Update app.py

app.py

@@ -64,6 +64,7 @@ sql_tokenizer = TapexTokenizer.from_pretrained(sql_model_name)
 sql_model = BartForConditionalGeneration.from_pretrained(sql_model_name)
 
 #sql_response = None
+conversation_history = []
 
 def predict(input, history=[]):
 
@@ -96,12 +97,18 @@ def predict(input, history=[]):
     return response, history
 
 
-def sqlquery(input):
-
-    sql_encoding = sql_tokenizer(table=table, query=input + sql_tokenizer.eos_token, return_tensors="pt")
+def sqlquery(input, conversation_history):
+    
+    input_text = " ".join(conversation_history) + " " + input
+    sql_encoding = sql_tokenizer(table=table, query=input_text + 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)
 
+    global conversation_history
+    
+    # Maintain the conversation history
+    conversation_history.append(user_input)
+    
     return sql_response
 
 
@@ -118,7 +125,8 @@ chat_interface = gr.Interface(
 sql_interface = gr.Interface(
     fn=sqlquery,
     theme="default",
-    inputs=gr.Textbox(prompt="You:"),
+    #inputs=gr.Textbox(prompt="You:"),
+    inputs=["text", "state"],
     outputs=gr.Textbox(),
     live=True,
     capture_session=True,