Update app.py

app.py

@@ -35,17 +35,31 @@ def respond(user_input, history):
 def on_reset_button_click():
     llm_svc.store={}
 
-# Define the main block
-if __name__ == '__main__':
-    logging.info("Starting AIVIz Bot")
-
-    with gr.Blocks() as demo:
-        # Create the ChatInterface
-        chat_interface = gr.ChatInterface(fn=respond)
-        # Create a button and bind the reset function to it
-        reset_button = gr.Button("Reset Chat Memory Cache")
+
+demo = gr.ChatInterface(
+    fn=respond,
+    additional_inputs=[
+        gr.Button("Reset Chat Memory Cache", elem_id="reset_button")
+    ]
+)
+
+reset_button = demo.additional_inputs[0]
+reset_button.click(on_reset_button_click)
+
+if __name__ == "__main__":
+    demo.launch(share=True)
+
+# # Define the main block
+# if __name__ == '__main__':
+#     logging.info("Starting AIVIz Bot")
+
+#     with gr.Blocks() as demo:
+#         # Create the ChatInterface
+#         chat_interface = gr.ChatInterface(fn=respond)
+#         # Create a button and bind the reset function to it
+#         reset_button = gr.Button("Reset Chat Memory Cache")
         
-        reset_button.click(on_reset_button_click)
+#         reset_button.click(on_reset_button_click)
 
-    # Launch the interface with a shareable link
-    demo.launch(share=True)
+#     # Launch the interface with a shareable link
+#     demo.launch(share=True)