Update app.py

app.py

@@ -9,14 +9,9 @@ client = OpenAI(api_key=OPENAI_API_KEY)
 OPEN_AI_MODEL = "gpt-4-1106-preview"
 
 # thread = gr.State(client.beta.threads.create())
-thread_id = None
-
-def initialize_thread():
-    global thread_id
-    if thread_id is None:
-        thread = client.beta.threads.create()
-        thread_id = thread.id
-    return thread_id, thread
+# thread_id = None
+
+
     
 def wait_on_run(run, thread):
     while run.status == "queued" or run.status == "in_progress":
@@ -108,17 +103,20 @@ def response_evaluation_for_case_tx(thread_id, query, question_text, input, outp
 
 
 def run_chat_in_all_cases(message, history, question_text,input, output, examples, code_written):
-  # thread = client.beta.threads.create()
-  # thread_id = thread.id 
-  global thread_id
-  thread_id = initialize_thread()
+  if "thread_id" in gr.get_session():
+      pass 
+  else:
+      thread = client.beta.threads.create()
+      thread_id = thread.id
+      gr.save_to_session(thread_id=thread_id)
+      gr.save_to_session(thread=thread)
   print(thread_id)
   if not message and not code_written:
-    ai_message = opening_statement(thread_id[0], question_text, input, output, examples, thread_id[1])
+    ai_message = opening_statement(thread_id, question_text, input, output, examples, thread)
   if not code_written:
-    ai_message = get_response_for_case_t0(thread_id[0], message, question_text, input, output, examples, thread_id[1])
+    ai_message = get_response_for_case_t0(thread_id, message, question_text, input, output, examples, thread)
   else:
-    ai_message = response_evaluation_for_case_tx(thread_id[0], message, question_text, input, output, examples, code_written, thread_id[1])
+    ai_message = response_evaluation_for_case_tx(thread_id, message, question_text, input, output, examples, code_written, thread)
   print({"question_text":question_text, "input":input, "output":output, "examples":examples, 
          "user_code":code_written, "query":message, "ai_message":ai_message})
   return ai_message