Update app.py

app.py

@@ -92,29 +92,74 @@ def text_to_audio(text, api_key, voice):
     return audio_filename
 
 def merge_audio_video(video_filename, audio_filename, output_filename, overlay_audio_path=None):
-    video_clip = VideoFileClip(video_filename)
-    audio_clip = AudioFileClip(audio_filename)
+    try:
+        video_clip = VideoFileClip(video_filename)
+        audio_clip = AudioFileClip(audio_filename)
+    except Exception as e:
+        print(f"Error loading video or audio clip: {e}")
+        return None
 
     audio_clips = [audio_clip]  # Start with the main audio clip
 
-    # If there's an overlay audio file, add it to the composite
-    if overlay_audio_path:
-        overlay_clip = AudioFileClip(overlay_audio_path).volumex(0.2)
-        audio_clips.append(overlay_clip.set_duration(audio_clip.duration))
+    # If there's an overlay audio file and it exists, add it to the composite
+    if overlay_audio_path and os.path.exists(overlay_audio_path):
+        try:
+            overlay_clip = AudioFileClip(overlay_audio_path).volumex(0.2)
+            audio_clips.append(overlay_clip.set_duration(audio_clip.duration))
+        except Exception as e:
+            print(f"Error loading overlay audio clip: {e}")
+            # Optionally handle the error or continue without the overlay
+    elif overlay_audio_path:
+        print("Overlay audio file does not exist or path is incorrect.")
 
     composite_audio_clip = CompositeAudioClip(audio_clips)
 
     if composite_audio_clip.duration > video_clip.duration:
         # Extend video with the last frame if audio is longer
-        last_frame = video_clip.to_ImageClip(t=video_clip.duration-1).set_duration(composite_audio_clip.duration - video_clip.duration)
-        video_clip = concatenate_videoclips([video_clip, last_frame])
-
-    final_clip = video_clip.set_audio(composite_audio_clip)
-    final_clip.write_videofile(output_filename, codec='libx264', audio_codec="aac")
-    video_clip.close()
-    audio_clip.close()
+        try:
+            last_frame = video_clip.to_ImageClip(t=video_clip.duration-1).set_duration(composite_audio_clip.duration - video_clip.duration)
+            video_clip = concatenate_videoclips([video_clip, last_frame])
+        except Exception as e:
+            print(f"Error extending video clip: {e}")
+            return None
+
+    try:
+        final_clip = video_clip.set_audio(composite_audio_clip)
+        final_clip.write_videofile(output_filename, codec='libx264', audio_codec="aac")
+    except Exception as e:
+        print(f"Error creating final video file: {e}")
+        return None
+    finally:
+        video_clip.close()
+        audio_clip.close()
+        if 'overlay_clip' in locals():
+            overlay_clip.close()
 
     return output_filename
+# def merge_audio_video(video_filename, audio_filename, output_filename, overlay_audio_path=None):
+#     video_clip = VideoFileClip(video_filename)
+#     audio_clip = AudioFileClip(audio_filename)
+
+#     audio_clips = [audio_clip]  # Start with the main audio clip
+
+#     # If there's an overlay audio file, add it to the composite
+#     if overlay_audio_path:
+#         overlay_clip = AudioFileClip(overlay_audio_path).volumex(0.2)
+#         audio_clips.append(overlay_clip.set_duration(audio_clip.duration))
+
+#     composite_audio_clip = CompositeAudioClip(audio_clips)
+
+#     if composite_audio_clip.duration > video_clip.duration:
+#         # Extend video with the last frame if audio is longer
+#         last_frame = video_clip.to_ImageClip(t=video_clip.duration-1).set_duration(composite_audio_clip.duration - video_clip.duration)
+#         video_clip = concatenate_videoclips([video_clip, last_frame])
+
+#     final_clip = video_clip.set_audio(composite_audio_clip)
+#     final_clip.write_videofile(output_filename, codec='libx264', audio_codec="aac")
+#     video_clip.close()
+#     audio_clip.close()
+
+#     return output_filename
 
 def main():
     st.title("Pixio Video to Voiceover 🎥🔮")