Update Gradio_UI.py

Gradio_UI.py

@@ -194,22 +194,16 @@ def stream_to_gradio(
             content={"path": final_answer.to_string(), "mime_type": "image/png"},
         )
     elif isinstance(final_answer, AgentAudio):
-        # Get the file path from the AgentAudio object (assuming it returns a string path)
-        local_path = final_answer.to_string()  # e.g. "/tmp/response.mp3"
-        
-        # Define a public folder to serve audio files
-        public_folder = "public_audio"
-        os.makedirs(public_folder, exist_ok=True)
-        
-        # Copy the file from /tmp to the public folder
-        public_path = os.path.join(public_folder, os.path.basename(local_path))
-        shutil.copy(local_path, public_path)
-        
         yield gr.ChatMessage(
             role="assistant",
-            content={"path": public_path, "mime_type": "audio/mpeg"},
+            content={"path": final_answer.to_string(), "mime_type": "audio/mpeg"},
+        )
+    elif isinstance(final_answer, str) and final_answer.strip().startswith("/tmp/"):
+        # Fallback: if the final answer is a string that is a file path, treat it as audio.
+        yield gr.ChatMessage(
+            role="assistant",
+            content={"path": final_answer.strip(), "mime_type": "audio/mpeg"},
         )
-
     else:
         yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")