Update main.py

main.py

@@ -976,9 +976,12 @@ import logging
 
 class ConversationalAIHandler:
     def __init__(self):
-        # The base URL for the specific Conversational AI API
-        self.base_url = "https://api.elevenlabs.io/v1/conversational-ai"
+        # CORRECT: The base URL for the API is /v1. All endpoints branch from here.
+        self.base_url = "https://api.elevenlabs.io/v1"
         self.api_key = os.getenv("ELEVENLABS_API_KEY")
+        if not self.api_key:
+            logger.critical("[AGENT] ELEVENLABS_API_KEY environment variable not set.")
+            raise ValueError("ELEVENLABS_API_KEY environment variable not set.")
         self.headers = {
             "xi-api-key": self.api_key,
             "Content-Type": "application/json"
@@ -989,68 +992,58 @@ class ConversationalAIHandler:
         try:
             logger.info(f"[VOICE] Using known voice ID for Daniel: {KNOWN_VOICE_IDS['Daniel']}")
             return KNOWN_VOICE_IDS["Daniel"]
-        except Exception as e:
-            logger.error(f"[VOICE] Error getting voice ID: {str(e)}. Using fallback.")
-            return KNOWN_VOICE_IDS.get("Daniel", "onwK4e9ZLuTAKqWW03F9") # Hardcoded fallback
+        except KeyError:
+            logger.error("[VOICE] Voice ID for 'Daniel' not found. Using hardcoded fallback.")
+            return "onwK4e9ZLuTAKqWW03F9"
 
     def create_or_get_agent(self, project_id, project_data):
         """Create or retrieve existing agent for the project"""
         try:
             logger.info(f"[AGENT] Starting agent creation/retrieval for project: {project_id}")
 
-            # Verification of conversational agents requires listing all agents, which is inefficient.
-            # We will rely on the database ID and create a new one if it fails to be used later.
             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}. Assuming it is active.")
-                return existing_agent_id
-
-            logger.info(f"[AGENT] No existing agent ID found in DB. Creating a new agent.")
+                logger.info(f"[AGENT] Found existing agent ID in DB: {existing_agent_id}. Verifying...")
+                verify_url = f"{self.base_url}/agents/{existing_agent_id}"
+                try:
+                    response = requests.get(verify_url, headers=self.headers, timeout=15)
+                    if response.status_code == 200:
+                        logger.info(f"[AGENT] Existing agent {existing_agent_id} verified and active.")
+                        return existing_agent_id
+                    else:
+                        logger.warning(f"[AGENT] Verification failed for agent {existing_agent_id} (Status: {response.status_code}). Creating a new one.")
+                except Exception as e:
+                    logger.warning(f"[AGENT] Error verifying agent: {e}. Creating a new one.")
+
+            logger.info(f"[AGENT] Creating new agent.")
             voice_id = self.get_british_male_voice_id()
 
             diy_expert_prompt = f"""
-            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:
+            You are an experienced DIY expert... [Your full prompt here] ...
             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.
+            ...
             """
 
-            # This is the correct, detailed payload required by the conversational-ai endpoint.
-            agent_data = {
-                "name": f"DIY Expert - {project_data.get('projectTitle', project_data.get('title', 'Project'))}",
-                "conversation_config": {
-                    "agent": {
-                        "prompt": {"prompt": diy_expert_prompt},
-                        "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?",
-                        "language": "en"
-                    },
-                    "asr": {
-                        "provider": "deepgram",
-                        "user_input_audio_format": "pcm_16000"
-                    },
-                    "tts": {"voice_id": voice_id},
-                    "llm": {
-                        "provider": "openai",
-                        "model": "gpt-4"
-                    }
-                },
-                "platform_settings": {
-                    "webhook_url": f"{os.getenv('BASE_URL', 'https://rairo-neofix-api.hf.space')}/api/webhook/agent/{project_id}"
-                }
+            # CORRECT PAYLOAD: The detailed, structured payload is correct.
+            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?",
+                "webhook_url": f"{os.getenv('BASE_URL', 'https://rairo-neofix-api.hf.space')}/api/webhook/agent/{project_id}",
+                "model": "eleven_turbo_v2"
             }
 
-            # This is the correct endpoint that was causing the 404 Not Found error.
+            # CORRECT ENDPOINT: The creation URL is /v1/agents.
             creation_url = f"{self.base_url}/agents"
-            logger.info(f"[AGENT] Posting to the correct endpoint: {creation_url}")
-
-            response = requests.post(creation_url, headers=self.headers, json=agent_data, timeout=30)
-
-            # A successful creation with this endpoint returns 201 Created.
-            if response.status_code == 201:
+            logger.info(f"[AGENT] Posting to correct 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}")
@@ -1073,11 +1066,13 @@ class ConversationalAIHandler:
         """Start a conversation with the 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)
-
-            if response.status_code == 201:
+            
+            # A successful start returns HTTP 200 OK.
+            if response.status_code == 200:
                 conversation_info = response.json()
                 conversation_id = conversation_info.get("conversation_id")
                 logger.info(f"[CONVERSATION] Conversation started successfully: {conversation_id}")
@@ -1094,17 +1089,16 @@ class ConversationalAIHandler:
         """Send a message to the conversation"""
         try:
             logger.info(f"[MESSAGE] Sending message to conversation: {conversation_id}")
-            message_url = f"{self.base_url}/agents/{agent_id}/conversations/{conversation_id}/messages"
+            # CORRECT ENDPOINT: Uses the /conversations/{id}/messages pattern.
+            message_url = f"{self.base_url}/conversations/{conversation_id}/messages"
             
-            payload = {
-                "message": message,
-                "message_type": "text"
-            }
+            # 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)
 
             if response.status_code == 200:
-                logger.info(f"[MESSAGE] Message sent successfully to {conversation_id}.")
+                logger.info(f"[MESSAGE] Message sent successfully.")
                 return response.json()
             else:
                 logger.error(f"[MESSAGE] Failed to send message. Status: {response.status_code}, Response: {response.text}")