Update app.py

app.py

@@ -91,26 +91,29 @@ 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):
+def merge_audio_video(video_filename, audio_filename, output_filename, overlay_audio_file=None):
     try:
         video_clip = VideoFileClip(video_filename)
         audio_clip = AudioFileClip(audio_filename)
     except Exception as e:
-        print(f"Error loading video or audio clip: {e}")
+        st.error(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 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.")
+    # If there's an overlay audio file, add it to the composite
+    overlay_audio_path = None
+    if overlay_audio_file is not None:
+        overlay_audio_path = save_temporary_audio_file(overlay_audio_file)
+        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:
+                st.error(f"Error loading overlay audio clip: {e}")
+                # Optionally handle the error or continue without the overlay
+        else:
+            st.error("Overlay audio file does not exist or path is incorrect.")
 
     composite_audio_clip = CompositeAudioClip(audio_clips)
 
@@ -120,50 +123,29 @@ def merge_audio_video(video_filename, audio_filename, output_filename, overlay_a
             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}")
+            st.error(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}")
+        st.error(f"Error creating final video file: {e}")
         return None
     finally:
         video_clip.close()
         audio_clip.close()
-        if 'overlay_clip' in locals():
-            overlay_clip.close()
+        if overlay_audio_path:
+            os.remove(overlay_audio_path)
 
     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 save_temporary_audio_file(uploaded_file):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
+        tmpfile.write(uploaded_file.getvalue())
+        return tmpfile.name
 
 def main():
-    st.title("Pixio Video to Voiceover 🎥🔮")
-
     if not check_password():
         return
 
@@ -217,10 +199,10 @@ def main():
     prompt = st.text_area("Edit the voiceover script prompt as needed:", value=initial_prompt.format(selected_duration=selected_duration), height=300)
 
 
-    if uploaded_file is not None and st.button("START PROCESSING"):
+if uploaded_file is not None and st.button("START PROCESSING"):
         with st.spinner("Processing..."):
             base64Frame, video_filename, video_duration = video_to_frames(uploaded_file, 1)
-            prompt = script_templates[selected_script_type].format(duration=selected_duration)  # Customize this based on your needs
+            prompt = script_templates[selected_script_type]
             text = frames_to_story(base64Frame, prompt, openai_key)
             audio_filename = text_to_audio(text, openai_key, voice_options[voice])
             output_video_filename = os.path.splitext(video_filename)[0] + "_output.mp4"
@@ -234,7 +216,6 @@ def main():
 if __name__ == "__main__":
     main()
 
-
 # from dotenv import load_dotenv
 # import streamlit as st
 # from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip