Update app.py

app.py

@@ -2,11 +2,9 @@ import gradio as gr
 import moviepy.editor as mp
 import numpy as np
 from PIL import Image
-import tempfile
 import os
 
-# Resize image while maintaining aspect ratio
-def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
+def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
     width, height = img.size
     target_width, target_height = target_size
 
@@ -21,87 +19,70 @@ def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(
         new_width = int(new_height * aspect_ratio)
 
     img_resized = img.resize((new_width, new_height))
+    
+    if new_width > target_width:
+        left = (new_width - target_width) // 2
+        img_resized = img_resized.crop((left, 0, left + target_width, target_height))
+    
     final_img = Image.new('RGB', target_size, padding_color)
-    padding_left = (target_width - new_width) // 2
-    padding_top = (target_height - new_height) // 2
-    final_img.paste(img_resized, (padding_left, padding_top))
+    final_img.paste(img_resized, (0, 0))
 
     return final_img
 
-# Video generation function with transition and debug logging
-def process_and_generate_video(audio_file, images):
-    debug_log = []
+def apply_zoom_effect(image_clip):
+    zoomed_clip = image_clip.resize(lambda t: 1 + 0.05 * t)  # Zoom in gradually
+    return zoomed_clip
 
+def process_and_generate_video(audio_file, images):
     try:
-        # Log the files received
-        debug_log.append(f"Audio file received: {audio_file}")
-        debug_log.append(f"Images received: {[img for img in images]}")
-
-        # Process audio file (use audio_file directly as it's a path)
-        audio = mp.AudioFileClip(audio_file)  # Use file path directly
+        audio = mp.AudioFileClip(audio_file)
         audio_duration = audio.duration
         image_clips = []
         image_count = len(images)
         image_duration = audio_duration / image_count
-
-        debug_log.append(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
-
-        # Process each image and create video clip
-        for img in images:
-            debug_log.append(f"Processing image: {img}")  # Debug print
-            img = Image.open(img)  # Open image from file 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)
-
+        
+        print(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
+        
+        for img_path in images:
+            img = Image.open(img_path)
+            img = resize_and_fit_image(img, target_size=(1280, 720))
+            
+            img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(30)
+            img_clip = apply_zoom_effect(img_clip)
+            
             # 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
-
+            if len(image_clips) > 0:
+                img_clip = img_clip.crossfadein(1)
+            
             image_clips.append(img_clip)
-
-        debug_log.append(f"Created {len(image_clips)} image clips.")
-
-        # Concatenate image clips with transitions
+        
+        print(f"Image clips: {len(image_clips)} clips created.")
+        
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
-
-        # Set output file path in a temporary location
-        output_path = '/tmp/generated_video.mp4'  # Temporary path for output
-
-        debug_log.append(f"Writing video to {output_path}...")
-
-        # Write video to file
-        video.write_videofile(output_path, codec='libx264', audio_codec='aac')
-
-        # Check if the video file exists
-        if os.path.exists(output_path):
-            debug_log.append(f"Video generated successfully at {output_path}")
-            return output_path, "\n".join(debug_log)  # Return the video path and debug log
-        else:
-            debug_log.append(f"Error: Video not generated at {output_path}")
-            return "Error: Video not generated.", "\n".join(debug_log)
-
+        
+        output_path = '/content/generated_video.mp4'
+        video.write_videofile(output_path, codec='libx264', audio_codec='aac', threads=4, fps=30, preset='ultrafast')
+        
+        return output_path  # Return the file path for Gradio output
+    
     except Exception as e:
-        debug_log.append(f"Error during video generation: {str(e)}")
-        return f"Error generating video: {str(e)}", "\n".join(debug_log)
+        print(f"Error during video generation: {str(e)}")
+        return f"Error generating video: {str(e)}"
 
-# Gradio interface setup
 def gradio_interface():
     with gr.Blocks() as demo:
         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
-                generate_button = gr.Button("Generate Video")  # Button to generate video
-
-            output_video = gr.Video(label="Generated Video")  # Video output display
-            output_logs = gr.Textbox(label="Debug Logs", interactive=False)  # Display debug logs
-
-        generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=[output_video, output_logs])
-
-    demo.launch()  # Launch the Gradio interface
+                mp3_input = gr.Audio(type="filepath", label="Upload MP3")
+                image_input = gr.File(type="filepath", file_types=[".jpg", ".png"], label="Upload Images", file_count="multiple")
+                generate_button = gr.Button("Generate Video")
+                
+            output_video = gr.Video(label="Generated Video")
+        
+        generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=output_video)
+    
+    demo.launch()
 
 # Run the interface
 gradio_interface()