removeing unneceary code

This commit is contained in:
Desktop
2025-11-12 04:20:43 +05:00
parent d1d2d768e5
commit 4bd158ce01
10 changed files with 130 additions and 1004 deletions

View File

@@ -514,73 +514,6 @@ class ImagesViewSet(SiteSectorModelViewSet):
'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'):
# Include queued prompts in response for TEST MODE
response_data = {
'success': True,
'images_generated': result.get('images_generated', 0),
'images_failed': result.get('images_failed', 0),
'count': result.get('count', 0),
'total_images': result.get('total_images', 0),
'message': 'Images generated successfully'
}
# Add test mode data if available
if result.get('queued_prompts'):
response_data['queued_prompts'] = result.get('queued_prompts')
response_data['test_mode'] = result.get('test_mode', False)
response_data['provider'] = result.get('provider')
response_data['model'] = result.get('model')
logger.info(f"[generate_images] Response: {response_data}")
return Response(response_data, 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):
"""
ViewSet for managing task content