Update app.py

app.py

@@ -980,9 +980,8 @@ voice_conversation_state = {
 #         return f"❌ Error: {str(e)}"
 
 def generate_podcast_ui(doc_ids, style, duration, voice1, voice2):
-    """UI wrapper for podcast generation"""
+    """UI wrapper for podcast generation (EXISTING FUNCTION - keep as is)"""
     try:
-        # Add detailed logging
         logger.info(f"generate_podcast_ui called with:")
         logger.info(f"  doc_ids: {doc_ids} (type: {type(doc_ids)})")
         logger.info(f"  style: {style}")
@@ -990,22 +989,18 @@ def generate_podcast_ui(doc_ids, style, duration, voice1, voice2):
         logger.info(f"  voice1: {voice1}")
         logger.info(f"  voice2: {voice2}")
         
-        # Handle various input formats
         if doc_ids is None:
             logger.warning("doc_ids is None")
             return ("⚠️ Please select at least one document", None, "No documents selected", "")
         
-        # Convert to list if needed
         if isinstance(doc_ids, str):
             logger.info(f"Converting string doc_id to list: {doc_ids}")
             doc_ids = [doc_ids]
         
-        # Check if empty
         if not doc_ids or len(doc_ids) == 0:
             logger.warning(f"doc_ids is empty or has length 0: {doc_ids}")
             return ("⚠️ Please select at least one document", None, "No documents selected", "")
         
-        # Filter out None or empty string values
         doc_ids = [doc_id for doc_id in doc_ids if doc_id and doc_id.strip()]
         
         if not doc_ids:
@@ -1094,6 +1089,105 @@ def load_dashboard_stats():
         logger.error(f"Error loading dashboard stats: {str(e)}")
         return (0, 0, 0.0, [], "❌ Error", "❌ Error", "❌ Error")
 
+# ADD THESE HELPER FUNCTIONS BEFORE create_gradio_interface():
+def load_podcast_library_ui():
+    """Load and display podcast library with audio players"""
+    try:
+        result = mcp_server.list_podcasts_sync(limit=50)
+        
+        if not result.get("success"):
+            return (
+                gr.Column(visible=False),
+                f"❌ Failed to load podcasts: {result.get('error', 'Unknown error')}"
+            )
+        
+        podcasts = result.get("podcasts", [])
+        
+        if not podcasts:
+            return (
+                gr.Column(visible=False),
+                "📭 No podcasts generated yet. Create your first podcast above!"
+            )
+        
+        # Build the library UI dynamically
+        with gr.Column(visible=True) as library_col:
+            for idx, podcast in enumerate(podcasts, 1):
+                with gr.Group():
+                    with gr.Row():
+                        # Column 1: Podcast Info (40%)
+                        with gr.Column(scale=2):
+                            gr.Markdown(f"### 🎙️ Podcast #{idx}")
+                            
+                            # Extract document names from metadata
+                            doc_names = []
+                            doc_ids = []  # Initialize here to avoid reference error
+                            
+                            if podcast.get("metadata"):
+                                doc_ids = podcast["metadata"].get("document_ids", [])
+                                # Get document names
+                                for doc_id in doc_ids:
+                                    try:
+                                        doc = mcp_server.run_async(
+                                            mcp_server.document_store.get_document(doc_id)
+                                        )
+                                        if doc:
+                                            doc_names.append(doc.filename)
+                                    except Exception as e:
+                                        logger.warning(f"Could not fetch document {doc_id}: {e}")
+                                        doc_names.append(f"Doc {doc_id[:8]}...")
+                            
+                            # Display document names
+                            if doc_names:
+                                gr.Markdown(f"**📄 Documents:** {', '.join(doc_names)}")
+                            else:
+                                doc_count = len(doc_ids) if doc_ids else 'N/A'
+                                gr.Markdown(f"**📄 Documents:** {doc_count} document(s)")
+                            
+                            # Podcast metadata
+                            style = podcast.get("metadata", {}).get("style", "Unknown")
+                            duration = podcast.get("metadata", {}).get("duration_minutes", "N/A")
+                            created = podcast.get("created_at", "Unknown")[:19] if podcast.get("created_at") else "Unknown"
+                            
+                            gr.Markdown(
+                                f"**🎨 Style:** {style.title()}  \n"
+                                f"**⏱️ Duration:** ~{duration} min  \n"
+                                f"**📅 Created:** {created}  \n"
+                                f"**🆔 ID:** `{podcast['id'][:16]}...`"
+                            )
+                        
+                        # Column 2: Audio Player (60%)
+                        with gr.Column(scale=3):
+                            audio_file = podcast.get("audio_file")
+                            
+                            if audio_file and os.path.exists(audio_file):
+                                gr.Audio(
+                                    value=audio_file,
+                                    type="filepath",
+                                    interactive=False,
+                                    show_label=False,
+                                    show_download_button=True,
+                                    waveform_options={"show_controls": True}
+                                )
+                            else:
+                                gr.Markdown("⚠️ *Audio file not found*")
+                                if audio_file:
+                                    gr.Markdown(f"*Expected path: {audio_file}*")
+                    
+                    # Optional: Show transcript in accordion
+                    with gr.Accordion(f"📝 View Transcript", open=False):
+                        transcript = podcast.get("transcript", "Transcript not available")
+                        gr.Markdown(transcript)
+        
+        status_msg = f"✅ Loaded {len(podcasts)} podcast{'s' if len(podcasts) != 1 else ''}"
+        return library_col, status_msg
+        
+    except Exception as e:
+        logger.error(f"Error loading podcast library: {str(e)}", exc_info=True)
+        return (
+            gr.Column(visible=False),
+            f"❌ Error loading library: {str(e)}"
+        )
+
 def create_gradio_interface():
     custom_theme = gr.themes.Soft(
         primary_hue=gr.themes.colors.indigo,
@@ -1530,40 +1624,34 @@ def create_gradio_interface():
                 gr.Markdown("""
                     # 🎙️ AI Podcast Studio
                     ## Transform Documents into Engaging Audio
-
                     Convert your documents into professional podcast conversations with AI-generated voices.
-
                     ### How It Works:
                     1. **Select Documents** - Choose 1-5 documents from your library
                     2. **Choose Style** - Pick conversation style (casual, educational, etc.)
                     3. **Set Duration** - Select podcast length (5-30 minutes)
                     4. **Select Voices** - Pick two AI hosts from available voices
                     5. **Generate** - AI creates natural dialogue discussing your content
-
                     ### Powered By:
                     - 🎵 **ElevenLabs AI** - Ultra-realistic voice synthesis
                     - 🤖 **LLM** - Intelligent content analysis and script generation
                     - 📚 **RAG** - Context-aware information retrieval
-
                     ---
                     """)
-                
+    
                 with gr.Row():
                     with gr.Column(scale=2):
                         # Configuration Panel
                         with gr.Group():
                             gr.Markdown("#### 📚 Select Content")
                             
-                            # IMPORTANT: CheckboxGroup that stores document IDs
                             podcast_doc_selector = gr.CheckboxGroup(
-                                choices=get_document_choices(),  # List of (label, value) tuples
+                                choices=get_document_choices(),
                                 label="Documents to Include",
                                 info="Choose 1-5 documents for best results",
                                 interactive=True,
-                                value=[]  # Start with empty selection
+                                value=[]
                             )
                             
-                            # Debug display (optional - remove in production)
                             gr.Markdown("*Selected document IDs will be used for podcast generation*")
                         
                         with gr.Accordion("🎨 Podcast Settings", open=True):
@@ -1616,26 +1704,25 @@ def create_gradio_interface():
                         )
                     
                     with gr.Column(scale=3):
-                        # Output Panel
+                        # Output Panel - Latest Generated
                         with gr.Group():
-                            gr.Markdown("#### 🎵 Generated Podcast")
+                            gr.Markdown("#### 🎵 Latest Generated Podcast")
                             
                             podcast_audio_player = gr.Audio(
-                                label="podcast_audio_player",
                                 type="filepath",
                                 interactive=False,
                                 autoplay=True,
-                                container=False
                                 show_label=False
                             )
-                            gr.DownloadButton("💾 Download MP3", value=podcast_audio_player)
                         
                         with gr.Accordion("📝 Transcript", open=False):
                             podcast_transcript_display = gr.Markdown(
                                 value="*Transcript will appear after generation...*"
                             )
                 
-                # Event handler - Connect the button to the function
+                podcast_library_container = gr.Column()
+
+                # Event handler for generate button
                 generate_podcast_btn.click(
                     fn=generate_podcast_ui,
                     inputs=[
@@ -1651,6 +1738,30 @@ def create_gradio_interface():
                         podcast_transcript_display,
                         podcast_id_display
                     ]
+                ).then(
+                    # Auto-refresh library after generation
+                    fn=load_podcast_library_ui,
+                    outputs=[podcast_library_container, podcast_library_status]
+                )
+                
+                # NEW SECTION: Podcast Library (at tab level, same as first gr.Row())
+                gr.Markdown("---")
+                gr.Markdown("### 📚 Podcast Library")
+                gr.Markdown("Browse and play all your generated podcasts")
+                
+                with gr.Row():
+                    refresh_podcast_library_btn = gr.Button("🔄 Refresh Library", variant="secondary")
+                    podcast_library_status = gr.Textbox(
+                        label="Library Status",
+                        value="Click 'Refresh Library' to load podcasts",
+                        interactive=False,
+                        scale=3
+                    )
+                                
+                # Event handler for refresh button
+                refresh_podcast_library_btn.click(
+                    fn=load_podcast_library_ui,
+                    outputs=[podcast_library_container, podcast_library_status]
                 )
                 
             with gr.Tab("❓ Ask Questions"):