Update app.py

app.py

@@ -3,30 +3,45 @@ import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
 import spaces
+import logging
+
+# Configura o logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 @spaces.GPU(duration=120)
-@spaces.GPU(duration=120) 
 def generate_music(description, melody_audio):
+    logging.info("Iniciando a geração de música.")
 
+    # Carrega o modelo pré-treinado
+    logging.info("Carregando o modelo pré-treinado.")
     model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
-    model.set_generation_params(duration=8)  
+    model.set_generation_params(duration=8)
 
     if description:
         description = [description]
         if melody_audio:
+            logging.info(f"Carregando a melodia de áudio de: {melody_audio}")
             melody, sr = torchaudio.load(melody_audio)
-            wav = model.generate_with_chroma(description, melody[None], sr) 
+            logging.info("Gerando música com descrição e melodia.")
+            wav = model.generate_with_chroma(description, melody[None], sr)
         else:
-            wav = model.generate(description)  
+            logging.info("Gerando música apenas com descrição.")
+            wav = model.generate(description)
     else:
-        wav = model.generate_unconditional(1)  
+        logging.info("Gerando música de forma incondicional.")
+        wav = model.generate_unconditional(1)
 
     output_path = 'output.wav'
+    logging.info(f"Salvando a música gerada em: {output_path}")
     audio_write(output_path, wav[0].cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
     
-    return output_path
+    # Verifica a forma do tensor de áudio e se foi salvo corretamente
+    logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
+    logging.info("Música gerada e salva com sucesso.")
 
+    return output_path
 
+# Define a interface Gradio
 description = gr.Textbox(label="Description", placeholder="acoustic, guitar, melody, trap, d minor, 90 bpm")
 melody_audio = gr.Audio(label="Melody Audio (optional)", type="filepath")
 output_path = gr.Audio(label="Generated Music", type="filepath")