Improve error handling and Basic Pitch implementation

app.py

@@ -5,11 +5,23 @@ from model_inference import (
 )
 
 def procesar_wav(input_wav_path: str):
-    # 1) Generar archivos midi
-    midi_path = inferir_basic_pitch(input_wav_path)
-    
-    # Devolver la ruta del archivo MIDI generado
-    return midi_path
+    if not input_wav_path:
+        return None
+        
+    try:
+        # 1) Generar archivos midi
+        midi_path = inferir_basic_pitch(input_wav_path)
+        
+        # Verificar que se generó el archivo
+        if midi_path and os.path.exists(midi_path):
+            return midi_path
+        else:
+            print("Error: No se pudo generar el archivo MIDI")
+            return None
+            
+    except Exception as e:
+        print(f"Error en procesar_wav: {e}")
+        return None
 
 demo = gr.Interface(
     fn=procesar_wav,

model_inference.py

@@ -30,8 +30,7 @@ def inferir_basic_pitch(input_file: str) -> str:
     midi_path = os.path.join(BASE_MIDI_DIR, f"{base_name}_basic_pitch.mid")
     
     try:
-        # Cargar modelo de Basic Pitch
-        basic_pitch_model = tf.saved_model.load(str(ICASSP_2022_MODEL_PATH))
+        print(f"Procesando archivo: {input_file}")
         
         # Realizar predicción y guardar archivo MIDI
         predict_and_save(
@@ -43,7 +42,18 @@ def inferir_basic_pitch(input_file: str) -> str:
             save_notes=False,
         )
         
-        return midi_path
+        # Basic Pitch genera el archivo con un nombre específico
+        # Necesitamos encontrar el archivo MIDI generado
+        generated_files = [f for f in os.listdir(BASE_MIDI_DIR) if f.endswith('.mid')]
+        if generated_files:
+            # Tomar el archivo más reciente
+            latest_file = max([os.path.join(BASE_MIDI_DIR, f) for f in generated_files], 
+                            key=os.path.getctime)
+            print(f"Archivo MIDI generado: {latest_file}")
+            return latest_file
+        else:
+            print("No se generó ningún archivo MIDI")
+            return None
         
     except Exception as e:
         print(f"Error durante la inferencia: {e}")