Update app.py

app.py

@@ -4,29 +4,39 @@ from dotenv import load_dotenv
 import os
 import time
 
+# Global variable to store the current thread ID
+current_thread_id = None
+
 examples = [
-    ["My Eucalyptus tree is struggling outside in the cold weather in europe"],
-    ["My callatea house plant is yellowing."],
-    ["We have a catcus as work that suddently started yellowing and wilting."]
+    ["My Eucalyptus tree is struggling outside in the cold weather in Europe"],
+    ["My calathea house plant is yellowing."],
+    ["We have a cactus at work that suddenly started yellowing and wilting."]
 ]
 
 load_dotenv()
 openai.api_key = os.getenv('OPENAI_API_KEY')
-assistant_id=os.getenv('ASSISTANT_ID')
+assistant_id = os.getenv('ASSISTANT_ID')
 client = openai.OpenAI(api_key=openai.api_key)
 
 def ask_openai(question):
+    global current_thread_id
+
     try:
-        thread = client.beta.threads.create()
+
+        if current_thread_id is None:
+            thread = client.beta.threads.create()
+            current_thread_id = thread.id
+        else:
+            thread = client.beta.threads.retrieve(thread_id=current_thread_id)
 
         client.beta.threads.messages.create(
-            thread_id=thread.id,
+            thread_id=current_thread_id,
             role="user",
             content=question,
         )
 
         run = client.beta.threads.runs.create(
-            thread_id=thread.id,
+            thread_id=current_thread_id,
             assistant_id=assistant_id
         )
 
@@ -36,19 +46,19 @@ def ask_openai(question):
 
         while not response_received and time.time() - start_time < timeout:
             run_status = client.beta.threads.runs.retrieve(
-                thread_id=thread.id,
+                thread_id=current_thread_id,
                 run_id=run.id,
             )
             if run_status.status == 'completed':
                 response_received = True
             else:
-                time.sleep(4)  
+                time.sleep(4)
 
         if not response_received:
             return "Response timed out."
 
         steps = client.beta.threads.runs.steps.list(
-            thread_id=thread.id,
+            thread_id=current_thread_id,
             run_id=run.id
         )
 
@@ -57,13 +67,13 @@ def ask_openai(question):
             if last_step.type == 'message_creation':
                 message_id = last_step.step_details.message_creation.message_id
                 message = client.beta.threads.messages.retrieve(
-                    thread_id=thread.id,
+                    thread_id=current_thread_id,
                     message_id=message_id
                 )
                 if message.content and message.content[0].type == 'text':
                     return message.content[0].text.value
         return "No response."
-        
+
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
@@ -71,9 +81,9 @@ iface = gr.Interface(
     fn=ask_openai,
     inputs=gr.Textbox(lines=5, placeholder="Hi there, I have a plant that's..."),
     outputs=gr.Markdown(),
-    title="Wecome to Tonic's Bulbi Plant Doctor",
+    title="Welcome to Tonic's Bulbi Plant Doctor",
     description="""Introduce your plant below. Be as descriptive as possible. Respond with additional information when prompted. Save your plants with Bulbi Plant Doctor""",
     examples=examples
 )
 
-iface.launch()
+iface.launch()