Update chatbot.py

chatbot.py

@@ -68,8 +68,8 @@ def demo_miny_memory():
     return memory
 
 def demo_chain(input_text,history):
-    #PROMPT = ChatPromptTemplate.from_template(template)
-    PROMPT = PromptTemplate(template=template, input_variables=["history", "input"])
+    PROMPT = ChatPromptTemplate.from_template(template)
+    #PROMPT = PromptTemplate(template=template, input_variables=["history", "input"])
     conversation = ConversationChain(
         prompt=PROMPT,
         llm=llm,
@@ -82,4 +82,4 @@ def demo_chain(input_text,history):
         "input" : input_text,
         "history" : history
     })
-    return chat_reply['response'].split('AI:')[-1]
+    return chat_reply['response'] #.split('AI:')[-1]