Update app.py

app.py

@@ -16,80 +16,122 @@ def gpu_decorator(func):
     else:
         return func
 
+
 @gpu_decorator
-def _fn(path, solver, nfe, tau, denoising, unlimited):
+def enhance_audio(
+    path, solver, nfe, tau, denoising, unlimited, show_info=gr.Info
+):
     if path is None:
         return "Please upload an audio file.", None
+    
     info = torchaudio.info(path)
     if not unlimited and (info.num_frames / info.sample_rate > 60):
         return "Only audio files shorter than 60 seconds are supported.", None
+    
     solver = solver.lower()
     nfe = int(nfe)
     lambd = 0.9 if denoising else 0.1
+    
     dwav, sr = torchaudio.load(path)
     dwav = dwav.mean(dim=0)
-    wav1, new_sr = denoise(dwav, sr)  # Remove o argumento device
-    wav2, new_sr = enhance(dwav, sr, nfe=nfe, solver=solver, lambd=lambd, tau=tau)  # Remove o argumento device
+    
+    if show_info:
+        show_info("Denoising audio...")
+    wav1, new_sr = denoise(dwav, sr)  
+    
+    if show_info:
+        show_info("Enhancing audio...")
+    wav2, new_sr = enhance(dwav, sr, nfe=nfe, solver=solver, lambd=lambd, tau=tau)  
+    
     wav1 = wav1.cpu().numpy()
     wav2 = wav2.cpu().numpy()
+    
     return (new_sr, wav1), (new_sr, wav2)
 
 
-
 with gr.Blocks() as app:
-    inputs = [
-        gr.Audio(type="filepath", label="Input Audio"),
-        gr.Dropdown(
-            choices=["Midpoint", "RK4", "Euler"],
-            value="Midpoint",
-            label="CFM ODE Solver (Midpoint is recommended)",
-        ),
-        gr.Slider(
-            minimum=1,
-            maximum=128,
-            value=64,
-            step=1,
-            label="CFM Number of Function Evaluations (higher values in general yield better quality but may be slower)",
-        ),
-        gr.Slider(
-            minimum=0,
-            maximum=1,
-            value=0.5,
-            step=0.01,
-            label="CFM Prior Temperature (higher values can improve quality but can reduce stability)",
-        ),
-        gr.Checkbox(
-            value=False,
-            label="Denoise Before Enhancement (tick if your audio contains heavy background noise)",
-        ),
-        gr.Checkbox(
-            value=False,
-            label="Allow Unlimited Audio Length (supports files longer than 60 seconds)",
-        ),
-    ]
-    outputs = [
-        gr.Audio(label="Output Denoised Audio"),
-        gr.Audio(label="Output Enhanced Audio"),
-    ]
-    # Interface Gradio
-    interface = gr.Interface(
-        fn=_fn,
-        title="Audio Enhancement",
-        description="AI-driven audio enhancement powered by Resemble AI.",
-        inputs=inputs,
-        outputs=outputs,
+    gr.Markdown("# Audio Enhancement with Resemble AI")
+
+    with gr.Row():
+        with gr.Column():
+            # Entradas
+            audio_input = gr.Audio(type="filepath", label="Input Audio")
+            solver_dropdown = gr.Dropdown(
+                choices=["Midpoint", "RK4", "Euler"],
+                value="Midpoint",
+                label="CFM ODE Solver",
+                info="Midpoint is recommended",
+            )
+            nfe_slider = gr.Slider(
+                minimum=1,
+                maximum=128,
+                value=64,
+                step=1,
+                label="CFM Number of Function Evaluations",
+                info="Higher values generally yield better quality but may be slower",
+            )
+            tau_slider = gr.Slider(
+                minimum=0,
+                maximum=1,
+                value=0.5,
+                step=0.01,
+                label="CFM Prior Temperature",
+                info="Higher values can improve quality but can reduce stability",
+            )
+            denoising_checkbox = gr.Checkbox(
+                value=False,
+                label="Denoise Before Enhancement",
+                info="Tick if your audio contains heavy background noise",
+            )
+            unlimited_checkbox = gr.Checkbox(
+                value=False,
+                label="Allow Unlimited Audio Length",
+                info="Supports files longer than 60 seconds",
+            )
+            enhance_btn = gr.Button("Enhance Audio", variant="primary")
+        
+        with gr.Column():
+            # Saídas
+            denoised_audio_output = gr.Audio(label="Output Denoised Audio")
+            enhanced_audio_output = gr.Audio(label="Output Enhanced Audio")
+
+    # Associação do botão `enhance_btn` à função `enhance_audio`
+    enhance_btn.click(
+        enhance_audio,
+        inputs=[
+            audio_input,
+            solver_dropdown,
+            nfe_slider,
+            tau_slider,
+            denoising_checkbox,
+            unlimited_checkbox,
+        ],
+        outputs=[
+            denoised_audio_output,
+            enhanced_audio_output,
+        ],
     )
-    app = interface.queue()
 
 
-def main():
+@click.command()
+@click.option("--port", "-p", default=None, type=int, help="Port to run the app on")
+@click.option("--host", "-H", default=None, help="Host to run the app on")
+@click.option(
+    "--share",
+    "-s",
+    default=False,
+    is_flag=True,
+    help="Share the app via Gradio share link",
+)
+@click.option("--api", "-a", default=True, is_flag=True, help="Allow API access")
+def main(port, host, share, api):
     global app
     print("Starting app...")
-    app.launch(share=True)
+    app.queue(api_open=api).launch(server_name=host, server_port=port, share=share, show_api=api)
 
 
 if __name__ == "__main__":
     if not USING_SPACES:
         main()
     else:
-        app.launch()
+        app.queue().launch()