Update app.py

app.py

@@ -97,15 +97,25 @@ prompt_template = PromptTemplate.from_template('''system role :{context} \
         assistance: 
         ''')
 
+import random
+import time
+
 # 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,
-)
+with gr.Blocks() as chat:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+
+    def respond(message, chat_history):
+        # Use the global conversation variable here
+        response = conversation.run(message)
+        chat_history.append((message, response))
+        return response, chat_history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
 # Launch Gradio Interface
-iface.launch()
+if __name__ == "__main__":
+    chat.launch()
 
 # gr.load("models/ksh-nyp/llama-2-7b-chat-TCMKB").launch()