Update app.py

app.py

@@ -10,7 +10,14 @@ OPEN_AI_MODEL = "gpt-4-1106-preview"
 
 # thread = gr.State(client.beta.threads.create())
 
-
+def initialize_thread():
+    session = gr.interface.get_session()
+    if "thread_id" not in session:
+        thread = client.beta.threads.create()
+        session["thread_id"] = thread.id
+        session["thread"] = thread
+    return session["thread_id"], session["thread"]
+    
 def wait_on_run(run, thread):
     while run.status == "queued" or run.status == "in_progress":
         run = client.beta.threads.runs.retrieve(
@@ -101,14 +108,17 @@ 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 
+  # thread = client.beta.threads.create()
+  # thread_id = thread.id 
+  thread_id = initialize_thread()
+  session = gr.interface.get_session()
+  print(thread_id)
   if not message and not code_written:
-    ai_message = opening_statement(thread_id, question_text, input, output, examples, thread)
+    ai_message = opening_statement(thread_id[0], question_text, input, output, examples, thread_id[1])
   if not code_written:
-    ai_message = get_response_for_case_t0(thread_id, message, question_text, input, output, examples, thread)
+    ai_message = get_response_for_case_t0(thread_id[0], message, question_text, input, output, examples, thread_id[1])
   else:
-    ai_message = response_evaluation_for_case_tx(thread_id, message, question_text, input, output, examples, code_written, thread)
+    ai_message = response_evaluation_for_case_tx(thread_id[0], message, question_text, input, output, examples, code_written, thread_id[1])
   print({"question_text":question_text, "input":input, "output":output, "examples":examples, 
          "user_code":code_written, "query":message, "ai_message":ai_message})
   return ai_message