Update app.py

app.py

@@ -16,16 +16,13 @@ from gradio_client import Client
 warnings.filterwarnings('ignore')
 
 # Initialize story starters
-STORY_STARTERS = pd.DataFrame({
-    'category': ['Adventure', 'Mystery', 'Romance', 'Sci-Fi', 'Fantasy'],
-    'starter': [
-        'In a hidden temple deep in the Amazon...',
-        'The detective found an unusual note...',
-        'Two strangers meet on a rainy evening...',
-        'The space station received an unexpected signal...',
-        'A magical portal appeared in the garden...'
-    ]
-})
+STORY_STARTERS = [
+    ['Adventure', 'In a hidden temple deep in the Amazon...'],
+    ['Mystery', 'The detective found an unusual note...'],
+    ['Romance', 'Two strangers meet on a rainy evening...'],
+    ['Sci-Fi', 'The space station received an unexpected signal...'],
+    ['Fantasy', 'A magical portal appeared in the garden...']
+]
 
 # Initialize client outside of interface definition
 arxiv_client = None
@@ -76,14 +73,15 @@ def load_gallery():
             with open(story_file) as f:
                 story_text = f.read()
             
-            files.append({
-                "timestamp": timestamp,
-                "story_path": str(story_file),
-                "audio_path": str(audio_file) if audio_file.exists() else None,
-                "story_preview": story_text[:100] + "..."
-            })
+            # Format as list instead of dict for Gradio Dataframe
+            files.append([
+                timestamp,
+                story_text[:100] + "...",
+                str(story_file),
+                str(audio_file) if audio_file.exists() else None
+            ])
         
-        return sorted(files, key=lambda x: x["timestamp"], reverse=True)
+        return sorted(files, key=lambda x: x[0], reverse=True)
     except Exception as e:
         print(f"Error loading gallery: {str(e)}")
         return []
@@ -135,6 +133,18 @@ def process_story_and_audio(prompt, model_choice):
     except Exception as e:
         return f"Error: {str(e)}", None, None
 
+def play_gallery_audio(evt: gr.SelectData, gallery_data):
+    """Play audio from gallery selection"""
+    try:
+        selected_row = gallery_data[evt.index[0]]
+        audio_path = selected_row[3]  # Audio path is the fourth element
+        if audio_path and os.path.exists(audio_path):
+            return audio_path
+        return None
+    except Exception as e:
+        print(f"Error playing gallery audio: {str(e)}")
+        return None
+
 # Create the Gradio interface
 with gr.Blocks(title="AI Story Generator") as demo:
     gr.Markdown("""
@@ -180,20 +190,20 @@ with gr.Blocks(title="AI Story Generator") as demo:
             gr.Markdown("### 📚 Story Starters")
             story_starters = gr.Dataframe(
                 value=STORY_STARTERS,
-                headers=["category", "starter"],
+                headers=["Category", "Starter"],
                 interactive=False
             )
             
             gr.Markdown("### 🎬 Gallery")
             gallery = gr.Dataframe(
                 value=load_gallery(),
-                headers=["timestamp", "story_preview"],
+                headers=["Timestamp", "Preview", "Story Path", "Audio Path"],
                 interactive=False
             )
     
     # Event handlers
     def update_prompt(evt: gr.SelectData):
-        return STORY_STARTERS.iloc[evt.index[0]]["starter"]
+        return STORY_STARTERS[evt.index[0]][1]
     
     story_starters.select(update_prompt, None, prompt_input)
     
@@ -202,6 +212,12 @@ with gr.Blocks(title="AI Story Generator") as demo:
         inputs=[prompt_input, model_choice],
         outputs=[story_output, audio_output, gallery]
     )
+    
+    gallery.select(
+        fn=play_gallery_audio,
+        inputs=[gallery],
+        outputs=[audio_output]
+    )
 
 if __name__ == "__main__":
     demo.launch()