Update app.py

app.py

@@ -96,11 +96,16 @@ prompt_template = PromptTemplate.from_template('''system role :{context} \
         ''')
 
 # Define Gradio Interface
+# iface = gr.Interface(
+#     fn=lambda query: conversation.run(prompt_template.format(context=context, query=query)),
+#     inputs=gr.Textbox(),
+#     outputs=gr.Textbox(),
+#     live=True,
+# )
 iface = gr.Interface(
-    fn=lambda query: conversation.run(prompt_template.format(context=context, query=query)),
+    fn=lambda query, chat_history: (conversation.run(prompt_template.format(context=context, query=query), chat_history), chat_history),
     inputs=gr.Textbox(),
-    outputs=gr.Textbox(),
-    live=True,
+    outputs=gr.Chat(live=True, persistent=True),
 )
 
 # Launch Gradio Interface