Update app.py

app.py

@@ -4,28 +4,23 @@ import time
 from datetime import datetime
 import tempfile
 import os
-from moviepy.editor import ImageClip, concatenate_videoclips
-from gradio_client import Client
-from PIL import Image
 import edge_tts
 import asyncio
 import warnings
-import numpy as np
+from gradio_client import Client
 
 warnings.filterwarnings('ignore')
 
-# Initialize Gradio clients with public demo spaces
+# Initialize Gradio client
 def initialize_clients():
     try:
-        # Use a simpler public demo space
-        image_client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
-        return image_client
+        client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
+        return client
     except Exception as e:
-        print(f"Error initializing clients: {str(e)}")
+        print(f"Error initializing client: {str(e)}")
         return None
 
 if gr.NO_RELOAD:
-    # Initialize client in NO_RELOAD block to prevent multiple initializations
     CLIENT = initialize_clients()
 
 STORY_GENRES = [
@@ -40,10 +35,10 @@ STORY_GENRES = [
 ]
 
 STORY_STRUCTURES = {
-    "Three Act": "Setup (Introduction, Inciting Incident) -> Confrontation (Rising Action, Climax) -> Resolution (Falling Action, Conclusion)",
-    "Hero's Journey": "Ordinary World -> Call to Adventure -> Trials -> Transformation -> Return",
+    "Three Act": "Setup -> Confrontation -> Resolution",
+    "Hero's Journey": "Ordinary World -> Call to Adventure -> Trials -> Return",
     "Five Act": "Exposition -> Rising Action -> Climax -> Falling Action -> Resolution",
-    "Seven Point": "Hook -> Plot Turn 1 -> Pinch Point 1 -> Midpoint -> Pinch Point 2 -> Plot Turn 2 -> Resolution"
+    "Seven Point": "Hook -> Plot Turn 1 -> Pinch Point 1 -> Midpoint -> Plot Turn 2 -> Resolution"
 }
 
 async def generate_speech(text, voice="en-US-AriaNeural"):
@@ -85,22 +80,6 @@ def generate_story(prompt, model_choice):
     except Exception as e:
         return f"Error generating story: {str(e)}"
 
-def process_story(story_text, num_scenes=5):
-    """Break story into scenes for visualization"""
-    if not story_text:
-        return []
-        
-    sentences = story_text.split('.')
-    scenes = []
-    scene_length = max(1, len(sentences) // num_scenes)
-    
-    for i in range(0, len(sentences), scene_length):
-        scene = '. '.join(sentences[i:i+scene_length]).strip()
-        if scene:
-            scenes.append(scene)
-    
-    return scenes[:num_scenes]
-
 def story_generator_interface(prompt, genre, structure, model_choice, num_scenes, words_per_scene):
     """Main story generation and multimedia creation function"""
     try:
@@ -110,20 +89,23 @@ def story_generator_interface(prompt, genre, structure, model_choice, num_scenes
         # Generate story
         story = generate_story(story_prompt, model_choice)
         if story.startswith("Error"):
-            return story, None, None, None
+            return story, None
         
         # Generate speech
         audio_path = asyncio.run(generate_speech(story))
         
-        return story, None, audio_path, None
+        return story, audio_path
         
     except Exception as e:
         error_msg = f"An error occurred: {str(e)}"
-        return error_msg, None, None, None
+        return error_msg, None
 
 # Create Gradio interface
 with gr.Blocks(title="AI Story Generator") as demo:
-    gr.Markdown("# 🎭 AI Story Generator")
+    gr.Markdown("""
+    # 🎭 AI Story Generator
+    Generate creative stories with AI and listen to them!
+    """)
     
     with gr.Row():
         with gr.Column():
@@ -168,17 +150,31 @@ with gr.Blocks(title="AI Story Generator") as demo:
             story_output = gr.Textbox(
                 label="Generated Story",
                 lines=10,
-                interactive=False  # Changed from readonly to interactive=False
+                interactive=False
             )
     
     with gr.Row():
-        audio_output = gr.Audio(label="Story Narration")
+        audio_output = gr.Audio(
+            label="Story Narration",
+            type="filepath"
+        )
     
+    # Fixed the click event outputs
     generate_btn.click(
         fn=story_generator_interface,
-        inputs=[prompt_input, genre_input, structure_input, model_choice, num_scenes, words_per_scene],
-        outputs=[story_output, None, audio_output, None]  # Set image and video outputs to None for now
+        inputs=[
+            prompt_input,
+            genre_input,
+            structure_input,
+            model_choice,
+            num_scenes,
+            words_per_scene
+        ],
+        outputs=[
+            story_output,
+            audio_output
+        ]
     )
 
 if __name__ == "__main__":
-    demo.launch(reload=True)
+    demo.launch(reload=True)