Add Generate Image Prompts Functionality: Implement new AI function for generating image prompts, update API endpoints, and integrate with frontend actions for content management.

2025-11-11 17:40:08 +00:00
parent f4d62448cf
commit fa696064e2
9 changed files with 333 additions and 16 deletions
--- a/backend/igny8_core/modules/writer/views.py
+++ b/backend/igny8_core/modules/writer/views.py
@@ -455,4 +455,57 @@ class ContentViewSet(SiteSectorModelViewSet):
            serializer.save(account=account)
        else:
            serializer.save()
+    
+    @action(detail=False, methods=['post'], url_path='generate_image_prompts', url_name='generate_image_prompts')
+    def generate_image_prompts(self, request):
+        """Generate image prompts for content records - same pattern as other AI functions"""
+        from igny8_core.ai.tasks import run_ai_task
+        
+        account = getattr(request, 'account', None)
+        ids = request.data.get('ids', [])
+        
+        if not ids:
+            return Response({
+                'error': 'No IDs provided',
+                'type': 'ValidationError'
+            }, status=status.HTTP_400_BAD_REQUEST)
+        
+        account_id = account.id if account else None
+        
+        # Queue Celery task
+        try:
+            if hasattr(run_ai_task, 'delay'):
+                task = run_ai_task.delay(
+                    function_name='generate_image_prompts',
+                    payload={'ids': ids},
+                    account_id=account_id
+                )
+                return Response({
+                    'success': True,
+                    'task_id': str(task.id),
+                    'message': 'Image prompt generation started'
+                }, status=status.HTTP_200_OK)
+            else:
+                # Fallback to synchronous execution
+                result = run_ai_task(
+                    function_name='generate_image_prompts',
+                    payload={'ids': ids},
+                    account_id=account_id
+                )
+                if result.get('success'):
+                    return Response({
+                        'success': True,
+                        'prompts_created': result.get('count', 0),
+                        'message': 'Image prompts generated successfully'
+                    }, status=status.HTTP_200_OK)
+                else:
+                    return Response({
+                        'error': result.get('error', 'Image prompt generation failed'),
+                        'type': 'TaskExecutionError'
+                    }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        except Exception as e:
+            return Response({
+                'error': str(e),
+                'type': 'ExecutionError'
+            }, status=status.HTTP_500_INTERNAL_SERVER_ERROR)