Update app.py

app.py

@@ -1,6 +1,11 @@
 from llama_index import Prompt, SimpleDirectoryReader, LLMPredictor, PromptHelper, StorageContext, ServiceContext, GPTVectorStoreIndex, load_index_from_storage
 from langchain.chat_models import ChatOpenAI
-from langchain.prompts import StringPromptTemplate
+from langchain.prompts import (
+    ChatPromptTemplate,
+    HumanMessagePromptTemplate,
+    MessagesPlaceholder,
+    SystemMessagePromptTemplate,
+)
 import gradio as gr
 import sys
 import os
@@ -29,17 +34,13 @@ def chatbot(input_text):
     # load the index from disk
     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"
-    #    "Dado esto, por favor responde a la pregunta: {input_text}\n"
-    #)
-    #QA_TEMPLATE = PromptTemplate(TEMPLATE_STR)
-    # query the index and get the response
-    # Instantiation using initializer
-    prompt = """\
-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. Dado esto, por favor responde a la pregunta: {input_text}")
-"""
-    #prompt = PromptTemplate(input_variables=input_text, template="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. Dado esto, por favor responde a la pregunta: {input_text}")
+    prompt = ChatPromptTemplate(
+    messages=[
+        SystemMessagePromptTemplate.from_template(
+            "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. Dado esto, por favor responde a todas las preguntas."
+        ),
+        HumanMessagePromptTemplate.from_template("{input_text}"),
+    ])
     response = query_engine.query(prompt)
     return response.response