Update app.py

app.py

@@ -28,6 +28,14 @@ def check_password():
             st.error("Incorrect password")
         return False
 
+
+def save_temporary_audio_file(uploaded_file):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav", mode='wb') as tmpfile:  # Ensure mode is set to 'wb' for binary write
+        # Directly write the uploaded file content to the temporary file
+        tmpfile.write(uploaded_file.read())  # Use .read() to read bytes from the uploaded file
+        return tmpfile.name
+
+        
 def video_to_frames(video_file, frame_sampling_rate=1):
     with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmpfile:
         tmpfile.write(video_file.read())
@@ -201,15 +209,14 @@ def main():
 
     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:
+                # Save the uploaded overlay audio file to a temporary file
                 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.get(selected_script_type, "Default prompt if script type not found").format(selected_duration=selected_duration)
+            base64Frame, video_filename, video_duration = video_to_frames(uploaded_file, frame_sampling_rate=1)
+            prompt = script_templates[selected_script_type].format(selected_duration=selected_duration)  # Fixed to directly access script_templates
             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"
@@ -223,7 +230,7 @@ def main():
                 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
+                # Consider adding functionality for the user to download/view the final video
 
 if __name__ == "__main__":
     main()