Add Image Generation from Prompts: Implement new functionality to generate images from prompts, including backend processing, API integration, and frontend handling with progress modal. Update settings and registry for new AI function.

backend/igny8_core/modules/writer/views.py

@@ -513,6 +513,60 @@ class ImagesViewSet(SiteSectorModelViewSet):
             'count': len(grouped_data),
             'results': grouped_data
         }, status=status.HTTP_200_OK)
+    
+    @action(detail=False, methods=['post'], url_path='generate_images', url_name='generate_images')
+    def generate_images(self, request):
+        """Generate images from prompts for image records"""
+        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_images_from_prompts',
+                    payload={'ids': ids},
+                    account_id=account_id
+                )
+                return Response({
+                    'success': True,
+                    'task_id': str(task.id),
+                    'message': 'Image generation started'
+                }, status=status.HTTP_200_OK)
+            else:
+                # Fallback to synchronous execution
+                result = run_ai_task(
+                    function_name='generate_images_from_prompts',
+                    payload={'ids': ids},
+                    account_id=account_id
+                )
+                if result.get('success'):
+                    return Response({
+                        'success': True,
+                        'images_generated': result.get('count', 0),
+                        'images_failed': result.get('images_failed', 0),
+                        'message': 'Images generated successfully'
+                    }, status=status.HTTP_200_OK)
+                else:
+                    return Response({
+                        'error': result.get('error', 'Image 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)
 
 
 class ContentViewSet(SiteSectorModelViewSet):