Update app.py

app.py

@@ -8,28 +8,34 @@ import sys
 from tools.asesoramiento_tool import asesorar_consulta
 from tools.triaje_tool import triaje_primer_tool
 
-# Ruta para almacenar el modelo
+# Configuración y carga del modelo de lenguaje especializado
 model_dir = os.path.join(os.path.dirname(__file__), "models")
 os.makedirs(model_dir, exist_ok=True)
 model_path = os.path.join(model_dir, "en_core_sci_md")
 
-# Descargar y cargar el modelo si no existe
 if not os.path.exists(model_path):
     os.system(f"{sys.executable} -m spacy download en_core_sci_md --target {model_dir}")
 nlp = spacy.load(model_path)
 
+# Cargar modelos de speech-to-text y text-generation
 whisper_model = WhisperModel("tiny", device="cpu", compute_type="int8")
 health_nlp = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def speech_to_text(audio_path):
-    segments, _ = whisper_model.transcribe(audio_path, language="es")
-    return " ".join([segment.text for segment in segments])
+    try:
+        segments, _ = whisper_model.transcribe(audio_path, language="es")
+        return " ".join([segment.text for segment in segments])
+    except Exception as e:
+        return f"Error en la transcripción: {str(e)}"
 
 def text_to_speech(text):
-    tts = gTTS(text, lang="es")
-    audio_file = "response.mp3"
-    tts.save(audio_file)
-    return audio_file
+    try:
+        tts = gTTS(text, lang="es")
+        audio_file = "response.mp3"
+        tts.save(audio_file)
+        return audio_file
+    except Exception as e:
+        return None
 
 def process_input(audio, mode="asesoramiento"):
     if audio is None:
@@ -42,6 +48,7 @@ def process_input(audio, mode="asesoramiento"):
         respuesta = resultado["respuesta"]
     elif mode == "triaje":
         doc = nlp(query.lower())
+        # Se mejora la extracción utilizando una lista ampliada de entidades y posibles sinónimos
         symptoms = [ent.text for ent in doc.ents if ent.label_ in ["SYMPTOM", "DISEASE"]]
         data = {
             "sintomas": " ".join(symptoms) if symptoms else query,
@@ -73,4 +80,4 @@ with gr.Blocks(title="Agente Médico Speech-to-Speech") as demo:
     
     btn.click(fn=process_input, inputs=[audio_input, mode_select], outputs=[audio_output, text_output])
 
-demo.launch()
+demo.launch()