Update app.py

app.py

@@ -3,12 +3,10 @@ import moviepy.editor as mp
 import numpy as np
 from PIL import Image
 import os
+import sys
 
-# Utility function to log messages
-def log_message(message):
-    with open("/tmp/video_generation_log.txt", "a") as log_file:
-        log_file.write(message + "\n")
-    print(message)
+# Redirect stdout to ensure logs are visible
+sys.stdout = sys.stderr
 
 # Resize image while maintaining aspect ratio
 def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
@@ -26,13 +24,8 @@ def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
         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)
-    final_img.paste(img_resized, (0, 0))
+    final_img.paste(img_resized, ((target_width - new_width) // 2, (target_height - new_height) // 2))
 
     return final_img
 
@@ -41,58 +34,55 @@ def apply_zoom_effect(image_clip):
     zoomed_clip = image_clip.resize(lambda t: 1 + 0.05 * t)  # Zoom in gradually
     return zoomed_clip
 
+# Video generation function
 def process_and_generate_video(audio_file, images):
     try:
-        log_message("Starting video generation...")
+        print("Starting video generation...")
 
-        # Check if the file paths are strings, and convert to actual file-like objects
-        if isinstance(audio_file, str):
-            log_message(f"Audio file path: {audio_file}")
-            audio = mp.AudioFileClip(audio_file)
-        else:
-            raise ValueError("Expected a valid file path for audio.")
+        # Validate input files
+        if not audio_file:
+            raise ValueError("No audio file provided.")
+        if not images or len(images) == 0:
+            raise ValueError("No images provided.")
 
+        # Load audio file
+        print(f"Audio file path: {audio_file}")
+        audio = mp.AudioFileClip(audio_file)
         audio_duration = audio.duration
-        image_clips = []
-        image_count = len(images)
-        image_duration = audio_duration / image_count
-
-        log_message(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
 
-        # Iterate over images, resize them, and create video clips
-        for img_path in images:
-            if isinstance(img_path, str):  # Ensure img_path is a string (file path)
-                log_message(f"Processing image: {img_path}")
-                img = Image.open(img_path)
-                img = resize_and_fit_image(img, target_size=(1280, 720))
+        print(f"Audio duration: {audio_duration} seconds")
 
-                # Create image clip and apply zoom effect
-                img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(30)
-                img_clip = apply_zoom_effect(img_clip)
+        # Process images
+        image_clips = []
+        image_duration = audio_duration / len(images)
 
-                # Add transition effect - Crossfade In
-                if len(image_clips) > 0:
-                    img_clip = img_clip.crossfadein(1)
+        for img_path in images:
+            print(f"Processing image: {img_path}")
+            img = Image.open(img_path)
+            img = resize_and_fit_image(img, target_size=(1280, 720))
 
-                image_clips.append(img_clip)
+            # Create image clip with zoom effect
+            img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(30)
+            img_clip = apply_zoom_effect(img_clip)
+            image_clips.append(img_clip)
 
-        log_message(f"Image clips: {len(image_clips)} clips created.")
+        print(f"Generated {len(image_clips)} image clips.")
 
-        # Concatenate image clips with audio
+        # Concatenate image clips and set audio
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
 
-        # Set output path
-        output_path = '/tmp/generated_video.mp4'
-        log_message(f"Saving video to {output_path}")
-        video.write_videofile(output_path, codec='libx264', audio_codec='aac', threads=4, fps=30, preset='ultrafast')
+        # Save video to temporary directory
+        output_path = "/tmp/generated_video.mp4"
+        print(f"Saving video to {output_path}...")
+        video.write_videofile(output_path, codec="libx264", audio_codec="aac")
 
-        return output_path  # Return the file path for Gradio output
+        return output_path
 
     except Exception as e:
-        error_message = f"Error during video generation: {str(e)}"
-        log_message(error_message)
-        return error_message
+        error_message = f"Error during video generation: {e}"
+        print(error_message)  # Log error
+        return error_message  # Return error as feedback
 
 # Gradio interface setup
 def gradio_interface():
@@ -104,11 +94,15 @@ def gradio_interface():
                 generate_button = gr.Button("Generate Video")
                 
             output_video = gr.Video(label="Generated Video")
-        
-        # Generate button triggers video creation process
-        generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=output_video)
+            error_output = gr.Textbox(label="Error Log", interactive=False)
+
+        generate_button.click(
+            fn=process_and_generate_video,
+            inputs=[mp3_input, image_input],
+            outputs=[output_video, error_output]
+        )
     
     demo.launch()
 
-# Run the interface
+# Run the Gradio interface
 gradio_interface()