Update app.py

app.py

@@ -16,6 +16,7 @@ import gradio as gr
 from gradio import components
 import textwrap
 import datetime
+from llama_index.core import Settings
 
 openai.api_key = os.environ.get('openai_key')
 os.environ["OPENAI_API_KEY"] = os.environ.get('openai_key')
@@ -57,17 +58,20 @@ def responder(pregunta):
     if len(prompt_question.split()) > max_tokens:
         prompt_question = ' '.join(prompt_question.split()[:max_tokens])
 
-    # Respuesta con prompt sumado a la pregunta
-    respuesta = query_engine.query(prompt_question)
-
-    # Muestra la cadena que se va formando palabra por palabra
-    partial_message = ""
-    for chunk in respuesta.response_gen:
-        partial_message += chunk
-        yield partial_message
-
+    try:
+        # Respuesta con prompt sumado a la pregunta
+        respuesta = query_engine.query(prompt_question)
 
+        # Muestra la cadena que se va formando palabra por palabra
+        partial_message = ""
+        for chunk in respuesta.response_gen:
+            partial_message += chunk
+            yield partial_message
+    except Exception as e:
+        # Si hay algún error, devolver un mensaje de error
+        yield "Error, inténtelo nuevamente!"
 
+        
 # Definir la interfaz de usuario con Gradio
 with gr.Blocks(theme='sudeepshouche/minimalist') as demo: