Update app.py

app.py

@@ -64,7 +64,7 @@ sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
 sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
 
 #sql_response = None
-conversation_history = []
+#conversation_history = []
 
 def chat(input, history=[]):
 
@@ -92,14 +92,14 @@ def chat(input, history=[]):
 
     # convert the tokens to text, and then split the responses into the right format
     response = tokenizer.decode(history[0]).split("<|endoftext|>")
-    #response = [(response[i], response[i + 1]) for i in range(0, len(response) - 1, 2)]  # convert to tuples of list
+    response = [(response[i], response[i + 1]) for i in range(0, len(response) - 1, 2)]  # convert to tuples of list
     
     return response, history
 
 
-def sqlquery(input):
+def sqlquery(input, history[]):
 
-    global conversation_history
+    #global conversation_history
     
     #input_text = " ".join(conversation_history) + " " + input
     sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
@@ -117,16 +117,16 @@ def sqlquery(input):
     return output
     '''
 
-    conversation_history.append((input, sql_response))
+    history.append((input, sql_response))
     #conversation_history.append(("User", input))
     #conversation_history.append(("Bot", sql_response))
 
     # Build conversation string
-    conversation = "\n".join([f"User: {user_msg}\nBot: {resp_msg}" for user_msg, resp_msg in conversation_history])
-    #conversation = "\n".join([f"{sender}: {msg}" for sender, msg in conversation_history])
+    #conversation = "\n".join([f"User: {user_msg}\nBot: {resp_msg}" for user_msg, resp_msg in history])
+    #conversation = "\n".join([f"{sender}: {msg}" for sender, msg in history])
     
     #return conversation
-    return conversation_history
+    return sql_response, history
 
     '''
     html = "<div class='chatbot'>"
@@ -154,9 +154,10 @@ sql_interface = gr.Interface(
     fn=sqlquery,
     theme="default",
     css=".footer {display:none !important}",
-    inputs=gr.Textbox(prompt="You:"),
+    #inputs=gr.Textbox(prompt="You:"),
     #outputs=gr.Textbox(),
-    outputs=["chatbot"],
+    inputs=["text", "state"],
+    outputs=["chatbot", "state"],
     #live=True,
     #capture_session=True,
     title="ST SQL Chat",