Update app.py

app.py

@@ -4,7 +4,7 @@ import tempfile
 import numpy as np
 import cv2
 from PIL import Image
-from moviepy import AudioFileClip, VideoFileClip, ImageClip
+import subprocess
 import random
 import time
 
@@ -73,11 +73,27 @@ def add_sparkle_effect(frame, density=0.001):
             cv2.circle(frame, (int(x), int(y)), size, color, -1)
     return frame
 
+def get_audio_duration(audio_path):
+    """Get the duration of an audio file using ffprobe"""
+    cmd = [
+        'ffprobe', 
+        '-v', 'error', 
+        '-show_entries', 'format=duration', 
+        '-of', 'default=noprint_wrappers=1:nokey=1', 
+        audio_path
+    ]
+    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    return float(result.stdout.strip())
+
 def create_video(image_path, audio_path, effect_type, output_path, fps=24):
     """Create a video from an image and audio with the selected effect"""
-    # Load audio to get duration
-    audio_clip = AudioFileClip(audio_path)
-    duration = audio_clip.duration
+    # Get audio duration
+    try:
+        duration = get_audio_duration(audio_path)
+    except Exception as e:
+        st.error(f"Error getting audio duration: {e}")
+        # Fallback to a default duration if ffprobe fails
+        duration = 30.0
     
     # Load image
     img = cv2.imread(image_path)
@@ -128,21 +144,24 @@ def create_video(image_path, audio_path, effect_type, output_path, fps=24):
     # Release the video writer
     out.release()
     
-    # Alternative approach using direct moviepy functionalities
-    # First create a mute video clip
-    video_clip = VideoFileClip(temp_video_path)
-    
-    # Then set its audio to our audio clip
-    final_clip = video_clip.set_audio(audio_clip)
+    # Combine the silent video with audio using ffmpeg directly
+    cmd = [
+        'ffmpeg',
+        '-i', temp_video_path,  # Input video file
+        '-i', audio_path,       # Input audio file
+        '-map', '0:v',          # Use video from first input
+        '-map', '1:a',          # Use audio from second input
+        '-c:v', 'copy',         # Copy the video codec
+        '-c:a', 'aac',          # Use AAC for audio
+        '-shortest',            # End when the shortest input ends
+        output_path             # Output file
+    ]
     
-    # Write the result to our output file
-    final_clip.write_videofile(output_path, codec='libx264')
+    # Run ffmpeg command
+    subprocess.run(cmd, check=True)
     
-    # Clean up temporary file and close clips
+    # Clean up temporary file
     os.remove(temp_video_path)
-    video_clip.close()
-    audio_clip.close()
-    final_clip.close()
     
     progress_bar.progress(1.0)
     return output_path