diff --git a/agents/admin.py b/agents/admin.py index 9bcf368..e2f55c6 100644 --- a/agents/admin.py +++ b/agents/admin.py @@ -1,5 +1,5 @@ from django.contrib import admin -from .models import AgentCategory, Agent, AgentExecution +from .models import AgentCategory, Agent, AgentExecution, ChatSession, ChatMessage @admin.register(AgentCategory) class AgentCategoryAdmin(admin.ModelAdmin): @@ -10,8 +10,8 @@ class AgentCategoryAdmin(admin.ModelAdmin): @admin.register(Agent) class AgentAdmin(admin.ModelAdmin): - list_display = ['name', 'category', 'price', 'is_active', 'created_at'] - list_filter = ['category', 'is_active', 'created_at'] + list_display = ['name', 'category', 'agent_type', 'price', 'is_active', 'created_at'] + list_filter = ['category', 'agent_type', 'is_active', 'created_at'] search_fields = ['name', 'description', 'short_description'] prepopulated_fields = {'slug': ('name',)} readonly_fields = ['created_at', 'updated_at'] @@ -22,3 +22,21 @@ class AgentExecutionAdmin(admin.ModelAdmin): list_filter = ['status', 'created_at', 'agent__category'] search_fields = ['agent__name', 'user__email'] readonly_fields = ['created_at', 'completed_at'] + +@admin.register(ChatSession) +class ChatSessionAdmin(admin.ModelAdmin): + list_display = ['session_id', 'agent', 'user', 'status', 'fee_charged', 'created_at'] + list_filter = ['status', 'agent__category', 'created_at'] + search_fields = ['session_id', 'agent__name', 'user__email'] + readonly_fields = ['session_id', 'created_at', 'updated_at', 'completed_at'] + +@admin.register(ChatMessage) +class ChatMessageAdmin(admin.ModelAdmin): + list_display = ['session', 'message_type', 'content_preview', 'timestamp'] + list_filter = ['message_type', 'timestamp'] + search_fields = ['session__session_id', 'content'] + readonly_fields = ['timestamp'] + + def content_preview(self, obj): + return obj.content[:50] + "..." if len(obj.content) > 50 else obj.content + content_preview.short_description = 'Content Preview' diff --git a/agents/management/commands/create_five_whys_agent.py b/agents/management/commands/create_five_whys_agent.py new file mode 100644 index 0000000..dac2df0 --- /dev/null +++ b/agents/management/commands/create_five_whys_agent.py @@ -0,0 +1,76 @@ +from django.core.management.base import BaseCommand +from agents.models import AgentCategory, Agent + +class Command(BaseCommand): + help = 'Create the 5 Whys chat-based analysis agent' + + def handle(self, *args, **options): + # Create or get the Analysis category + category, created = AgentCategory.objects.get_or_create( + slug='analysis', + defaults={ + 'name': 'Analysis & Problem Solving', + 'description': 'Advanced analytical tools for problem-solving and decision making', + 'icon': '🧠', + 'is_active': True + } + ) + + if created: + self.stdout.write(f'✅ Created category: {category.name}') + else: + self.stdout.write(f'📂 Using existing category: {category.name}') + + # Create the 5 Whys agent + agent, created = Agent.objects.get_or_create( + slug='five-whys-analysis', + defaults={ + 'name': '5 Whys Analysis', + 'short_description': 'Interactive problem-solving using the proven 5 Whys methodology', + 'description': '''Discover the root cause of any problem through guided conversation using the 5 Whys technique. + +This interactive agent helps you systematically drill down to the core issue by asking "why" five times. Perfect for: +• Troubleshooting operational problems +• Understanding process failures +• Identifying systemic issues +• Improving quality and efficiency + +The conversation-based approach ensures you think deeply about each layer of the problem, leading to more effective solutions.''', + 'category': category, + 'price': 15.00, + 'agent_type': 'chat', # This is a chat-based agent + 'form_schema': None, # Chat agents don't use form schemas + 'webhook_url': 'http://localhost:5678/webhook/5-whys-web', # N8N webhook URL + 'is_active': True + } + ) + + if created: + self.stdout.write( + self.style.SUCCESS(f'🎉 Successfully created 5 Whys Analysis agent!') + ) + self.stdout.write(f' 💬 Agent Type: {agent.agent_type}') + self.stdout.write(f' 💰 Price: {agent.price} AED') + self.stdout.write(f' 🔗 Webhook: {agent.webhook_url}') + self.stdout.write(f' 📂 Category: {agent.category.name}') + else: + self.stdout.write( + self.style.WARNING(f'⚠️ 5 Whys Analysis agent already exists') + ) + + # Update existing agent to ensure it's chat-based + if agent.agent_type != 'chat': + agent.agent_type = 'chat' + agent.form_schema = None + agent.save() + self.stdout.write( + self.style.SUCCESS(f'✅ Updated existing agent to chat-based') + ) + + self.stdout.write('') + self.stdout.write('🚀 Next steps:') + self.stdout.write(' 1. Ensure N8N webhook is running on localhost:5678') + self.stdout.write(' 2. Visit /agents/five-whys-analysis/ to test the chat interface') + self.stdout.write(' 3. Start a conversation to test the 5 Whys methodology') + self.stdout.write('') + self.stdout.write('💡 The agent is now ready for interactive problem-solving!') \ No newline at end of file diff --git a/agents/migrations/0002_agent_agent_type_alter_agent_form_schema_chatsession_and_more.py b/agents/migrations/0002_agent_agent_type_alter_agent_form_schema_chatsession_and_more.py new file mode 100644 index 0000000..798c6ea --- /dev/null +++ b/agents/migrations/0002_agent_agent_type_alter_agent_form_schema_chatsession_and_more.py @@ -0,0 +1,160 @@ +# Generated by Django 5.2.4 on 2025-08-01 04:01 + +import django.db.models.deletion +import uuid +from django.conf import settings +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ("agents", "0001_initial"), + migrations.swappable_dependency(settings.AUTH_USER_MODEL), + ] + + operations = [ + migrations.AddField( + model_name="agent", + name="agent_type", + field=models.CharField( + choices=[("form", "Form-based"), ("chat", "Chat-based")], + default="form", + help_text="Agent interaction type", + max_length=10, + ), + ), + migrations.AlterField( + model_name="agent", + name="form_schema", + field=models.JSONField( + blank=True, help_text="JSON schema for agent input form", null=True + ), + ), + migrations.CreateModel( + name="ChatSession", + fields=[ + ( + "id", + models.UUIDField( + default=uuid.uuid4, + editable=False, + primary_key=True, + serialize=False, + ), + ), + ( + "session_id", + models.CharField( + help_text="Unique session identifier", + max_length=100, + unique=True, + ), + ), + ( + "status", + models.CharField( + choices=[ + ("active", "Active"), + ("completed", "Completed"), + ("abandoned", "Abandoned"), + ("failed", "Failed"), + ], + default="active", + max_length=20, + ), + ), + ( + "context_data", + models.JSONField( + default=dict, help_text="Session context and progress tracking" + ), + ), + ("fee_charged", models.DecimalField(decimal_places=2, max_digits=10)), + ("created_at", models.DateTimeField(auto_now_add=True)), + ("updated_at", models.DateTimeField(auto_now=True)), + ("completed_at", models.DateTimeField(blank=True, null=True)), + ( + "agent", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + related_name="chat_sessions", + to="agents.agent", + ), + ), + ( + "user", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + to=settings.AUTH_USER_MODEL, + ), + ), + ], + options={ + "ordering": ["-created_at"], + }, + ), + migrations.CreateModel( + name="ChatMessage", + fields=[ + ( + "id", + models.UUIDField( + default=uuid.uuid4, + editable=False, + primary_key=True, + serialize=False, + ), + ), + ( + "message_type", + models.CharField( + choices=[ + ("user", "User Message"), + ("agent", "Agent Response"), + ("system", "System Message"), + ], + max_length=10, + ), + ), + ("content", models.TextField()), + ( + "metadata", + models.JSONField( + default=dict, + help_text="Additional message data like webhook responses", + ), + ), + ("timestamp", models.DateTimeField(auto_now_add=True)), + ( + "session", + models.ForeignKey( + on_delete=django.db.models.deletion.CASCADE, + related_name="messages", + to="agents.chatsession", + ), + ), + ], + options={ + "ordering": ["timestamp"], + }, + ), + migrations.AddIndex( + model_name="chatsession", + index=models.Index( + fields=["session_id"], name="agents_chat_session_0d9cb4_idx" + ), + ), + migrations.AddIndex( + model_name="chatsession", + index=models.Index( + fields=["user", "-created_at"], name="agents_chat_user_id_f8983d_idx" + ), + ), + migrations.AddIndex( + model_name="chatmessage", + index=models.Index( + fields=["session", "timestamp"], name="agents_chat_session_e8eaed_idx" + ), + ), + ] diff --git a/agents/models.py b/agents/models.py index e5537df..25ae543 100644 --- a/agents/models.py +++ b/agents/models.py @@ -17,6 +17,11 @@ class AgentCategory(models.Model): return self.name class Agent(models.Model): + AGENT_TYPE_CHOICES = [ + ('form', 'Form-based'), + ('chat', 'Chat-based'), + ] + id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) name = models.CharField(max_length=200) slug = models.SlugField(unique=True) @@ -24,7 +29,8 @@ class Agent(models.Model): description = models.TextField() category = models.ForeignKey(AgentCategory, on_delete=models.CASCADE, related_name='agents') price = models.DecimalField(max_digits=10, decimal_places=2) - form_schema = models.JSONField(help_text="JSON schema for agent input form") + agent_type = models.CharField(max_length=10, choices=AGENT_TYPE_CHOICES, default='form', help_text="Agent interaction type") + form_schema = models.JSONField(help_text="JSON schema for agent input form", null=True, blank=True) webhook_url = models.URLField(help_text="n8n webhook URL for execution") is_active = models.BooleanField(default=True) created_at = models.DateTimeField(auto_now_add=True) @@ -62,3 +68,55 @@ class AgentExecution(models.Model): def __str__(self): return f"{self.agent.name} - {self.user.email} - {self.status}" + +class ChatSession(models.Model): + STATUS_CHOICES = [ + ('active', 'Active'), + ('completed', 'Completed'), + ('abandoned', 'Abandoned'), + ('failed', 'Failed'), + ] + + id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) + session_id = models.CharField(max_length=100, unique=True, help_text="Unique session identifier") + agent = models.ForeignKey(Agent, on_delete=models.CASCADE, related_name='chat_sessions') + user = models.ForeignKey('authentication.User', on_delete=models.CASCADE) + status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='active') + context_data = models.JSONField(default=dict, help_text="Session context and progress tracking") + fee_charged = models.DecimalField(max_digits=10, decimal_places=2) + created_at = models.DateTimeField(auto_now_add=True) + updated_at = models.DateTimeField(auto_now=True) + completed_at = models.DateTimeField(null=True, blank=True) + + class Meta: + ordering = ['-created_at'] + indexes = [ + models.Index(fields=['session_id']), + models.Index(fields=['user', '-created_at']), + ] + + def __str__(self): + return f"{self.agent.name} - {self.user.email} - {self.session_id}" + +class ChatMessage(models.Model): + MESSAGE_TYPE_CHOICES = [ + ('user', 'User Message'), + ('agent', 'Agent Response'), + ('system', 'System Message'), + ] + + id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) + session = models.ForeignKey(ChatSession, on_delete=models.CASCADE, related_name='messages') + message_type = models.CharField(max_length=10, choices=MESSAGE_TYPE_CHOICES) + content = models.TextField() + metadata = models.JSONField(default=dict, help_text="Additional message data like webhook responses") + timestamp = models.DateTimeField(auto_now_add=True) + + class Meta: + ordering = ['timestamp'] + indexes = [ + models.Index(fields=['session', 'timestamp']), + ] + + def __str__(self): + return f"{self.session.session_id} - {self.message_type} - {self.timestamp}" diff --git a/agents/templates/agents/agent_chat.html b/agents/templates/agents/agent_chat.html new file mode 100644 index 0000000..9a55fe6 --- /dev/null +++ b/agents/templates/agents/agent_chat.html @@ -0,0 +1,590 @@ +{% extends 'base.html' %} +{% load static %} + +{% block title %}{{ agent.name }} - Quantum Tasks AI{% endblock %} + +{% block extra_css %} + + +{% endblock %} + +{% block content %} + + +
+ + {% include "components/agent_header.html" with agent_title=agent.name agent_subtitle=agent.short_description %} + + + {% include "components/quick_agents_panel.html" %} + + {% if user.is_authenticated %} + {% if user.wallet_balance >= agent.price or chat_session %} + + {% if chat_session %} +
+
+ 💬 Chat Session: {{ chat_session.get_status_display }} +
+
ID: {{ chat_session.session_id }}
+
+ {% endif %} + + +
+
+

+ {{ agent.category.icon }} + {{ agent.name }} +

+

{{ agent.description }}

+
+ +
+ {% if messages %} + {% for message in messages %} +
+
+ {{ message.content|linebreaks }} +
{{ message.timestamp|date:"H:i" }}
+
+
+ {% endfor %} + {% else %} +
+
🤖
+
Welcome to {{ agent.name }}!
+
Send a message to start your conversation
+
+ {% endif %} +
+ +
+ {{ agent.name }} is typing... +
+ +
+ {% if not chat_session or chat_session.status == 'active' %} +
+ {% csrf_token %} + + +
+ {% else %} +
+ Chat session {{ chat_session.get_status_display|lower }}. + Start New Chat +
+ {% endif %} +
+
+ {% else %} + +
+
+

+ {{ agent.category.icon }} + {{ agent.name }} +

+
+
+
+
💰
+
Insufficient Balance
+
+ You need {{ agent.price }} AED to start a chat session. +
+ + 💰 Top Up Wallet + +
+
+
+ {% endif %} + {% else %} + +
+
+

+ {{ agent.category.icon }} + {{ agent.name }} +

+
+
+
+
🔐
+
Login Required
+
+ Please login to start chatting with {{ agent.name }}. +
+ + 🔐 Login to Continue + +
+
+
+ {% endif %} + + + {% include "components/how_it_works_widget.html" with steps="agents" %} +
+{% endblock %} + +{% block extra_js %} + + + +{% endblock %} \ No newline at end of file diff --git a/agents/urls.py b/agents/urls.py index ad7de97..3a46bf5 100644 --- a/agents/urls.py +++ b/agents/urls.py @@ -11,6 +11,13 @@ urlpatterns = [ path('api/execute/', views.execute_agent, name='execute_agent'), path('api/executions/', views.execution_list, name='execution_list'), path('api/executions//', views.execution_detail, name='execution_detail'), + + # Chat API endpoints + path('api/chat/start/', views.start_chat_session, name='start_chat_session'), + path('api/chat/send/', views.send_chat_message, name='send_chat_message'), + path('api/chat/history//', views.get_chat_history, name='get_chat_history'), + path('api/chat/end/', views.end_chat_session, name='end_chat_session'), + path('api/', views.agent_list, name='agent_list'), path('api//', views.agent_detail, name='agent_detail_api'), diff --git a/agents/views.py b/agents/views.py index 809cc6b..b89e977 100644 --- a/agents/views.py +++ b/agents/views.py @@ -7,7 +7,7 @@ from django.shortcuts import get_object_or_404, render from django.utils import timezone from django.contrib.auth.decorators import login_required from django.db import models -from .models import Agent, AgentExecution, AgentCategory +from .models import Agent, AgentExecution, AgentCategory, ChatSession, ChatMessage from .serializers import AgentSerializer, AgentExecutionSerializer import requests import json @@ -234,9 +234,14 @@ def format_agent_message(agent_slug, input_data): # Web interface views @login_required def agent_detail_view(request, slug): - """Render agent detail page with dynamic form""" + """Render agent detail page with dynamic form or chat interface""" agent = get_object_or_404(Agent, slug=slug, is_active=True) + # Handle chat-based agents + if agent.agent_type == 'chat': + return chat_agent_view(request, agent) + + # Handle form-based agents (existing behavior) context = { 'agent': agent, 'timestamp': int(time.time()) # For cache busting @@ -273,3 +278,271 @@ def agents_marketplace(request): } return render(request, 'agents/marketplace.html', context) + + +# Chat-based agent views +def chat_agent_view(request, agent): + """Render chat interface for chat-based agents""" + chat_session = None + messages = [] + + if request.user.is_authenticated: + # Get or create active chat session + chat_session = ChatSession.objects.filter( + agent=agent, + user=request.user, + status='active' + ).first() + + # Get session ID from URL parameter if resuming a session + session_id = request.GET.get('session') + if session_id and not chat_session: + chat_session = ChatSession.objects.filter( + session_id=session_id, + agent=agent, + user=request.user + ).first() + + # Get messages for the session + if chat_session: + messages = ChatMessage.objects.filter(session=chat_session).order_by('timestamp') + + context = { + 'agent': agent, + 'chat_session': chat_session, + 'messages': messages, + 'timestamp': int(time.time()) + } + + return render(request, 'agents/agent_chat.html', context) + + +@api_view(['POST']) +@permission_classes([IsAuthenticated]) +def start_chat_session(request): + """Start a new chat session""" + agent_slug = request.data.get('agent_slug') + + if not agent_slug: + return Response({'error': 'agent_slug is required'}, status=status.HTTP_400_BAD_REQUEST) + + agent = get_object_or_404(Agent, slug=agent_slug, is_active=True, agent_type='chat') + + # Check wallet balance + if hasattr(request.user, 'wallet_balance') and request.user.wallet_balance < agent.price: + return Response({'error': 'Insufficient wallet balance'}, status=status.HTTP_400_BAD_REQUEST) + + # Check for existing active session + existing_session = ChatSession.objects.filter( + agent=agent, + user=request.user, + status='active' + ).first() + + if existing_session: + return Response({ + 'session_id': existing_session.session_id, + 'message': 'Active session already exists' + }) + + # Create new chat session + session_id = f"{int(time.time() * 1000)}_{uuid.uuid4().hex[:8]}" + + chat_session = ChatSession.objects.create( + session_id=session_id, + agent=agent, + user=request.user, + fee_charged=agent.price, + status='active' + ) + + # Deduct fee from wallet (if wallet system is implemented) + # This would integrate with the existing wallet system + + # Send welcome message + welcome_message = f"Welcome to {agent.name}! I'm here to help you with 5 Whys analysis. What problem would you like to analyze?" + + ChatMessage.objects.create( + session=chat_session, + message_type='agent', + content=welcome_message + ) + + return Response({ + 'session_id': chat_session.session_id, + 'message': 'Chat session started successfully' + }) + + +@api_view(['POST']) +@permission_classes([IsAuthenticated]) +def send_chat_message(request): + """Send a message in a chat session""" + session_id = request.data.get('session_id') + message_content = request.data.get('message', '').strip() + + if not session_id or not message_content: + return Response({'error': 'session_id and message are required'}, status=status.HTTP_400_BAD_REQUEST) + + # Get chat session + chat_session = get_object_or_404( + ChatSession, + session_id=session_id, + user=request.user, + status='active' + ) + + # Save user message + user_message = ChatMessage.objects.create( + session=chat_session, + message_type='user', + content=message_content + ) + + # Prepare webhook payload + webhook_payload = { + "message": { + "text": f"Chat message: {message_content}. Provide helpful guidance about 5 Whys analysis. Do not generate the final report - just chat and help the user understand their problem." + }, + "sessionId": session_id, + "userId": str(request.user.id), + "agentId": chat_session.agent.slug, + "messageType": "chat" + } + + try: + # Validate webhook URL + validate_webhook_url(chat_session.agent.webhook_url) + + # Send to webhook + response = requests.post( + chat_session.agent.webhook_url, + json=webhook_payload, + timeout=30, + headers={'Content-Type': 'application/json'} + ) + + if response.status_code == 200: + response_data = response.json() + agent_response = response_data.get('response', 'I received your message but couldn\'t generate a response.') + + # Save agent response + agent_message = ChatMessage.objects.create( + session=chat_session, + message_type='agent', + content=agent_response, + metadata={'webhook_response': response_data} + ) + + # Update session timestamp + chat_session.updated_at = timezone.now() + chat_session.save() + + return Response({ + 'user_message': { + 'id': str(user_message.id), + 'content': user_message.content, + 'timestamp': user_message.timestamp.isoformat() + }, + 'agent_message': { + 'id': str(agent_message.id), + 'content': agent_message.content, + 'timestamp': agent_message.timestamp.isoformat() + } + }) + else: + # Webhook error + error_message = "I'm having trouble processing your message right now. Please try again." + agent_message = ChatMessage.objects.create( + session=chat_session, + message_type='agent', + content=error_message, + metadata={'error': f'Webhook returned {response.status_code}'} + ) + + return Response({ + 'user_message': { + 'id': str(user_message.id), + 'content': user_message.content, + 'timestamp': user_message.timestamp.isoformat() + }, + 'agent_message': { + 'id': str(agent_message.id), + 'content': agent_message.content, + 'timestamp': agent_message.timestamp.isoformat() + } + }, status=status.HTTP_202_ACCEPTED) + + except Exception as e: + # Handle webhook errors + error_message = "I'm experiencing technical difficulties. Please try again later." + agent_message = ChatMessage.objects.create( + session=chat_session, + message_type='agent', + content=error_message, + metadata={'error': str(e)} + ) + + return Response({ + 'user_message': { + 'id': str(user_message.id), + 'content': user_message.content, + 'timestamp': user_message.timestamp.isoformat() + }, + 'agent_message': { + 'id': str(agent_message.id), + 'content': agent_message.content, + 'timestamp': agent_message.timestamp.isoformat() + } + }, status=status.HTTP_500_INTERNAL_SERVER_ERROR) + + +@api_view(['GET']) +@permission_classes([IsAuthenticated]) +def get_chat_history(request, session_id): + """Get chat history for a session""" + chat_session = get_object_or_404( + ChatSession, + session_id=session_id, + user=request.user + ) + + messages = ChatMessage.objects.filter(session=chat_session).order_by('timestamp') + + message_data = [] + for message in messages: + message_data.append({ + 'id': str(message.id), + 'message_type': message.message_type, + 'content': message.content, + 'timestamp': message.timestamp.isoformat() + }) + + return Response({ + 'session_id': session_id, + 'status': chat_session.status, + 'messages': message_data + }) + + +@api_view(['POST']) +@permission_classes([IsAuthenticated]) +def end_chat_session(request): + """End a chat session""" + session_id = request.data.get('session_id') + + if not session_id: + return Response({'error': 'session_id is required'}, status=status.HTTP_400_BAD_REQUEST) + + chat_session = get_object_or_404( + ChatSession, + session_id=session_id, + user=request.user, + status='active' + ) + + chat_session.status = 'completed' + chat_session.completed_at = timezone.now() + chat_session.save() + + return Response({'message': 'Chat session ended successfully'}) diff --git a/docs_update_summary.txt b/docs_update_summary.txt index 697b5a5..9c7264c 100644 --- a/docs_update_summary.txt +++ b/docs_update_summary.txt @@ -1,28 +1,17 @@ === Documentation Auto-Update Summary === -Update Date: 2025-08-01 09:22:55 +Update Date: 2025-08-01 09:23:11 Recent Commits: + - 11d28a1 📚 Update documentation after GitHub push - f6970b6 🎨 Complete Phase 1 UI optimization with button hover fixes - 277e7ec 📄 Auto-update documentation timestamp after security fixes - - c8ad34f 🔒 Implement critical security fixes for production readiness - -Agents Changes: - - agents/templates/agents/agent_detail.html - - agents/templates/agents/marketplace.html - - static/css/agent-base.css - - static/css/agent-detail.css Documentation Changes: - CLAUDE.md -Frontend Changes: - - static/css/base.css - - static/css/marketplace.css - Backend Changes: - docs_update_summary.txt -Updated Documentation Files: - - /home/amit/projects/quantum_ai_v2/CLAUDE.md +No documentation files required updates. === End Summary === \ No newline at end of file