Update Gradio_UI.py

Gradio_UI.py

@@ -299,13 +299,18 @@ class GradioUI:
 
         # NEW: Add a helper function to extract audio file path from stored messages
         def extract_audio(messages):
+            # Iterate backwards over messages until we find one with audio content.
             if not messages:
                 return None
-            last_message = messages[-1]
-            if isinstance(last_message.content, dict):
-                mime = last_message.content.get("mime_type", "")
-                if mime.startswith("audio"):
-                    return last_message.content.get("path")
+            for msg in reversed(messages):
+                # Only process messages that are objects with a "content" attribute.
+                if not hasattr(msg, "content"):
+                    continue
+                # If the content is a dict and contains an audio mime type, return the file path.
+                if isinstance(msg.content, dict):
+                    mime = msg.content.get("mime_type", "")
+                    if mime.startswith("audio"):
+                        return msg.content.get("path")
             return None
 
         with gr.Blocks(fill_height=True) as demo: