Update app.py

app.py

@@ -98,23 +98,35 @@ def chat(input, history=[]):
 
 
 def sqlquery(input):
+
+    history = gr.get_state() or []
     
     #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 + "\n")
     conversation_history.append("Bot: " + " ".join(sql_response) + "\n" )
 
     output = " ".join(conversation_history)
     return output
-    '''    
-    return sql_response
+    '''
+
+    history.append((input, sql_response))
+    gr.set_state(history)
+    html = "<div class='chatbot'>"
+    for user_msg, resp_msg in history:
+        html += f"<div class='user_msg'>{user_msg}</div>"
+        html += f"<div class='resp_msg'>{resp_msg}</div>"
+    html += "</div>"
+    return html
+    
+    #return sql_response
 
 
 chat_interface = gr.Interface(
@@ -127,6 +139,7 @@ chat_interface = gr.Interface(
     description="Type your message in the box above, and the chatbot will respond.",
 )
 
+"""
 sql_interface = gr.Interface(
     fn=sqlquery,
     theme="default",
@@ -137,6 +150,14 @@ sql_interface = gr.Interface(
     title="ST SQL Chat",
     description="Type your message in the box above, and the chatbot will respond.",
 )
+"""
+
+iface = gr.Interface(chat, "text", "html", css="""
+    .chatbox {display:flex;flex-direction:column}
+    .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
+    .user_msg {background-color:cornflowerblue;color:white;align-self:start}
+    .resp_msg {background-color:lightgray;align-self:self-end}
+""", allow_screenshot=False, allow_flagging=False)
 
 combine_interface = gr.TabbedInterface(
     interface_list=[
@@ -147,4 +168,5 @@ combine_interface = gr.TabbedInterface(
 )
 
 if __name__ == '__main__':
-    combine_interface.launch()
+    #combine_interface.launch()
+    iface.launch(debug=True)