Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import fitz  # PyMuPDF
 from groq import Groq
 from langchain_groq import ChatGroq
+import json
 
 RAID_WEIGHTS = {
     "Dolor": 0.21,
@@ -65,11 +66,29 @@ def evaluate_raid(patient_text, clinical_record):
         # Extraer JSON de la respuesta
         start = response.content.find('{')
         end = response.content.rfind('}') + 1
-        json_response = json.loads(response.content[start:end])
+        try:
+            json_response = json.loads(response.content[start:end])
+        except json.JSONDecodeError:
+            # Si falla el parseo directo, intentar extraer substring
+            json_str = response.content.split('```json')[1].split('```')[0].strip()
+            json_response = json.loads(json_str)
         
         # Validar estructura
         if not all(key in json_response['raid_scores'] for key in RAID_WEIGHTS):
             raise ValueError("Faltan componentes del RAID en la respuesta")
+
+                # Validación adicional
+        if not isinstance(json_response, dict) or 'raid_scores' not in json_response:
+            raise ValueError("Respuesta en formato incorrecto")
+            
+        # Cálculo de verificación
+        calculated_total = sum(
+            json_response['raid_scores'][k] * v 
+            for k, v in RAID_WEIGHTS.items()
+        )
+        
+        # Redondear a 2 decimales para coincidir
+        json_response['raid_total'] = round(calculated_total, 2)
             
         return json_response