Update chatbot.py

chatbot.py

@@ -115,62 +115,61 @@ Do you usually discuss your problems and concerns with your mother or a mother-l
 """},
 ]
 
-def chatbot(user_id, input, state):
-    if state is None:
-        state = initial_messages.copy()
+def chatbot(input, state):
+    user_id = state[0]
+    messages = state[1]
     
     if input:
-        state.append({"role": "user", "content": input})
+        messages.append({"role": "user", "content": input})
         response = client.chat.completions.create(
             model="gpt-3.5-turbo",
-            messages=state
+            messages=messages
         )
         reply = response.choices[0].message["content"]
-        state.append({"role": "assistant", "content": reply})
+        messages.append({"role": "assistant", "content": reply})
         
         conversation = ""
-        for message in state[2:]:
+        for message in messages[2:]:
             role = "You" if message["role"] == "user" else "AttachmentBot"
             conversation += f"{role}: {message['content']}\n"
 
         store_transcript(user_id, conversation)
-        return conversation, 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")
-    user_id_state = gr.State()
     auth_message = gr.Textbox(visible=False)
 
     chat_input = gr.Textbox(lines=7, label="Chat with AttachmentBot", visible=False)
-    chat_output = gr.Textbox(label="Conversation")
-    state = gr.State()
+    chat_output = gr.Textbox(label="Conversation", visible=False)
+    state = gr.State([None, initial_messages.copy()])
 
     def login(username, password):
         user_id = authenticate(username, password)
         if user_id:
-            return gr.update(visible=True), user_id, ""
+            return gr.update(visible=True), gr.update(visible=True), [user_id, initial_messages.copy()], ""
         else:
-            return gr.update(visible=False), None, "Invalid credentials"
+            return gr.update(visible=False), gr.update(visible=False), [None, initial_messages.copy()], "Invalid credentials"
 
     def register(username, password):
         user_id = register_user(username, password)
         if user_id:
-            return gr.update(visible=True), user_id, "Registration successful, you can now login."
+            return gr.update(visible=True), gr.update(visible=True), [user_id, initial_messages.copy()], "Registration successful, you can now login."
         else:
-            return gr.update(visible=False), None, "Registration failed, try a different username."
+            return gr.update(visible=False), gr.update(visible=False), [None, initial_messages.copy()], "Registration failed, try a different username."
 
-    login_button.click(login, inputs=[username, password], outputs=[chat_input, user_id_state, auth_message])
-    register_button.click(register, inputs=[username, password], outputs=[chat_input, user_id_state, auth_message])
+    login_button.click(login, inputs=[username, password], outputs=[chat_input, chat_output, state, auth_message])
+    register_button.click(register, inputs=[username, password], outputs=[chat_input, chat_output, state, auth_message])
 
     chat_interface = gr.Interface(
         fn=chatbot,
-        inputs=[user_id_state, chat_input, state],
+        inputs=[chat_input, state],
         outputs=[chat_output, state],
         title="AttachmentBot",
-        description="Let me survey you about your attachment with certain people in your life. To begin, enter 'start'.",
+        description="Let me survey you about your attachment with certain people in your life. To begin, enter 'start'."
     )
 
     demo.launch()