Update app.py

app.py

@@ -1,15 +1,17 @@
 import gradio as gr
-from temp import chat, reset, update_tokens, INPUT_TOTAL_TOKENS, OUTPUT_TOTAL_TOKENS
+from temp import MailChatBot
+
+MODEL = MailChatBot()
 
 with gr.Blocks(fill_height=True, fill_width=True) as demo:
-    gr.ChatInterface(chat)
+    gr.ChatInterface(MODEL.chat)
     with gr.Row():
-        input_tokens = gr.Textbox(value=INPUT_TOTAL_TOKENS, label='Input Tokens')
-        output_tokens = gr.Textbox(value=OUTPUT_TOTAL_TOKENS, label='Output Tokens')
+        input_tokens = gr.Textbox(value=0, label='Input Tokens')
+        output_tokens = gr.Textbox(value=0, label='Output Tokens')
         with gr.Column():
             btn = gr.Button("Reset")        
             count_tokens = gr.Button('Count Tokens')
-        btn.click(reset, outputs=[output_tokens, input_tokens])
-        count_tokens.click(update_tokens, outputs=[input_tokens, output_tokens])
+        btn.click(MODEL.reset, outputs=[output_tokens, input_tokens])
+        count_tokens.click(MODEL.get_tokens, outputs=[input_tokens, output_tokens])
 
 demo.launch(share=True)