Feature: Extract and return audio file from agent output

app.py

@@ -7,7 +7,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import create_agent
 import tempfile
 import shutil
-
+import re
 # Get API keys from Hugging Face Secrets
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY")
@@ -107,7 +107,30 @@ def gradio_handler(age, gender, topic, pdf_file):
             output = loop.run_until_complete(run_agent_dynamic(age, gender, topic, pdf_path))
         finally:
             loop.close()
-        return f"✅ Success!\n\n{str(output)}", None
+        
+        # Extract audio file path from the output
+        output_str = str(output)
+        audio_path = None
+        
+        # Look for the audio file mention in the output
+        if "File saved as:" in output_str:
+            # Extract filename from the output
+            import re
+            match = re.search(r'File saved as: ([\w,_\-\.]+\.mp3)', output_str)
+            if match:
+                filename = match.group(1)
+                # The file is saved in current directory, which is /app
+                audio_path = os.path.join("/app", filename)
+                
+                # Check if file exists
+                if not os.path.exists(audio_path):
+                    # Try current working directory
+                    audio_path = filename
+                    if not os.path.exists(audio_path):
+                        audio_path = None
+        
+        return f"✅ Success!\n\n{output_str}", audio_path
+        
     except Exception as e:
         return f"❌ Error: {str(e)}\n\nCheck logs for details.", None
     finally: