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=[]):
 
@@ -97,35 +97,24 @@ def chat(input, history=[]):
     return response, history
 
 
-def sqlquery(input, history=[]):
+def sqlquery(input):
     
     #input_text = " ".join(conversation_history) + " " + input
     sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
-    #sql_outputs = sql_model.generate(**sql_encoding)
-    #sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
+    sql_outputs = sql_model.generate(**sql_encoding)
+    sql_response = sql_tokenizer.batch_decode(sql_outputs, skip_special_tokens=True)
 
-    sql_input_ids = sql_encoding["input_ids"]
-     # append the new user input tokens to the chat history
-    bot_input_ids = torch.cat([torch.LongTensor(history),  sql_input_ids], dim=-1)
-      
-    # generate a response
-    history = sql_model.generate(bot_input_ids, max_length=1000, pad_token_id=sql_tokenizer.eos_token_id).tolist()
-
-     # convert the tokens to text, and then split the responses into the right format
-    response = sql_tokenizer.batch_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
     
-    '''
     global conversation_history
     
     # Maintain the conversation history
-    conversation_history.append("User: " + input)
-    conversation_history.append("Bot: " + " ".join(sql_response) )
+    conversation_history.append("User: " + input + "<|endoftext|>")
+    conversation_history.append("Bot: " + " ".join(sql_response) + "<|endoftext|>" )
 
     output = " ".join(conversation_history)
     return output
-    '''
-    return response, history
+    
+    #return sql_response
 
 
 chat_interface = gr.Interface(
@@ -141,10 +130,8 @@ chat_interface = gr.Interface(
 sql_interface = gr.Interface(
     fn=sqlquery,
     theme="default",
-    #inputs=gr.Textbox(prompt="You:"),
-    #outputs=gr.Textbox(),
-    inputs=["text", "state"],
-    outputs=["chatbot", "state"],
+    inputs=gr.Textbox(prompt="You:"),
+    outputs=gr.Textbox(),
     live=True,
     capture_session=True,
     title="ST SQL Chat",