Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import json
 from transformers import pipeline
-from moviepy.editor import ImageClip, concatenate_videoclips
 from PIL import Image
 import numpy as np
 import os
@@ -34,7 +33,7 @@ def generate_cartoon(script_text):
     """
     Return result in JSON format.
     """
-    
+
     response = text_generator(prompt, max_new_tokens=1024)[0]['generated_text']
 
     try:
@@ -42,13 +41,11 @@ def generate_cartoon(script_text):
     except:
         return "Script expansion failed. Please refine input."
 
-    characters = set()
-    backgrounds = set()
-    scene_clips = []
-
     os.makedirs("generated_images", exist_ok=True)
     os.makedirs("generated_audio", exist_ok=True)
 
+    scene_results = []
+
     for idx, scene in enumerate(scene_data.get("scenes", [])):
         # Generate Background Image
         background_prompt = scene.get("background_description", "cartoon background")
@@ -62,25 +59,17 @@ def generate_cartoon(script_text):
         audio_path = f"generated_audio/scene_{idx+1}.wav"
         audio_output['audio'].export(audio_path, format="wav")
 
-        # Create Scene Video Clip
-        image_clip = ImageClip(bg_path).set_duration(5).fadein(1).fadeout(1)
-        image_clip = image_clip.set_audio(audio_path)
-        scene_clips.append(image_clip)
-
-    # Merge Scenes
-    final_video = concatenate_videoclips(scene_clips)
-    final_video_path = "final_cartoon_video.mp4"
-    final_video.write_videofile(final_video_path, fps=24)
+        scene_results.append((bg_path, audio_path))
 
-    return final_video_path
+    return scene_results
 
 # Gradio UI
 demo = gr.Interface(
     fn=generate_cartoon,
     inputs=gr.Textbox(label="Enter Cartoon Script", lines=10),
-    outputs=gr.Video(label="Generated Cartoon Video"),
-    title="Cartoon Video Generator",
-    description="Enter a short cartoon story script and generate a cartoon video (7-10 mins) automatically."
+    outputs=gr.Gallery(label="Generated Scenes with Audio"),
+    title="Cartoon Scene Generator",
+    description="Enter a cartoon story script to generate scenes with background images and audio. Video merging should be done separately in Google Colab."
 )
 
 if __name__ == "__main__":