Update main.py

main.py

@@ -966,17 +966,10 @@ KNOWN_VOICE_IDS = {
     "Antoni": "ErXwobaYiN019PkySvjV",  # American male (fallback)
 }
 
-import os
-import time
-import requests
-import logging
-
-# Assumes 'db_ref', 'logger', and 'KNOWN_VOICE_IDS' are initialized elsewhere in your application.
-# from elevenlabs import ElevenLabs # This is no longer needed for the handler's logic.
 
 class ConversationalAIHandler:
     def __init__(self):
-        # CORRECT: The base URL for the API is /v1. All endpoints branch from here.
+        # Per documentation, the base URL for all v1 API calls.
         self.base_url = "https://api.elevenlabs.io/v1"
         self.api_key = os.getenv("ELEVENLABS_API_KEY")
         if not self.api_key:
@@ -988,19 +981,16 @@ class ConversationalAIHandler:
         }
 
     def get_british_male_voice_id(self):
-        """Get British male voice ID"""
-        try:
-            logger.info(f"[VOICE] Using known voice ID for Daniel: {KNOWN_VOICE_IDS['Daniel']}")
-            return KNOWN_VOICE_IDS["Daniel"]
-        except KeyError:
-            logger.error("[VOICE] Voice ID for 'Daniel' not found. Using hardcoded fallback.")
-            return "onwK4e9ZLuTAKqWW03F9"
+        """Gets a known British male voice ID."""
+        logger.info(f"[VOICE] Using known voice ID for Daniel.")
+        return KNOWN_VOICE_IDS.get("Daniel", "onwK4e9ZLuTAKqWW03F9") # Default fallback
 
     def create_or_get_agent(self, project_id, project_data):
-        """Create or retrieve existing agent for the project"""
+        """Creates or retrieves an agent, strictly following the /v1/agents documentation."""
         try:
             logger.info(f"[AGENT] Starting agent creation/retrieval for project: {project_id}")
 
+            # First, check for and verify an existing agent in the database.
             existing_agent_id = db_ref.child(f'projects/{project_id}/agent_id').get()
             if existing_agent_id:
                 logger.info(f"[AGENT] Found existing agent ID in DB: {existing_agent_id}. Verifying...")
@@ -1015,30 +1005,34 @@ class ConversationalAIHandler:
                 except Exception as e:
                     logger.warning(f"[AGENT] Error verifying agent: {e}. Creating a new one.")
 
-            logger.info(f"[AGENT] Creating new agent.")
+            logger.info(f"[AGENT] Creating new agent as per API documentation.")
             voice_id = self.get_british_male_voice_id()
 
             diy_expert_prompt = f"""
-            You are an experienced DIY expert... [Your full prompt here] ...
+            You are an experienced DIY expert with years of hands-on experience in Home Appliance Repair, Automotive Maintenance, Gardening & Urban Farming,
+            Upcycling & Sustainable Crafts, or DIY Project Creation. You speak in a friendly, knowledgeable British manner and provide 
+            practical, actionable advice. You're working on this specific project:
             Project: {project_data.get('projectTitle', 'DIY Project')}
             Description: {project_data.get('projectDescription', '')}
             Initial Plan: {project_data.get('initialPlan', '')}
-            ...
+            Provide helpful, step-by-step guidance while being encouraging and safety-conscious. 
+            Ask clarifying questions when needed and share relevant tips from your experience.
             """
 
-            # CORRECT PAYLOAD: The detailed, structured payload is correct.
+            # This payload EXACTLY matches the fields in the /v1/agents/create documentation.
+            # The complex 'conversation_config' is NOT part of this documented endpoint.
             agent_payload = {
                 "name": f"DIY Expert - {project_data.get('projectTitle', project_id)}",
                 "prompt": diy_expert_prompt,
                 "voice_id": voice_id,
-                "initial_message": "Hello! I'm your Sozo DIY assistant. I'm here to help you with your project. What would you like to work on today?",
+                "initial_message": "Hello! I'm your Sozo DIY expert assistant. I'm here to help you with your project. What would you like to work on today?",
                 "webhook_url": f"{os.getenv('BASE_URL', 'https://rairo-neofix-api.hf.space')}/api/webhook/agent/{project_id}",
                 "model": "eleven_turbo_v2"
             }
 
-            # CORRECT ENDPOINT: The creation URL is /v1/agents.
+            # This is the exact endpoint from the documentation.
             creation_url = f"{self.base_url}/agents"
-            logger.info(f"[AGENT] Posting to correct endpoint: {creation_url}")
+            logger.info(f"[AGENT] Posting to official endpoint: {creation_url}")
             
             response = requests.post(creation_url, headers=self.headers, json=agent_payload, timeout=30)
             
@@ -1048,14 +1042,12 @@ class ConversationalAIHandler:
                 agent_id = agent_info.get("agent_id")
                 logger.info(f"[AGENT] New agent created successfully: {agent_id}")
 
-                db_ref.child(f'projects/{project_id}').update({
-                    'agent_id': agent_id,
-                    'agent_created_at': int(time.time())
-                })
-                logger.info(f"[AGENT] Agent ID stored in database.")
+                db_ref.child(f'projects/{project_id}').update({'agent_id': agent_id})
                 return agent_id
             else:
+                # If this 404 error persists, the issue is not in the code but in the environment/account.
                 logger.error(f"[AGENT] Failed to create agent. Status: {response.status_code}, Response: {response.text}")
+                logger.error("[AGENT] ACTION REQUIRED: This 404 error indicates the API endpoint was not found for your account. Please verify your ElevenLabs subscription includes the Agents API.")
                 raise Exception(f"Failed to create agent: HTTP {response.status_code} - {response.text}")
 
         except Exception as e:
@@ -1063,15 +1055,13 @@ class ConversationalAIHandler:
             raise
 
     def start_conversation(self, agent_id):
-        """Start a conversation with the agent"""
+        """Starts a conversation with the created agent."""
         try:
             logger.info(f"[CONVERSATION] Starting conversation with agent: {agent_id}")
-            # CORRECT ENDPOINT: Follows the /agents/{id}/conversations pattern.
             conversation_url = f"{self.base_url}/agents/{agent_id}/conversations"
 
             response = requests.post(conversation_url, headers=self.headers, json={}, timeout=30)
             
-            # A successful start returns HTTP 200 OK.
             if response.status_code == 200:
                 conversation_info = response.json()
                 conversation_id = conversation_info.get("conversation_id")
@@ -1080,19 +1070,16 @@ class ConversationalAIHandler:
             else:
                 logger.error(f"[CONVERSATION] Failed to start conversation. Status: {response.status_code}, Response: {response.text}")
                 raise Exception(f"Failed to start conversation: {response.text}")
-
         except Exception as e:
             logger.error(f"[CONVERSATION] An exception occurred in start_conversation: {e}")
             raise
 
     def send_message(self, agent_id, conversation_id, message):
-        """Send a message to the conversation"""
+        """Sends a message to an active conversation."""
         try:
+            # Note: agent_id is not used in the URL but kept for consistent function signature.
             logger.info(f"[MESSAGE] Sending message to conversation: {conversation_id}")
-            # CORRECT ENDPOINT: Uses the /conversations/{id}/messages pattern.
             message_url = f"{self.base_url}/conversations/{conversation_id}/messages"
-            
-            # CORRECT PAYLOAD: The payload requires the "text" of the message.
             payload = {"text": message}
 
             response = requests.post(message_url, headers=self.headers, json=payload, timeout=30)
@@ -1103,7 +1090,6 @@ class ConversationalAIHandler:
             else:
                 logger.error(f"[MESSAGE] Failed to send message. Status: {response.status_code}, Response: {response.text}")
                 raise Exception(f"Failed to send message: {response.text}")
-
         except Exception as e:
             logger.error(f"[MESSAGE] An exception occurred in send_message: {e}")
             raise