Update app.py

app.py

@@ -1,20 +1,20 @@
+
 try:
-    import spaces  # Presumindo que spaces represente o ambiente do Hugging Face Spaces.
+    import spaces  # Para ambientes específicos como Hugging Face Spaces
     USING_SPACES = True
 except ImportError:
     USING_SPACES = False
 
-import argparse
-from functools import partial
 import gradio as gr
 import torch
 import torchaudio
+from functools import partial
 from resemble_enhance.enhancer.inference import denoise, enhance
 
+
+
+
 def gpu_decorator(func):
-    """
-    Decorador para determinar o dispositivo (CPU ou GPU) baseado no ambiente.
-    """
     def wrapper(*args, **kwargs):
         device = "cpu"
         if not USING_SPACES and torch.cuda.is_available():
@@ -23,11 +23,9 @@ def gpu_decorator(func):
         return func(*args, **kwargs)
     return wrapper
 
+
 @gpu_decorator
 def _fn(path, solver, nfe, tau, denoising, unlimited, device="cpu"):
-    """
-    Função para processar o áudio com as opções especificadas.
-    """
     if path is None:
         return "Please upload an audio file.", None
 
@@ -50,11 +48,12 @@ def _fn(path, solver, nfe, tau, denoising, unlimited, device="cpu"):
 
     return (new_sr, wav1), (new_sr, wav2)
 
-def create_app(args):
+
+def create_app():
     """
-    Cria a interface do Gradio com as configurações especificadas.
+    Cria a interface do Gradio.
     """
-    inputs: list = [
+    inputs = [
         gr.Audio(type="filepath", label="Input Audio"),
         gr.Dropdown(
             choices=["Midpoint", "RK4", "Euler"],
@@ -79,40 +78,42 @@ def create_app(args):
             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: list = [
+    outputs = [
         gr.Audio(label="Output Denoised Audio"),
         gr.Audio(label="Output Enhanced Audio"),
     ]
 
+    # Interface Gradio
     interface = gr.Interface(
-        fn=partial(_fn, unlimited=args.unlimited),
-        title="Resemble Enhance",
-        description="AI-driven audio enhancement for your audio files, powered by Resemble AI.",
+        fn=_fn,
+        title="Audio Enhancement",
+        description="AI-driven audio enhancement powered by Resemble AI.",
         inputs=inputs,
         outputs=outputs,
-        live=False,  # Evitar execução em tempo real.
-        queue=True,  # Adiciona suporte a fila.
     )
-    return interface
 
-def main(port=None, host="0.0.0.0", share=False, api=False):
+    return interface.queue()  # Ativa filas para processamento
+
+
+def main(port=7860, host="0.0.0.0", share=False, api=False):
     """
-    Função principal para inicializar o aplicativo no ambiente adequado.
+    Inicializa a aplicação dependendo do ambiente.
     """
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--unlimited", action="store_true", help="Permitir áudios maiores que 60 segundos.")
-    args = parser.parse_args()
-
     global app
-    app = create_app(args)
     print("Starting app...")
-    app.queue(api_open=api).launch(server_name=host, server_port=port, share=share, show_api=api)
+    app = create_app()
+    app.launch(server_name=host, server_port=port, share=share, show_api=api)
+
 
 if __name__ == "__main__":
     if not USING_SPACES:
         main()
     else:
-        app = create_app(argparse.Namespace(unlimited=True))
-        app.queue().launch()
+        app = create_app()
+        app.launch()