Update app.py

app.py

@@ -99,17 +99,18 @@ def predict(input, history=[]):
 
 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")
+    #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)
 
-    global conversation_history
+    #global conversation_history
     
     # Maintain the conversation history
-    conversation_history.append(user_input)
+    conversation_history.append("User: " + input)
+    conversation_history.append("Bot: " + sql_response)
     
-    return sql_response
+    return conversation_history
 
 
 chat_interface = gr.Interface(