Resolving Timeout error

agent.py

@@ -23,7 +23,7 @@ class AnalysisDescription(BaseModel):
     metadata: Dict[str, Any]
     confidence_score: float
 
-MODEL = "mistral-medium-latest"
+MODEL = "mistral-large-latest"
 
 def play_wav(url: str, save_path: str = "/tmp/audio.wav"):
     try:
@@ -242,9 +242,15 @@ def simulate_analyze_json_data(json_data: Dict[str, Any], analysis_type: str = "
 def simulate_text_to_speech(text: str, voice_settings: Dict[str, Any] = None) -> str:
     print(f"Converting to speech: {text[:100]}...")
     save_path = "/tmp/generated_speech.wav"
-    tts = gTTS(text=text, lang="en")
-    tts.save(save_path)
-    return f"file://{os.path.abspath(save_path)}"
+    try:
+        tts = gTTS(text=text, lang="en")
+        tts.save(save_path)
+        if not os.path.exists(save_path):
+            raise FileNotFoundError(f"Failed to save audio to {save_path}")
+        return f"file://{os.path.abspath(save_path)}"
+    except Exception as e:
+        print(f"Error generating speech: {str(e)}")
+        return f"Error: {str(e)}"
 
 async def process_document_workflow(client: Mistral, file_path: str, document_type: str = "climate_report"):
     print("Starting document processing workflow...")

app.py

@@ -29,6 +29,11 @@ def initialize_client_and_agents(api_key: str):
                 "json_analyzer_agent_id": json_analyzer_agent.id,
                 "speech_agent_id": speech_agent.id
             }
+            print("Agents initialized successfully.")
+            print(f"Document Agent ID: {doc_agent.id}")
+            print(f"Image Agent ID: {image_agent.id}")
+            print(f"JSON Analyzer Agent ID: {json_analyzer_agent.id}")
+            print(f"Speech Agent ID: {speech_agent.id}")
         except Exception as e:
             return None, f"Error initializing client: {str(e)}"
     return client, agents
@@ -241,4 +246,4 @@ with gr.Blocks(css=custom_css) as demo:
         )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)