Update app.py

app.py

@@ -8,22 +8,17 @@ client = OpenAI()
 assistant_id=os.environ["OPENAI_ASSISTANT_ID"]
 
 def get_thread_id(streaming_chat_thread_id_state):
-    # if not streaming_chat_thread_id_state:
-    #   streaming_chat_thread_id_state = gr.State({"thread_id": None})
-    # print("======thread_id======")
-    # print(type(streaming_chat_thread_id_state))
-    # print(streaming_chat_thread_id_state)
-    # try:
-    #     thread_id = streaming_chat_thread_id_state["thread_id"]
-    #     print(thread_id)
-    # except TypeError as e:
-    #     thread_id = None        
-    thread_id = streaming_chat_thread_id
-    if not thread_id or thread_id == "Demo":
+    if not streaming_chat_thread_id_state:
+      streaming_chat_thread_id_state = gr.State({ "thread_id": None})
+    print("======thread_id======")
+    thread_id = streaming_chat_thread_id_state["thread_id"]
+    print(thread_id)
+
+    if not thread_id:
         thread = client.beta.threads.create()
         thread_id = thread.id
         print(f"new thread_id: {thread_id}")
-        # streaming_chat_thread_id_state["thread_id"] = thread_id
+        streaming_chat_thread_id_state["thread_id"] = thread_id
     else:
         print(f"old thread_id: {thread_id}")
     return thread_id
@@ -46,11 +41,11 @@ def openai_chat(user_message, thread_id):
             total += delta
             yield total
 
-def chat(user_message, history, streaming_chat_thread_id):
-  thread_id =get_thread_id(streaming_chat_thread_id)
+def chat(user_message, history, streaming_chat_thread_id_state):
+  thread_id =get_thread_id(streaming_chat_thread_id_state)
   yield from openai_chat(user_message, thread_id)
 
 history = []
-streaming_chat_thread_id = "Demo"# = gr.State({"thread_id": None})
-additional_inputs = [streaming_chat_thread_id]
+streaming_chat_thread_id_state = gr.State({ "thread_id": None})
+additional_inputs = [streaming_chat_thread_id_state]
 gr.ChatInterface(chat, examples=[["What are the troy rack mounting options?"], ["what are the steps for helen"], ["what are the main comparison between troy and janus specs"]], title="User Support Bot", additional_inputs=additional_inputs).launch(debug=True)