vaishnav commited on
Commit
3a0580c
·
1 Parent(s): 3ce8148

update app.py and requirements.py

Browse files
Files changed (2) hide show
  1. app.py +12 -7
  2. requirements.txt +0 -0
app.py CHANGED
@@ -36,17 +36,22 @@ def on_reset_button_click():
36
  llm_svc.store={}
37
 
38
 
39
- # Define the main block
40
  if __name__ == '__main__':
41
  logging.info("Starting AIVIz Bot")
42
 
43
  with gr.Blocks() as demo:
44
- # Create the ChatInterface
45
- chat_interface = gr.ChatInterface(fn=respond)
46
- # Create a button and bind the reset function to it
47
- reset_button = gr.Button("Reset Chat Memory Cache")
 
48
 
49
- reset_button.click(on_reset_button_click)
 
 
 
 
 
50
 
51
- # Launch the interface with a shareable link
52
  demo.launch(share=True)
 
36
  llm_svc.store={}
37
 
38
 
 
39
  if __name__ == '__main__':
40
  logging.info("Starting AIVIz Bot")
41
 
42
  with gr.Blocks() as demo:
43
+ gr.Markdown("# 🚢 AIVIz Bot - Vessel Trajectory Prediction")
44
+ gr.Markdown("Welcome! Ask me anything about vessel tracking, AI models.")
45
+
46
+ with gr.Row():
47
+ chat_interface = gr.ChatInterface(fn=respond)
48
 
49
+ with gr.Row():
50
+ reset_button = gr.Button("🔄 Reset Chat Memory Cache")
51
+ reset_status = gr.Textbox(label="Status", interactive=False)
52
+
53
+ # Bind reset button to function
54
+ reset_button.click(fn=on_reset_button_click, outputs=reset_status)
55
 
56
+ # Launch the interface
57
  demo.launch(share=True)
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ