Upload 2 files

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import table_load as tl
-import comm
 
 
 css="""
@@ -35,7 +34,7 @@ with gr.Blocks(css=css) as demo:
     load_table.click(tl.load_file, None, langchain_status, queue=False)    
     load_table.click(tl.table_loader, inputs=[raw_table, openai_key], outputs=[langchain_status], queue=False)
    
-    question.submit(comm.respond, [question, chatbot], [question, chatbot])
+    question.submit(tl.respond, [question, chatbot], [question, chatbot])
     clr_btn.click(lambda: None, None, chatbot, queue=False)
     
 demo.launch()

table_load.py

@@ -1,4 +1,4 @@
-"""functions related to tables loading"""
+"""functions"""
 
 def load_file():
     return "Loading..."
@@ -33,3 +33,20 @@ def table_loader(table_file, open_ai_key):
     else:
         return "Wrong file format! Upload excel file or csv!"
     
+def run(query):
+    from langchain.callbacks import get_openai_callback
+    
+    with get_openai_callback() as cb:
+        response = (agent.run(query))
+        costs = (f"Total Cost (USD): ${cb.total_cost}")
+        output = f'{response} \n {costs}'
+        return output
+    
+def respond(message, chat_history):
+    import time
+    
+    bot_message = run(message)
+    chat_history.append((message, bot_message))
+    time.sleep(0.5)
+    return "", chat_history
+