Update main.py

main.py

@@ -969,7 +969,6 @@ KNOWN_VOICE_IDS = {
 
 class ConversationalAIHandler:
     def __init__(self):
-        # 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:
@@ -981,20 +980,17 @@ class ConversationalAIHandler:
         }
 
     def get_british_male_voice_id(self):
-        """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
+        return KNOWN_VOICE_IDS.get("Daniel", "onwK4e9ZLuTAKqWW03F9")
 
     def create_or_get_agent(self, project_id, project_data):
-        """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...")
-                verify_url = f"{self.base_url}/agents/{existing_agent_id}"
+                verify_url = f"{self.base_url}/convai/agents/{existing_agent_id}"
                 try:
                     response = requests.get(verify_url, headers=self.headers, timeout=15)
                     if response.status_code == 200:
@@ -1005,7 +1001,7 @@ 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 as per API documentation.")
+            logger.info(f"[AGENT] Creating new agent as per Convai API documentation.")
             voice_id = self.get_british_male_voice_id()
 
             diy_expert_prompt = f"""
@@ -1019,35 +1015,30 @@ class ConversationalAIHandler:
             Ask clarifying questions when needed and share relevant tips from your experience.
             """
 
-            # 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 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"
+                "conversation_config": {
+                    "name": f"DIY Expert - {project_data.get('projectTitle', project_id)}",
+                    "voice_id": voice_id,
+                    "model_id": "eleven_turbo_v2",
+                    "first_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?",
+                    "prompt": diy_expert_prompt
+                }
             }
 
-            # This is the exact endpoint from the documentation.
-            creation_url = f"{self.base_url}/agents"
+            creation_url = f"{self.base_url}/convai/agents/create"
             logger.info(f"[AGENT] Posting to official endpoint: {creation_url}")
-            
+
             response = requests.post(creation_url, headers=self.headers, json=agent_payload, timeout=30)
-            
-            # A successful creation returns HTTP 200 OK.
+
             if response.status_code == 200:
                 agent_info = response.json()
                 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})
                 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.")
+                logger.error("[AGENT] ACTION REQUIRED: This error may indicate a subscription issue or malformed payload.")
                 raise Exception(f"Failed to create agent: HTTP {response.status_code} - {response.text}")
 
         except Exception as e:
@@ -1055,13 +1046,12 @@ class ConversationalAIHandler:
             raise
 
     def start_conversation(self, agent_id):
-        """Starts a conversation with the created agent."""
         try:
             logger.info(f"[CONVERSATION] Starting conversation with agent: {agent_id}")
-            conversation_url = f"{self.base_url}/agents/{agent_id}/conversations"
+            conversation_url = f"{self.base_url}/convai/agents/{agent_id}/conversations"
 
             response = requests.post(conversation_url, headers=self.headers, json={}, timeout=30)
-            
+
             if response.status_code == 200:
                 conversation_info = response.json()
                 conversation_id = conversation_info.get("conversation_id")
@@ -1075,9 +1065,7 @@ class ConversationalAIHandler:
             raise
 
     def send_message(self, agent_id, conversation_id, message):
-        """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}")
             message_url = f"{self.base_url}/conversations/{conversation_id}/messages"
             payload = {"text": message}
@@ -1094,7 +1082,6 @@ class ConversationalAIHandler:
             logger.error(f"[MESSAGE] An exception occurred in send_message: {e}")
             raise
 
-
 def calculate_cost(duration_seconds):
     """Calculate cost based on conversation duration"""
     minutes = (duration_seconds + 59) // 60