Update app.py

app.py

@@ -49,7 +49,10 @@ chat_history = []
 
 def chat_output(inputs):
     output = chat_interface(inputs)
-    chat_history.append((inputs, output))
+    answer = output.split('AI:')
+    if(len(answer) == 1):
+        chat_history.append((inputs, answer[0]))
+    chat_history.append((inputs, answer[0]))
     return chat_history
 
 template = """You are an AI having conversation with a human. 
@@ -74,7 +77,7 @@ with gr.Blocks() as demo:
     textbox_component = gr.Textbox(placeholder="Can I help you to book a hotel?", container=False, label = "input", scale=7)
 
     demo.chatbot_interface = gr.Interface(
-        fn=chat_interface, 
+        fn=chat_output, 
         inputs=[
             textbox_component
         ],