Update app.py

app.py

@@ -4,6 +4,12 @@ import numpy as np
 from PIL import Image
 import os
 
+# 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)
+
 # Resize image while maintaining aspect ratio
 def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
     width, height = img.size
@@ -37,8 +43,11 @@ def apply_zoom_effect(image_clip):
 
 def process_and_generate_video(audio_file, images):
     try:
+        log_message("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.")
@@ -48,11 +57,12 @@ def process_and_generate_video(audio_file, images):
         image_count = len(images)
         image_duration = audio_duration / image_count
 
-        print(f"Audio duration: {audio_duration} seconds, Image count: {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))
 
@@ -66,21 +76,23 @@ def process_and_generate_video(audio_file, images):
 
                 image_clips.append(img_clip)
 
-        print(f"Image clips: {len(image_clips)} clips created.")
+        log_message(f"Image clips: {len(image_clips)} clips created.")
 
         # Concatenate image clips with audio
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
 
         # Set output path
-        output_path = '/content/generated_video.mp4'
+        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')
 
         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)}"
+        error_message = f"Error during video generation: {str(e)}"
+        log_message(error_message)
+        return error_message
 
 # Gradio interface setup
 def gradio_interface():