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=[]):
 
@@ -99,7 +99,7 @@ def chat(input, history=[]):
 
 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,9 +117,9 @@ def sqlquery(input, history=[]):
     return output
     '''
 
-    history.append((input, sql_response))
-    #conversation_history.append(("User", input))
-    #conversation_history.append(("Bot", 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 history])
@@ -154,10 +154,10 @@ sql_interface = gr.Interface(
     fn=sqlquery,
     theme="default",
     css=".footer {display:none !important}",
-    #inputs=gr.Textbox(prompt="You:"),
-    #outputs=gr.Textbox(),
-    inputs=["text", "state"],
-    outputs=["chatbot", "state"],
+    inputs=gr.Textbox(prompt="You:"),
+    outputs=gr.Textbox(),
+    #inputs=["text", "state"],
+    #outputs=["chatbot", "state"],
     #live=True,
     #capture_session=True,
     title="ST SQL Chat",