Update app.py

app.py

@@ -50,9 +50,9 @@ chat_history = []
 def chat_output(inputs):
     output = chat_interface(inputs)
     answer = output.split('AI:')
-    if(len(answer) == 1):
-        chat_history.append((inputs, answer[0]))
-    chat_history.append((inputs, answer[0]))
+    if(len(answer) == 2):
+        chat_history.append((inputs, answer[1]))
+    chat_history.append((inputs, answer[1]))
     return chat_history
 
 template = """You are an AI having conversation with a human.