Update app.py

app.py

@@ -7,9 +7,8 @@ import io
 import openai
 import os
 import requests
-import tempfile
 
-# Set up page configuration first
+# Set up page configuration
 st.set_page_config(page_title="AI Voiceover", page_icon="🔮")
 
 # Load environment variables
@@ -118,7 +117,6 @@ def merge_audio_video(video_filename, audio_filename, output_filename, overlay_a
     return output_filename
 
 def main():
-    # st.set_page_config(page_title="AI Voiceover", page_icon="🔮")
     st.title("Pixio Video to Voiceover 🎥🔮")
 
     if not check_password():
@@ -130,27 +128,10 @@ def main():
         return
 
     uploaded_file = st.file_uploader("Select a video file", type=["mp4", "avi"])
-    if uploaded_file is not None:
-        st.video(uploaded_file)
-
-overlay_audio_file = st.file_uploader("Upload overlay audio (optional)", type=["mp3", "wav"])
-
-# After uploading, write it to a temporary file
-if overlay_audio_file is not None:
-    # Create a temporary file
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
-        overlay_audio_path = temp_file.name
-        # Read content from the uploaded file
-        content = overlay_audio_file.read()
-        # Write content to the temporary file
-        temp_file.write(content)
-    
-    # Now overlay_audio_path contains the path to the temporary file
-    # and can be safely used with MoviePy
+    overlay_audio_file = st.file_uploader("Upload overlay audio (optional)", type=["mp3", "wav"])
 
     voice_options = {'Echo (Male)': 'echo', 'Fable (Male)': 'fable', 'Onyx (Male)': 'onyx', 'Nova (Female)': 'nova', 'Shimmer (Female)': 'shimmer', 'Alloy (Female)': 'alloy'}
-    option = st.selectbox('Choose the voice you want', list(voice_options.keys()))
-    classify = voice_options[option]
+    voice = st.selectbox('Choose the voice you want', list(voice_options.keys()))
 
     duration_options = list(range(10, 121, 10)) # 10 to 120 seconds, in 10-second intervals
     selected_duration = st.selectbox('Select the desired video duration (seconds)', duration_options)
@@ -186,8 +167,9 @@ if overlay_audio_file is not None:
     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
             text = frames_to_story(base64Frame, prompt, openai_key)
-            audio_filename = text_to_audio(text, openai_key, classify)
+            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)