Update chatbot.py

chatbot.py

@@ -3,6 +3,7 @@ import gradio as gr
 import os
 import psycopg2
 from urllib.parse import urlparse
+import json
 
 # Initialize OpenAI client
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -118,7 +119,7 @@ def store_survey_responses(user_id, responses):
         INSERT INTO survey_responses (user_id, responses)
         VALUES (%s, %s);
         '''
-        cur.execute(insert_query, (user_id, responses))
+        cur.execute(insert_query, (user_id, json.dumps(responses)))  # Convert dictionary to JSON string
         conn.commit()
         cur.close()
         conn.close()
@@ -176,18 +177,22 @@ def chatbot(input, state):
         return conversation, [user_id, messages]
 
 with gr.Blocks() as demo:
-    username = gr.Textbox(label="Username")
-    password = gr.Textbox(label="Password", type="password")
-    login_button = gr.Button("Login")
-    register_button = gr.Button("Register")
-    auth_message = gr.Textbox(visible=False)
-
-    chat_input = gr.Textbox(lines=7, label="Chat with AttachmentBot", visible=False)
-    chat_output = gr.Textbox(label="Conversation", visible=False)
-    state = gr.State([None, initial_messages.copy()])
-
-    survey_inputs = create_survey_inputs()
-    submit_survey = gr.Button("Submit Survey")
+    with gr.Row():
+        with gr.Column():
+            username = gr.Textbox(label="Username")
+            password = gr.Textbox(label="Password", type="password")
+            login_button = gr.Button("Login")
+            register_button = gr.Button("Register")
+            auth_message = gr.Textbox(visible=False)
+        
+        with gr.Column():
+            survey_inputs = create_survey_inputs()
+            submit_survey = gr.Button("Submit Survey")
+        
+        with gr.Column():
+            chat_input = gr.Textbox(lines=7, label="Chat with AttachmentBot", visible=False)
+            chat_output = gr.Textbox(label="Conversation", visible=False)
+            state = gr.State([None, initial_messages.copy()])
 
     def login(username, password):
         user_id = authenticate(username, password)