updated code

app.py

@@ -56,41 +56,47 @@ def get_news(topic):
         print("Error occured during API Request", e)
 
 
-class AssistantManager:
+class AssistantManager ():
     thread_id = None
     assistant_id = None
     
-    def __init__(self, model: str = model):
+    def __init__(self, model: str = model, assistant_id  = None, thread_id = None):
         self.client = client
         self.model = model
-        self.assistant = None
-        self.thread = None
         self.run = None
         self.summary = None
+        self.assistant_id = assistant_id
+        self.thread = thread_id
 
-        # Retrieve existing assistant and thread if IDs are already set
-        if AssistantManager.assistant_id:
-            self.assistant = self.client.beta.assistants.retrieve(
-                assistant_id=AssistantManager.assistant_id
-            )
-        if AssistantManager.thread_id:
+        # Retrieve existing thread if IDs are already set          
+        if self.thread_id:
             self.thread = self.client.beta.threads.retrieve(
-                thread_id=AssistantManager.thread_id
+                thread_id=self.thread_id
             )
 
     def create_assistant(self, name, instructions, tools):
-        if not self.assistant:
+
+        # Retrieve existing assistant and thread if IDs are already set
+        try:
+            self.assistant = self.client.beta.assistants.retrieve(
+                assistant_id=self.assistant_id
+            )
+        except openai.NotFoundError:
+            # If the assistant doesn't exist, create a new one
             assistant_obj = self.client.beta.assistants.create(
                 name=name, instructions=instructions, tools=tools, model=self.model
             )
-            AssistantManager.assistant_id = assistant_obj.id
+            self.assistant_id = assistant_obj.id
             self.assistant = assistant_obj
-            print(f"AssisID:::: {self.assistant.id}")
+            print(f"Created new assistant with ID: {self.assistant.id}")
+        except Exception as e:
+            # If any other error occurs, raise the error
+            raise e
 
     def create_thread(self):
         if not self.thread:
             thread_obj = self.client.beta.threads.create()
-            AssistantManager.thread_id = thread_obj.id
+            self.thread_id = thread_obj.id
             self.thread = thread_obj
             print(f"ThreadID::: {self.thread.id}")
 
@@ -121,11 +127,6 @@ class AssistantManager:
             self.summary = "\n".join(summary)
             print(f"SUMMARY-----> {role.capitalize()}: ==> {response}")
 
-            # for msg in messages:
-            #     role = msg.role
-            #     content = msg.content[0].text.value
-            #     print(f"SUMMARY-----> {role.capitalize()}: ==> {content}")
-
     def call_required_functions(self, required_actions):
         if not self.run:
             return
@@ -183,17 +184,17 @@ class AssistantManager:
 
 
 def main():
-    news = get_news("bitcoin")
-    print(news[0])
-
-    manager = AssistantManager()
 
+    manager = AssistantManager(assistant_id = 'asst_7adqkvmsU2ovkQsPiBFbh5yV')
+    # asst_lGnxol2MqmLr4x6OA4xJYXAq me
+    # asst_7adqkvmsU2ovkQsPiBFbh5yV Charles
+    
     # Streamlit interface
     st.title("News Summariser")
 
     # 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"):
         instructions = st.text_input("Enter topic:")