mirror of
https://github.com/thecyberlearn/quantum-ai.git
synced 2026-08-18 09:53:00 +00:00
bfbb7d3515
## Issues Fixed - Added missing status endpoint for polling mechanism - Fixed content extraction priority to handle ad_copy field and raw_response.output - Corrected include_emoji field handling (form sends 'yes' not 'on') - Updated URL patterns to include proper status and result endpoints ## Technical Changes - Added social_ads_generator_status() view for polling - Updated content extraction priority: ad_copy_content > content > ad_copy > raw_response.output - Fixed boolean field handling for include_emoji checkbox - Consistent with other agents' polling patterns ## URL Updates - Added /status/<uuid:request_id>/ for polling - Added /result/<uuid:request_id>/ for results - Consistent API structure across all agents ## Content Field Priority Based on actual JSON response showing "output" field: 1. ad_copy_content (primary field) 2. content (fallback) 3. ad_copy (model field) 4. raw_response.output (actual webhook response) 5. formatted_ad (processed content) 6. output_text (legacy field) ## Result - Social ads generator now properly handles form submission - Correct content extraction from webhook response - Consistent polling mechanism with other agents - No more network errors during ad generation 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
204 lines
8.4 KiB
Python
204 lines
8.4 KiB
Python
from django.shortcuts import render, redirect
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.contrib import messages
|
|
from django.http import JsonResponse
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
from django.utils.decorators import method_decorator
|
|
from django.views import View
|
|
from agent_base.models import BaseAgent
|
|
from .models import SocialAdsGeneratorRequest, SocialAdsGeneratorResponse
|
|
from .processor import SocialAdsGeneratorProcessor
|
|
import json
|
|
|
|
|
|
def social_ads_generator_detail(request):
|
|
"""Detail page for Social Ads Generator agent"""
|
|
try:
|
|
agent = BaseAgent.objects.get(slug='social-ads-generator')
|
|
except BaseAgent.DoesNotExist:
|
|
messages.error(request, 'Social Ads Generator agent not found.')
|
|
return redirect('core:homepage')
|
|
|
|
if request.method == 'POST':
|
|
# Handle AJAX requests
|
|
if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
|
|
if not request.user.is_authenticated:
|
|
return JsonResponse({'error': 'Authentication required'}, status=401)
|
|
|
|
# Check wallet balance
|
|
if not request.user.has_sufficient_balance(agent.price):
|
|
return JsonResponse({'error': 'Insufficient wallet balance'}, status=400)
|
|
|
|
try:
|
|
# Create request object (no wallet deduction yet)
|
|
agent_request = SocialAdsGeneratorRequest.objects.create(
|
|
user=request.user,
|
|
agent=agent,
|
|
cost=agent.price,
|
|
description=request.POST.get('description'),
|
|
social_platform=request.POST.get('social_platform', 'facebook'),
|
|
include_emoji=request.POST.get('include_emoji') == 'yes',
|
|
language=request.POST.get('language', 'English'),
|
|
)
|
|
|
|
# Process request
|
|
processor = SocialAdsGeneratorProcessor()
|
|
result = processor.process_request(
|
|
request_obj=agent_request,
|
|
user_id=request.user.id,
|
|
)
|
|
|
|
# Refresh user from database to get updated wallet balance
|
|
request.user.refresh_from_db()
|
|
|
|
return JsonResponse({
|
|
'success': True,
|
|
'request_id': str(agent_request.id),
|
|
'message': 'Social ads generation started',
|
|
'wallet_balance': float(request.user.wallet_balance)
|
|
})
|
|
|
|
except Exception as e:
|
|
return JsonResponse({'error': str(e)}, status=500)
|
|
|
|
# Regular form submission (redirect to avoid resubmission)
|
|
return redirect('social_ads_generator:detail')
|
|
|
|
# GET request - show form
|
|
context = {
|
|
'agent': agent,
|
|
}
|
|
return render(request, 'social_ads_generator/detail.html', context)
|
|
|
|
|
|
@method_decorator(csrf_exempt, name='dispatch')
|
|
class SocialAdsGeneratorProcessView(View):
|
|
"""Process Social Ads Generator requests"""
|
|
|
|
def post(self, request):
|
|
if not request.user.is_authenticated:
|
|
return JsonResponse({'error': 'Authentication required'}, status=401)
|
|
|
|
try:
|
|
# Parse request data
|
|
data = json.loads(request.body)
|
|
|
|
# Get agent
|
|
agent = BaseAgent.objects.get(slug='social-ads-generator')
|
|
|
|
# Check wallet balance
|
|
if not request.user.has_sufficient_balance(agent.price):
|
|
return JsonResponse({'error': 'Insufficient wallet balance'}, status=400)
|
|
|
|
# Create request object (no wallet deduction yet - only after successful processing)
|
|
agent_request = SocialAdsGeneratorRequest.objects.create(
|
|
user=request.user,
|
|
agent=agent,
|
|
cost=agent.price,
|
|
description=data.get('description'),
|
|
social_platform=data.get('social_platform', 'facebook'),
|
|
include_emoji=data.get('include_emoji', False),
|
|
language=data.get('language', 'English'),
|
|
)
|
|
|
|
# Process request
|
|
processor = SocialAdsGeneratorProcessor()
|
|
result = processor.process_request(
|
|
request_obj=agent_request,
|
|
user_id=request.user.id,
|
|
)
|
|
|
|
# Refresh user from database to get updated wallet balance
|
|
request.user.refresh_from_db()
|
|
|
|
return JsonResponse({
|
|
'success': True,
|
|
'request_id': str(agent_request.id),
|
|
'message': 'Social Ads Generator request processed successfully',
|
|
'wallet_balance': float(request.user.wallet_balance)
|
|
})
|
|
|
|
except BaseAgent.DoesNotExist:
|
|
return JsonResponse({'error': 'Social Ads Generator agent not found'}, status=404)
|
|
except Exception as e:
|
|
return JsonResponse({'error': str(e)}, status=500)
|
|
|
|
|
|
@login_required
|
|
def social_ads_generator_status(request, request_id):
|
|
"""Get status for a specific request (for polling)"""
|
|
try:
|
|
agent_request = SocialAdsGeneratorRequest.objects.get(
|
|
id=request_id,
|
|
user=request.user
|
|
)
|
|
|
|
if hasattr(agent_request, 'response'):
|
|
response = agent_request.response
|
|
# Refresh user to get current wallet balance
|
|
request.user.refresh_from_db()
|
|
|
|
return JsonResponse({
|
|
'success': response.success,
|
|
'status': agent_request.status,
|
|
'content': getattr(response, 'ad_copy', None),
|
|
'ad_copy_content': getattr(response, 'ad_copy', None),
|
|
'hashtags': getattr(response, 'hashtags', None),
|
|
'targeting_suggestions': getattr(response, 'targeting_suggestions', None),
|
|
'formatted_ad': getattr(response, 'formatted_ad', None),
|
|
'raw_response': getattr(response, 'raw_response', None),
|
|
'processing_time': float(response.processing_time) if response.processing_time else None,
|
|
'error_message': response.error_message,
|
|
'wallet_balance': float(request.user.wallet_balance)
|
|
})
|
|
else:
|
|
return JsonResponse({
|
|
'success': False,
|
|
'status': agent_request.status,
|
|
'message': 'Processing in progress...'
|
|
})
|
|
|
|
except SocialAdsGeneratorRequest.DoesNotExist:
|
|
return JsonResponse({'error': 'Request not found'}, status=404)
|
|
except Exception as e:
|
|
return JsonResponse({'error': str(e)}, status=500)
|
|
|
|
|
|
@login_required
|
|
def social_ads_generator_result(request, request_id):
|
|
"""Get result for a specific request"""
|
|
try:
|
|
agent_request = SocialAdsGeneratorRequest.objects.get(
|
|
id=request_id,
|
|
user=request.user
|
|
)
|
|
|
|
if hasattr(agent_request, 'response'):
|
|
response = agent_request.response
|
|
# Refresh user to get current wallet balance
|
|
request.user.refresh_from_db()
|
|
|
|
return JsonResponse({
|
|
'success': response.success,
|
|
'status': agent_request.status,
|
|
'content': getattr(response, 'ad_copy', None),
|
|
'ad_copy_content': getattr(response, 'ad_copy', None),
|
|
'hashtags': getattr(response, 'hashtags', None),
|
|
'targeting_suggestions': getattr(response, 'targeting_suggestions', None),
|
|
'formatted_ad': getattr(response, 'formatted_ad', None),
|
|
'raw_response': getattr(response, 'raw_response', None),
|
|
'processing_time': float(response.processing_time) if response.processing_time else None,
|
|
'error_message': response.error_message,
|
|
'wallet_balance': float(request.user.wallet_balance)
|
|
})
|
|
else:
|
|
return JsonResponse({
|
|
'success': False,
|
|
'status': agent_request.status,
|
|
'message': 'Processing in progress...'
|
|
})
|
|
|
|
except SocialAdsGeneratorRequest.DoesNotExist:
|
|
return JsonResponse({'error': 'Request not found'}, status=404)
|
|
except Exception as e:
|
|
return JsonResponse({'error': str(e)}, status=500) |