Update app.py

app.py

@@ -279,9 +279,21 @@ def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics
     return f"{name}.mp4", haslyrics
 
 def gradio_interface(audio_file, lyrics, output_name, fps=30, vidwidth=1280, vidheight=720, oscres=512, img=None, tit=None, ast=None):
+    if audio_file is None:
+        raise ValueError("No audio file provided.")
+        
+    # Save the uploaded file to a temporary location
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_audio_file:
+        temp_audio_file.write(audio_file.read())
+        temp_audio_file_path = temp_audio_file.name
+
     resolution = f"{vidwidth}x{vidheight}"
     res = tuple(map(int, resolution.split('x')))
-    video_file, haslyrics = main(audio_file, output_name, fps=fps, res=res, oscres=oscres, lyrics=lyrics, img=img, tit=tit, ast=ast)
+    video_file, haslyrics = main(temp_audio_file_path, output_name, fps=fps, res=res, oscres=oscres, lyrics=lyrics, img=img, tit=tit, ast=ast)
+    
+    # Clean up the temporary file
+    os.remove(temp_audio_file_path)
+    
     time.sleep(5)
     
     shutil.rmtree("out")
@@ -319,8 +331,11 @@ with gr.Blocks() as demo:
     output_video = gr.Video(label="Output")
     srt_output = gr.File(label="SRT Output", visible=False)
 
-    # Connect the main function
-    demo.load(
+    # Add a submit button and link it to the gradio_interface function
+    submit_btn = gr.Button("Generate Video")
+
+    # Bind the button to the function
+    submit_btn.click(
         fn=gradio_interface,
         inputs=[audio_file, lyrics_file, output_name, fps_slider, vidwidth_slider, vidheight_slider, oscres_slider, cover_img, title_input, artist_input],
         outputs=[output_video, srt_output]