Update chatbot.py

chatbot.py

@@ -115,26 +115,26 @@ Do you usually discuss your problems and concerns with your mother or a mother-l
 """},
 ]
 
-def chatbot(user_id, input):
-    if not hasattr(chatbot, "messages"):
-        chatbot.messages = initial_messages.copy()
+def chatbot(user_id, input, state):
+    if state is None:
+        state = initial_messages.copy()
     
     if input:
-        chatbot.messages.append({"role": "user", "content": input})
+        state.append({"role": "user", "content": input})
         response = client.chat.completions.create(
             model="gpt-3.5-turbo",
-            messages=chatbot.messages
+            messages=state
         )
         reply = response.choices[0].message["content"]
-        chatbot.messages.append({"role": "assistant", "content": reply})
+        state.append({"role": "assistant", "content": reply})
         
         conversation = ""
-        for message in chatbot.messages[2:]:
+        for message in state[2:]:
             role = "You" if message["role"] == "user" else "AttachmentBot"
             conversation += f"{role}: {message['content']}\n"
 
         store_transcript(user_id, conversation)
-        return conversation
+        return conversation, state
 
 with gr.Blocks() as demo:
     username = gr.Textbox(label="Username")
@@ -145,11 +145,12 @@ with gr.Blocks() as demo:
 
     chat_input = gr.Textbox(lines=7, label="Chat with AttachmentBot")
     chat_output = gr.Textbox(label="Conversation")
+    state = gr.State()
 
     chat_interface = gr.Interface(
         fn=chatbot,
-        inputs=[user_id_state, chat_input],
-        outputs=chat_output,
+        inputs=[user_id_state, 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'.",
         visible=False  # Initially hidden, shown after login