updated key

app.py

@@ -8,15 +8,6 @@ openai_api_key = st.secrets["openai_api_key"]
 client = openai.OpenAI(api_key = openai_api_key)
 model = "gpt-3.5-turbo-16k"
 
-# # ==  Create our Assistant
-personal_trainer_assis = client.beta.assistants.create(
-    name="Personal Trainer",
-    instructions="""You are the best personal trainer and nutritionist who knows how to get clients to build lean muscles.\n
-     You've trained high-caliber athletes and movie stars. """,
-    model=model,
-)
-asistant_id = personal_trainer_assis.id
-
 # === Thread an empty thread
 thread = client.beta.threads.create()
 thread_id = thread.id
@@ -30,7 +21,7 @@ def wait_for_run_completion(client, thread_id, run_id, sleep_interval=5):
     """
     while True:
         try:
-            run = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run_id)
+            run = client.beta.threads.runs.retrieve(thread_id = thread_id, run_id = run_id)
             if run.completed_at:
                 elapsed_time = run.completed_at - run.created_at
                 formatted_elapsed_time = time.strftime(
@@ -50,37 +41,61 @@ def wait_for_run_completion(client, thread_id, run_id, sleep_interval=5):
         logging.info("Waiting for run to complete...")
         time.sleep(sleep_interval)
 
+def create_new_assistant():
+    personal_trainer_assis = client.beta.assistants.create(
+        name="Personal Trainer",
+        instructions="""You are the best personal trainer and nutritionist who knows how to get clients to build lean muscles.\n
+        You've trained high-caliber athletes and movie stars. """,
+        model=model,
+    )
+    print(f"Created new assistant with ID: {personal_trainer_assis.id}")
+    return personal_trainer_assis.id
+
 def main():
     # Streamlit interface
-    st.title("Personal trainer")
+    st.title("Personal Trainer")
 
     # Note that API key's running out of budget
     contact_url = "https://www.linkedin.com/in/linhvuu"
-    st.write("I am running out of energy. Please contact [my assistant](%s) to wake me up." % contact_url)
-
+    st.write("If no result returns, it means I am running out of energy. Please contact [Linh Vuu](%s) to wake me up." % contact_url)
+    
     with st.form(key="user_input_form"):
+
         question = st.text_input("Enter question:")
         submit_button = st.form_submit_button(label="Response to my question above")
 
         if submit_button:
+
             # ==== Create a Message ====
             message = question
             message = client.beta.threads.messages.create(
-                thread_id=thread_id, role="user", content = message
+                thread_id = thread_id, role = "user", content = message
             )
 
+            # Define assistant_id before the try-except block
+            assistant_id = "asst_v5C5Qy4KGXTRHyty7n1wLLa6"
+            run = None
+
             # === Run our Assistant ===
-            run = client.beta.threads.runs.create(
-                thread_id=thread_id,
-                assistant_id=asistant_id
-            )
+            try:
+                if assistant_id is None:
+                    # If assistant_id is None, create a new assistant
+                    assistant_id = create_new_assistant()
+
+                run = client.beta.threads.runs.create(
+                    thread_id = thread_id,
+                    assistant_id = assistant_id
+                )
+            except openai.NotFoundError:
+                # If assistant_id does not exist, create a new assistant and retry
+                assistant_id = create_new_assistant()
+                run = client.beta.threads.runs.create(
+                    thread_id = thread_id,
+                    assistant_id = assistant_id
+                )
 
             # === Run ===
             wait_for_run_completion(client=client, thread_id=thread_id, run_id=run.id)
 
 if __name__ == "__main__":
-    main()
-
-# ==== Steps --- Logs ==
-# run_steps = client.beta.threads.runs.steps.list(thread_id=thread_id, run_id=run.id)
-# print(f"Steps---> {run_steps.data[0]}")
+    main()