Update app.py

app.py

@@ -25,11 +25,14 @@ def construct_index(directory_path):
     index.storage_context.persist(persist_dir="index.json")
     return index
 
+
+def chatbot(input_text):
+    response = query_engine.query(input_text)
+    return response.response
+
 def chatbot(input_text):
     # load the index from disk
-    storage_context = StorageContext.from_defaults(persist_dir="index.json")
-    # load index
-    index = load_index_from_storage(storage_context)
+    query_engine = index.as_query_engine()
     # define custom Prompt
     TEMPLATE_STR = (
         "Quiero que actues como un asistente personal de un cliente del Banco Galicia. Tu nombre es Gala. Me brindas información sobre mi resument de tarjeta de credito VISA. Si la respuesta no esta en el documento, respondeme de forma creativa que no lo sabes, pero que podes ayudarme con otra pregunta. Nunca te enojes y no contestes preguntas politicas o religiosas. \n"
@@ -38,7 +41,7 @@ def chatbot(input_text):
     )
     QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
     # query the index and get the response
-    response = index.query(QA_TEMPLATE)
+    response = query_engine.query(input_text)
     return response.response
 
 iface = gr.Interface(fn=chatbot,