Update app.py

app.py

@@ -13,7 +13,8 @@ def get_inference_client():
     return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def extract_and_clean_json(text):
-    json_match = re.search(r'\{.*\}', text, re.DOTALL)
+    # Busca el JSON válido en el texto
+    json_match = re.search(r'\{(?:[^{}]|(?R))*\}', text, re.DOTALL)
     if json_match:
         json_str = json_match.group()
         json_str = re.sub(r'[\n\t\r]', '', json_str)
@@ -39,13 +40,11 @@ def generate_presentation_content(topic, client):
     
     response = client.text_generation(prompt, max_new_tokens=1500, temperature=0.7)
     
-    # Mensaje de depuración para imprimir la respuesta original
     st.text("Respuesta original del modelo:")
     st.code(response)
-
+    
     json_str = extract_and_clean_json(response)
     
-    # Mensaje de depuración para imprimir el JSON después de la limpieza
     st.text("JSON procesado (después de limpieza):")
     st.code(json_str)