Update app.py

app.py

@@ -6,6 +6,43 @@ API_URL = "https://ia.nebuia.com/api/v1/integrator/extractor/from/excel"
 API_KEY = "2J7NFWQ-236PGAM-2S9RD3C-2DG54CY"
 API_SECRET = "a47abf97-866b-4154-b29c-346c9b02919e"
 
+import re
+import json
+
+def extract_json_from_response(response_text):
+    """
+    Detecta si la respuesta contiene un bloque JSON con formato markdown ```json ```
+    y extrae su contenido, o devuelve el texto original si no se encuentra ese patrón.
+    
+    Args:
+        response_text (str): El texto de respuesta que puede contener bloques JSON
+        
+    Returns:
+        str: El contenido JSON extraído o el texto original
+    """
+    # Patrón para encontrar bloques de código JSON con formato markdown
+    json_pattern = r"```json\s+([\s\S]*?)\s+```"
+    
+    # Buscar coincidencias
+    match = re.search(json_pattern, response_text)
+    
+    if match:
+        # Extraer el contenido JSON del bloque
+        json_content = match.group(1).strip()
+        
+        # Verificar que sea un JSON válido
+        try:
+            # Intentar cargar el JSON para validarlo
+            json.loads(json_content)
+            return json_content
+        except json.JSONDecodeError:
+            # Si no es un JSON válido, devolver el texto original
+            return response_text
+    
+    # Si no hay coincidencias, devolver el texto original
+    return response_text
+
+# Modificación de la función send_excel_to_api para usar la función de extracción
 def send_excel_to_api(file_path, json_structure):
     headers = {
         'key': API_KEY,
@@ -25,8 +62,13 @@ def send_excel_to_api(file_path, json_structure):
         
         if response.status_code == 200:
             result = response.json()
-            response1 = json.dumps(result["payload"], indent=2, ensure_ascii=False)
-            return response1
+            # Si la respuesta contiene un payload, formatearla como JSON
+            if "payload" in result:
+                response_json = json.dumps(result["payload"], indent=2, ensure_ascii=False)
+                # Extraer JSON de la respuesta si está en formato markdown
+                return extract_json_from_response(response_json)
+            else:
+                return json.dumps(result, indent=2, ensure_ascii=False)
         else:
             return f"Error: Received status code {response.status_code}\n{response.text}"
     except requests.RequestException as e: