Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import openai
 import streamlit as st
 import os
@@ -11,8 +12,7 @@ class OpenAIClient:
         self.client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         self.thread_id = None
         self.file_id = ["file-Anvyhz2k0puV0NCjWH8mGsuk"]
-        self.assistant_id = "asst_Z7pj5QGUjiAlVpJzqsX2SJkg"
-        self.run_id = "run_A5wq14ZMQXBpaDnULFPky5Bb"
+        self.assistant_id = "asst_jQN8lzPx9NMK79et6kuJ27II
 
     def create_thread_id(self):
         if "thread_id" not in st.session_state:
@@ -25,22 +25,13 @@ class OpenAIClient:
             role="user",
             content=user_message,
         )
-        _run = self.client.beta.threads.runs.create(thread_id=st.session_state.get("thread_id"), assistant_id=self.assistant_id)
-        while True:
-            run = self.client.beta.threads.runs.retrieve(thread_id=st.session_state.get("thread_id"),
-                                                        run_id=_run.id).status
+        run = self.client.beta.threads.runs.create(thread_id=st.session_state.get("thread_id"), assistant_id=self.assistant_id).id
+        while run == "queued" or run == "progress":
+            run = self.client.beta.threads.runs.retrieve(thread_id=st.session_state.get("thread_id"),run_id=run).status
+            time.sleep(0.5)
 
-            time.sleep(3)
-            if run == "completed":
-                return self.get_response()
-            elif run == "queued":
-                continue
-            elif run == "failed" or run == "cancelled" or run == "expired":
-                return run
-            elif run == "in_progress":
-                continue
-            else:
-                break
+            
+                
 
     def get_response(self):
         messages = self.client.beta.threads.messages.list(