Update app.py

app.py

@@ -199,23 +199,34 @@ 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..."):
+            # Assuming save_temporary_audio_file and other necessary functions are defined
+            # Ensure overlay_audio_file is saved to a temporary file if present
+            overlay_audio_path = None
+            if overlay_audio_file is not None:
+                overlay_audio_path = save_temporary_audio_file(overlay_audio_file)
+
+            # Process the uploaded video file
             base64Frame, video_filename, video_duration = video_to_frames(uploaded_file, 1)
-            prompt = script_templates[selected_script_type]
+            prompt = script_templates.get(selected_script_type, "Default prompt if script type not found").format(selected_duration=selected_duration)
             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"
-            final_video_filename = merge_audio_video(video_filename, audio_filename, output_video_filename, overlay_audio_file)
-            st.video(final_video_filename)
+            final_video_filename = merge_audio_video(video_filename, audio_filename, output_video_filename, overlay_audio_path)
+
+            if final_video_filename:
+                st.video(final_video_filename)
 
-            os.remove(video_filename)
-            os.remove(audio_filename)
-            os.remove(final_video_filename)
+                # Cleanup
+                os.remove(video_filename)
+                os.remove(audio_filename)
+                if overlay_audio_path:
+                    os.remove(overlay_audio_path)
+                # Consider how to handle final_video_filename as it might be needed for the user to download/view
 
 if __name__ == "__main__":
     main()
-
 # from dotenv import load_dotenv
 # import streamlit as st
 # from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip