Update Gradio_UI.py

Gradio_UI.py

@@ -194,16 +194,21 @@ def stream_to_gradio(
             content={"path": final_answer.to_string(), "mime_type": "image/png"},
         )
     elif isinstance(final_answer, AgentAudio):
-        # Assuming your AgentAudio object stores the raw MP3 bytes in an attribute called "value"
-        audio_bytes = final_answer.value  
-        # Convert MP3 bytes to a numpy array using our helper function
-        audio_np = mp3_bytes_to_numpy(audio_bytes)
-        print("DEBUG audio_np type:", type(audio_np), "shape:", audio_np.shape)
+        # 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": final_answer.to_string(), "mime_type": "audio/mpeg"},
+            content={"path": public_path, "mime_type": "audio/mpeg"},
         )
-        print("DEBUG AgentAudio attributes:", vars(final_answer))
 
     else:
         yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
@@ -353,7 +358,7 @@ class GradioUI:
             )
 
             # NEW: Add a dedicated audio player component below the chatbot.
-            audio_player = gr.Audio(label="Audio Pronunciation", type="numpy")
+            audio_player = gr.Audio(label="Audio Pronunciation", type="filepath")
     
             if self.file_upload_folder is not None:
                 upload_file = gr.File(label="Upload a file")