Update app.py

app.py

@@ -19,10 +19,11 @@ class MultiAgents:
         print("Initializing MultiAgents.")
         model = os.getenv("DEFAULT_LLM_MODEL")
         
-        print("Router Agent")
+        # print("Router Agent")
         self.orchestrator = RouterAgent()
-        print("Research Agent")
+        # print("Research Agent")
         self.research_agent = ResearchAgent(model)
+        self.audio_agent = AudioAgent(model)
 
         print("Finished initializing agents. Now registering them.")
 
@@ -33,6 +34,12 @@ class MultiAgents:
             "Handles general factual questions requiring web search or knowledge retrieval. Good for questions about people, events, organizations, or facts that require research."
         )
 
+        self.orchestrator.register_agent(
+            "audio_transcript",
+            self.audio_agent,
+            "Handles transcription of an audio file, for example an mp3 file and subsequently finding answer from the audio file."
+        )
+
         print("MultiAgent successfully initialized")
         
     def __call__(self, question: str) -> str:
@@ -107,7 +114,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            if file_name:
+            if not file_name:
                 fetch_url = f"{api_url}/files/{task_id}"
                 print(f"Fetching file from {fetch_url} ")
                 try:
@@ -123,7 +130,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
                     print(f"Error fetching file: {e}")
                     continue
                 
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, file_name)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: