Update app.py

app.py

@@ -8,10 +8,19 @@ OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
 USER=os.getenv('USER')
 PASS=os.getenv('PASS')
 
+template = """You are a helpful assistant to answer all user queries.
+{history}
+User: {message}
+Chatbot:"""
+
+prompt = PromptTemplate(
+    input_variables=["history", "message"], template=template
+)
 
 def predict(message, history):
     history_openai_format = []
     for human, assistant in history:
+        history_openai_format.append({"role": "system", "content": prompt })
         history_openai_format.append({"role": "user", "content": human })
         history_openai_format.append({"role": "assistant", "content":assistant})
     history_openai_format.append({"role": "user", "content": message})