Update app.py

app.py

@@ -18,6 +18,7 @@ from tools import (
     answer_image_question,
     answer_excel_question,
     answer_python_question,
+    answer_audio_question,
     get_youtube_transcript,
 )
 # (Keep Constants as is)
@@ -58,6 +59,7 @@ class BasicAgent:
             answer_image_question,
             answer_excel_question,
             answer_python_question,
+            answer_audio_question,
             get_youtube_transcript,
             TavilySearch(
                 max_results=5,
@@ -267,7 +269,7 @@ class BasicAgent:
         if getattr(last_message, "tool_calls", None):
             return "tools"
     
-        return "final_process"
+        return "verify_answer"
     
     def retry_with_feedback(self, state: AgentState):
         notes = state.get("verification_notes", "")
@@ -357,7 +359,12 @@ class BasicAgent:
                     "file_path": file_path,
                     "question": question
                 })
-                return self.format_final_answer(question, raw)
+                result = self.react_graph.invoke({
+                    "messages": [
+                        HumanMessage(content=f"Question:\n{question}\n\nTool result:\n{raw}\n\nReturn only the final answer.")
+                    ]
+                })
+                return result["messages"][-1].content
     
         user_content = question
         if task_id: