update app.py

app.py

@@ -280,7 +280,7 @@ def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics
 
 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.")
+        raise gr.Error("Please Provide an Audio File")
 
     resolution = f"{vidwidth}x{vidheight}"
     res = tuple(map(int, resolution.split('x')))
@@ -288,58 +288,60 @@ def gradio_interface(audio_file, lyrics, output_name, fps=30, vidwidth=1280, vid
     
     # Clean up the temporary file
     os.remove(audio_file)
-    
-    time.sleep(5)
-    
     shutil.rmtree("out")
+
     srt_output = "out.srt" if haslyrics else None
-    return video_file, srt_output
+    return video_file, srt_output, haslyrics
 
-def update_srt_output_visibility(lyrics):
-    haslyrics = lyrics is not None
+def update_srt_output_visibility(haslyrics):
     return gr.update(visible=haslyrics)
 
 with gr.Blocks() as demo:
     gr.Markdown('Upload an MP3 file and configure parameters to create a visualization video.\nOptionally upload a word or line synced lyric file\nEnsure a blank line at the end to avoid conversion errors')
-    
-    with gr.Accordion(label="Audio Settings", open=True):
-        gr.Markdown('# Load your mp3 and lyrics file here')
-        audio_file = gr.File(label="Upload your MP3 file", file_count='single', file_types=['mp3'])
-        lyrics_file = gr.File(label="(Optional) Upload Lyrics as LRC or SRT", file_count='single', file_types=['lrc', 'srt'])
-    
-    with gr.Accordion(label="Video Output Settings"):
-        gr.Markdown('# Configure Video Output Here')
-        output_name = gr.Textbox(label="Output Video Name", value='video')
-        fps_slider = gr.Slider(label="Frames per Second", minimum=20, maximum=60, step=1, value=30)
-        vidwidth_slider = gr.Slider(label="Output Video Width", minimum=100, maximum=2000, value=1280, step=2)
-        vidheight_slider = gr.Slider(label="Output Video Height", minimum=100, maximum=2000, value=720, step=2)
-    
-    with gr.Accordion(label="Advanced Options"):
-        oscres_slider = gr.Slider(label="Number of Visualization Segments", minimum=256, maximum=2048, step=2, value=512)
-    
-    with gr.Accordion(label="Mp3 Metadata"):
-        gr.Markdown('# Add Metadata here if your mp3 does not have one')
-        cover_img = gr.Image(label='Cover Art')
-        title_input = gr.Textbox(label='Title')
-        artist_input = gr.Textbox(label='Artists')
 
-    output_video = gr.Video(label="Output")
-    srt_output = gr.File(label="SRT Output", visible=False)
+    with gr.Row():
+        # Inputs on the left
+        with gr.Column():
+            with gr.Accordion(label="Audio Settings", open=True):
+                gr.Markdown('# Load your mp3 and lyrics file here')
+                audio_file = gr.File(label="Upload your MP3 file", file_count='single', file_types=['mp3'])
+                lyrics_file = gr.File(label="(Optional) Upload Lyrics as LRC or SRT", file_count='single', file_types=['lrc', 'srt'])
+
+            with gr.Accordion(label="Video Output Settings", open=False):
+                gr.Markdown('# Configure Video Output Here')
+                output_name = gr.Textbox(label="Output Video Name", value='video')
+                fps_slider = gr.Slider(label="Frames per Second", minimum=20, maximum=60, step=1, value=30)
+                vidwidth_slider = gr.Slider(label="Output Video Width", minimum=100, maximum=2000, value=1280, step=2)
+                vidheight_slider = gr.Slider(label="Output Video Height", minimum=100, maximum=2000, value=720, step=2)
+        
+            with gr.Accordion(label="Advanced Options", open=False):
+                oscres_slider = gr.Slider(label="Number of Visualization Segments", minimum=256, maximum=2048, step=2, value=512)
+        
+            with gr.Accordion(label="Mp3 Metadata", open=False):
+                gr.Markdown('# Add Metadata here if your mp3 does not have one')
+                cover_img = gr.Image(label='Cover Art')
+                title_input = gr.Textbox(label='Title')
+                artist_input = gr.Textbox(label='Artists')
+
+            # Add a submit button
+            submit_btn = gr.Button("Generate Video")
 
-    # Add a submit button and link it to the gradio_interface function
-    submit_btn = gr.Button("Generate Video")
+        # Outputs on the right
+        with gr.Column():
+            output_video = gr.Video(label="Output")
+            srt_output = gr.File(label="SRT Output", visible=False)
 
     # 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]
+        outputs=[output_video, srt_output, gr.State()]
     )
     
     # Update visibility of SRT output
     demo.load(
         fn=update_srt_output_visibility,
-        inputs=lyrics_file,
+        inputs=gr.State(),
         outputs=srt_output
     )