Update app.py

app.py

@@ -417,50 +417,51 @@ def combine_video_with_audio(video_path, audio_path, output_path):
 
 # Main processing function
 def create_story_video(prompt, progress=gr.Progress()):
-    # Input validation
     if not prompt or len(prompt.strip()) < 5:
         return "Please enter a longer prompt (at least 5 characters).", None, None
 
     try:
-        # Step 1: Generate story
+        print("Step 1: Generating story...")
         progress(0, desc="Starting story generation...")
         story = generate_story(prompt)
+        print("Story generation complete.")
         progress(20, desc="Story generated successfully!")
 
-        # Step 2: Generate video
+        print("Step 2: Generating video...")
         progress(25, desc="Creating video animation (this may take several minutes)...")
         video_path = generate_video(story)
+        print("Video generation complete.")
         progress(60, desc="Video created successfully!")
 
-        # Step 3: Create audio summary
+        print("Step 3: Summarizing for audio...")
         progress(65, desc="Creating audio summary...")
         audio_summary = summary_of_summary(story, video_path)
+        print("Audio summary complete.")
         progress(80, desc="Creating audio narration...")
 
-        # Step 4: Generate audio with sentiment (async)
+        print("Step 4: Generating audio...")
         try:
-            # Set up event loop handling
             try:
                 loop = asyncio.get_event_loop()
             except RuntimeError:
                 loop = asyncio.new_event_loop()
                 asyncio.set_event_loop(loop)
-
             audio_file = loop.run_until_complete(
                 generate_audio_with_sentiment(audio_summary, sentiment_analyzer)
             )
+            print(f"Audio generated at: {audio_file}")
             progress(90, desc="Audio created successfully!")
         except Exception as e:
             print(f"Audio generation error: {str(e)}")
             return story, None, f"Audio generation failed: {str(e)}"
 
-        # Step 5: Combine video and audio
+        print("Step 5: Combining video and audio...")
         progress(95, desc="Combining video and audio...")
         output_path = 'final_video_with_audio.mp4'
         combine_video_with_audio(video_path, audio_file, output_path)
-
+        print("Combination complete.")
         progress(100, desc="Process complete!")
-        return story, output_path, audio_summary
+        return story, output_path, audio_file  # Return audio file path instead of summary
 
     except Exception as e:
         error_msg = f"Error: {str(e)}\n{traceback.format_exc()}"
@@ -481,7 +482,6 @@ with gr.Blocks(title="AI Story Video Generator", theme=gr.themes.Soft()) as demo
     gr.Markdown("# 🎬 AI Story Video Generator")
     gr.Markdown("Enter a one-sentence prompt to generate a complete story with video and narration.")
 
-    # Input section
     with gr.Row():
         prompt_input = gr.Textbox(
             label="Your Story Idea",
@@ -489,10 +489,7 @@ with gr.Blocks(title="AI Story Video Generator", theme=gr.themes.Soft()) as demo
             lines=2
         )
 
-    # Example prompts section
     gr.Markdown("### Try these example prompts:")
-
-    # Create examples using Gradio's examples feature
     with gr.Row():
         examples = gr.Examples(
             examples=[[prompt] for prompt in EXAMPLE_PROMPTS],
@@ -504,10 +501,8 @@ with gr.Blocks(title="AI Story Video Generator", theme=gr.themes.Soft()) as demo
         generate_button = gr.Button("Generate Story Video", variant="primary")
         clear_button = gr.Button("Clear", variant="secondary")
 
-    # Status indicator
     status_indicator = gr.Markdown("Ready to generate your story video...")
 
-    # Output section with tabs
     with gr.Tabs():
         with gr.TabItem("Results"):
             with gr.Row():
@@ -515,61 +510,36 @@ with gr.Blocks(title="AI Story Video Generator", theme=gr.themes.Soft()) as demo
                     video_output = gr.Video(label="Generated Video with Narration")
                 with gr.Column(scale=1):
                     story_output = gr.TextArea(label="Generated Story", lines=15, max_lines=30)
-                    summary_output = gr.TextArea(label="Audio Summary", lines=5)
+                    audio_output = gr.Audio(label="Audio Narration")  # Changed to Audio
 
         with gr.TabItem("Help & Information"):
             gr.Markdown("""
             ## How to use this tool
-
             1. Enter a creative one-sentence story idea in the input box
             2. Click "Generate Story Video" and wait for processing to complete
-            3. View your complete AI-generated story video with narration
-
+            3. View your story, narration audio, and final video
             ## Processing Steps
-
-            1. **Story Generation**: The AI expands your idea into a 15-20 sentence story
-            2. **Video Creation**: Each sentence is visualized through AI-generated animation
-            3. **Audio Narration**: The AI analyzes the sentiment and creates appropriate voiceover
-            4. **Final Compilation**: Video and audio are combined into your final story
-
-            ## Tips for Great Results
-
-            - Use clear, specific prompts that suggest a narrative arc
-            - Include interesting characters, settings, or situations
-            - Make your prompt realistic but with potential for development
-            - Try to suggest a potential conflict or discovery
-            
-            ## Note on Processing Time
-            
-            For faster testing, the app currently processes only the first 5 sentences of the story.
-            In a production environment, this limit would be removed.
-
-            ## Troubleshooting
-
-            If you encounter errors:
-            - Try a different prompt
-            - Ensure your prompt is clear and specific
-            - Check that all required models are properly loaded
+            - Story Generation: Expands your idea into a 15-20 sentence story
+            - Video Creation: Visualizes sentences with AI animation
+            - Audio Narration: Creates a voiceover with sentiment analysis
+            - Final Compilation: Combines video and audio
             """)
 
-    # Handle clearing
     def clear_outputs():
-        return "", None, ""
+        return "", None, None
 
-    # Connect interface elements
     generate_button.click(
         fn=create_story_video,
         inputs=prompt_input,
-        outputs=[story_output, video_output, summary_output],
+        outputs=[story_output, video_output, audio_output],  # Updated to audio_output
         api_name="generate"
     )
 
     clear_button.click(
         fn=clear_outputs,
         inputs=None,
-        outputs=[story_output, video_output, summary_output]
+        outputs=[story_output, video_output, audio_output]
     )
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch()