Update app.py

app.py

@@ -2,11 +2,14 @@ import gradio as gr
 import moviepy.editor as mp
 import numpy as np
 from PIL import Image
+import tempfile
+import shutil
 
-# Resize image with aspect ratio
+# Resize image while maintaining aspect ratio
 def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
     width, height = img.size
     target_width, target_height = target_size
+
     aspect_ratio = width / height
     target_aspect_ratio = target_width / target_height
 
@@ -27,34 +30,55 @@ def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(
 
 # Video generation function with transition
 def process_and_generate_video(audio_file, images):
-    audio = mp.AudioFileClip(audio_file)
-    audio_duration = audio.duration
-    image_clips = []
-    image_count = len(images)
-    image_duration = audio_duration / image_count
-
-    # Process each image and create a video clip
-    for img_path in images:
-        img = Image.open(img_path)
-        img = resize_image_with_aspect_ratio(img, target_size=(1280, 720))
-
-        img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(24)
-
-        # Add transition effect (crossfade)
-        if len(image_clips) > 0:
-            img_clip = img_clip.crossfadein(1)  # 1-second crossfade transition
-
-        image_clips.append(img_clip)
-
-    # Combine the image clips into a single video
-    video = mp.concatenate_videoclips(image_clips, method="compose")
-    video = video.set_audio(audio)
-
-    # Output path for the generated video
-    output_path = '/content/generated_video.mp4'
-    video.write_videofile(output_path, codec='libx264', audio_codec='aac')
-
-    return output_path
+    try:
+        # Create temporary folder to store image paths
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Save images to temporary directory
+            image_paths = []
+            for img in images:
+                temp_image_path = os.path.join(temp_dir, os.path.basename(img.name))
+                with open(temp_image_path, 'wb') as f:
+                    f.write(img.read())
+                image_paths.append(temp_image_path)
+
+            # Process audio file
+            audio = mp.AudioFileClip(audio_file.name)  # Use .name for filepath
+            audio_duration = audio.duration
+            image_clips = []
+            image_count = len(image_paths)
+            image_duration = audio_duration / image_count
+            
+            print(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
+            
+            # Process each image and create video clip
+            for img_path in image_paths:
+                img = Image.open(img_path)
+                img = resize_image_with_aspect_ratio(img, target_size=(1280, 720))
+                
+                # Create image clip with a crossfade transition effect
+                img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(24)
+                
+                # Add transition effect - Crossfade In
+                if len(image_clips) > 0:  # Apply transition only after the first image
+                    img_clip = img_clip.crossfadein(1)  # 1-second fade-in transition
+                
+                image_clips.append(img_clip)
+            
+            # Concatenate image clips with transitions
+            video = mp.concatenate_videoclips(image_clips, method="compose")
+            video = video.set_audio(audio)
+            
+            # Set output file path
+            output_path = '/tmp/generated_video.mp4'  # Temporary path for output
+            
+            # Write video to file
+            video.write_videofile(output_path, codec='libx264', audio_codec='aac')
+            
+            return output_path  # Return the file path for Gradio output
+        
+    except Exception as e:
+        print(f"Error during video generation: {str(e)}")
+        return f"Error generating video: {str(e)}"
 
 # Gradio interface setup
 def gradio_interface():
@@ -62,14 +86,14 @@ def gradio_interface():
         with gr.Row():
             with gr.Column():
                 mp3_input = gr.Audio(type="filepath", label="Upload MP3")  # MP3 input
-                image_input = gr.File(type="filepath", file_types=[".jpg", ".png"], label="Upload Images", file_count="multiple")  # Images input
+                image_input = gr.File(type="file", file_types=[".jpg", ".png"], label="Upload Images", file_count="multiple")  # Images input
                 generate_button = gr.Button("Generate Video")  # Button to generate video
-
+                
             output_video = gr.Video(label="Generated Video")  # Video output display
 
         generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=output_video)
 
     demo.launch()  # Launch the Gradio interface
 
-# Run the Gradio app
+# Run the interface
 gradio_interface()