Update app.py

app.py

@@ -103,22 +103,29 @@ def generate_text(prompt, max_length=2048):
 
 
 def mostrar_respuesta(pregunta):
-    json_obj={}
-    json_obj['respuesta']='Esperando'
-    json_obj['fuente']='Esperando'
-    json_obj['tema']='Esperando'
-    if pregunta!="":
+    respuesta_default = "No se pudo generar una respuesta adecuada."
+    json_obj = {
+        "respuesta": respuesta_default,
+        "fuente": respuesta_default,
+        "tema": respuesta_default
+    }
+    
+    if pregunta:
         try:
-          res= generate_text(pregunta, max_length=512)
-          # print(">> RES:",res)
-          inicio_json = res.find('{')
-          fin_json = res.rfind('}') + 1
-          json_str = res[inicio_json:fin_json]
-          json_obj = json.loads(json_str)
-          # print("json_obj:",json_obj)    
-          return json_obj["respuesta"], json_obj["fuente"], json_obj["tema"]
-        except:          
-          return json_obj["respuesta"], json_obj["fuente"], json_obj["tema"]    
+            res = generate_text(pregunta, max_length=512)
+            inicio_json = res.find('{')
+            fin_json = res.rfind('}') + 1
+            json_str = res[inicio_json:fin_json]
+            json_obj = json.loads(json_str)
+
+            # Verificar si el JSON contiene todas las claves necesarias
+            if all(key in json_obj for key in ["respuesta", "fuente", "tema"]):
+                return json_obj["respuesta"], json_obj["fuente"], json_obj["tema"]
+            else:
+                return res, respuesta_default, respuesta_default
+        except Exception as e:
+            print("Error al procesar la respuesta:", e)
+    
     return json_obj["respuesta"], json_obj["fuente"], json_obj["tema"]
 
 # Ejemplos de preguntas
@@ -136,7 +143,7 @@ iface = gr.Interface(
         gr.Textbox(label="Fuente", lines=1),
         gr.Textbox(label="Tema", lines=1)
     ],
-    title="Consulta Juridica basada en la Constitucion Politica del Peru",
+    title="HolaaaaaaaaaaaaaaaaaConsulta Juridica basada en la Constitucion Politica del Peru",
     description="Introduce tu pregunta sobre la Constitución Politica o una situación donde creas que tus derechos hayan sido vulnerados.",
     examples=ejemplos,
 )