Update app.py

app.py

@@ -1,38 +1,68 @@
 import gradio as gr
 from audiocraft.models import MusicGen, AudioGen
 from audiocraft.data.audio import audio_write
-import torchaudio
 import tempfile
+import torch
+
+# Verificar se há GPU disponível
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Usando dispositivo: {device}")
 
 # Carregar modelos
-musicgen = MusicGen.get_pretrained('facebook/musicgen-small')
-audiogen = AudioGen.get_pretrained('facebook/audiogen-medium')
+try:
+    musicgen = MusicGen.get_pretrained('facebook/musicgen-small').to(device)
+    audiogen = AudioGen.get_pretrained('facebook/audiogen-medium').to(device)
+    # Definir duração padrão
+    musicgen.set_generation_params(duration=5)  # Aumentei para 5 segundos
+    audiogen.set_generation_params(duration=5)
+except Exception as e:
+    print(f"Erro ao carregar modelos: {e}")
+    raise
 
-# Definir duração padrão
-musicgen.set_generation_params(duration=2)
-audiogen.set_generation_params(duration=2)
+def generate_audio(prompt, model, model_name):
+    try:
+        print(f"Gerando {model_name} para:", prompt)
+        wav = model.generate([prompt])[0]
+        print("Áudio gerado, salvando...")
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
+            audio_write(
+                fp.name, 
+                wav.cpu(), 
+                model.sample_rate, 
+                strategy="loudness",
+                loudness_compressor=True
+            )
+            print(f"Arquivo {model_name} salvo em", fp.name)
+            return fp.name
+    except Exception as e:
+        print(f"Erro ao gerar {model_name}: {e}")
+        raise
 
 def generate_music(prompt):
-    print("Gerando música para:", prompt)
-    wav = musicgen.generate([prompt])[0]
-    print("Áudio gerado, salvando...")
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
-        audio_write(fp.name, wav.cpu(), musicgen.sample_rate, strategy="loudness", loudness_compressor=True)
-        print("Arquivo salvo em", fp.name)
-        return fp.name       
+    return generate_audio(prompt, musicgen, "música")
 
 def generate_sound_effect(prompt):
-    print("Gerando som para:", prompt)
-    wav = audiogen.generate([prompt])[0]
-    print("Som gerado, salvando...")
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
-        audio_write(fp.name, wav.cpu(), audiogen.sample_rate, strategy="loudness", loudness_compressor=True)
-        print("Arquivo salvo em", fp.name)
-        return fp.name
+    return generate_audio(prompt, audiogen, "efeito sonoro")
 
-music_interface = gr.Interface(fn=generate_music, inputs=gr.Text(label="Prompt para música"), outputs=gr.Audio(type="filepath"), title="🎵 MusicGen")
-sound_interface = gr.Interface(fn=generate_sound_effect, inputs=gr.Text(label="Prompt para efeito sonoro"), outputs=gr.Audio(type="filepath"), title="🔊 AudioGen")
+# Configuração da interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎵 Gerador de Áudio com MusicGen e AudioGen")
+    with gr.Tabs():
+        with gr.TabItem("MusicGen 🎵"):
+            gr.Markdown("## Gerar Música")
+            with gr.Row():
+                music_input = gr.Text(label="Descreva a música que deseja")
+                music_output = gr.Audio(label="Música Gerada", type="filepath")
+            music_button = gr.Button("Gerar Música")
+            music_button.click(fn=generate_music, inputs=music_input, outputs=music_output)
+            
+        with gr.TabItem("AudioGen 🔊"):
+            gr.Markdown("## Gerar Efeitos Sonoros")
+            with gr.Row():
+                sound_input = gr.Text(label="Descreva o efeito sonoro que deseja")
+                sound_output = gr.Audio(label="Som Gerado", type="filepath")
+            sound_button = gr.Button("Gerar Som")
+            sound_button.click(fn=generate_sound_effect, inputs=sound_input, outputs=sound_output)
 
-# Interface combinada
-demo = gr.TabbedInterface([music_interface, sound_interface], ["MusicGen", "AudioGen"])
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()