Update app.py

app.py

@@ -1,34 +1,54 @@
 import gradio as gr
 import openai
 from dotenv import load_dotenv
-from gradio_client import Client
 import os
 import time
 
+current_thread_id = None
+
+title = "# Welcome to 🙋🏻‍♂️Tonic's🕵🏻‍♂️Bulbi🪴Plant👩🏻‍⚕️Doctor!"
+description = """Here you can use Bulbi - an OpenAI agent that helps you save your plants! 
+OpenAI doesnt let you use Agents without paying for it, so I made you an interface you can use for free !
+### How to use:
+- Introduce your🌵plant below. 
+- Be as🌿descriptive as possible. 
+- **Respond with additional🗣️information when prompted.**
+- Save your plants with👨🏻‍⚕️Bulbi Plant Doctor! 
+### Join us:
+[Join my active builders' server on discord](https://discord.gg/VqTxc76K3u). Let's build together!
+Big thanks to 🤗Huggingface Organisation for the🫂Community Grant"""
+
 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",True, None],
+    ["My calathea house plant is yellowing.",True, None],
+    ["We have a cactus at work that suddenly started yellowing and wilting.",True, None]
 ]
 
 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)
+thread_ids = {}
 
+def ask_openai(question, start_new_thread=True, selected_thread_id=None):
+    global thread_ids
 
-def ask_openai(question):
     try:
-        thread = client.beta.threads.create()
+        if start_new_thread or selected_thread_id not in thread_ids:
+            thread = client.beta.threads.create()
+            current_thread_id = thread.id
+            thread_ids[current_thread_id] = thread.id
+        else:
+            current_thread_id = thread_ids[selected_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
         )
 
@@ -38,36 +58,41 @@ 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."
 
-        # Ensure steps is defined here
         steps = client.beta.threads.runs.steps.list(
-            thread_id=thread.id,
+            thread_id=current_thread_id,
             run_id=run.id
         )
 
-        response_text = "No response."
         if steps.data:
             last_step = steps.data[-1]
             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':
                     response_text = message.content[0].text.value
+                else:
+                    return "No response."
+        else:
+            return "No response."
+
+        # Initialize Gradio client
+        gradio_client = Client("https://tonic1-tulu.hf.space/--replicas/t5vxm/")
 
-        gradio_client = Client("https://tonic1-tulu.hf.space/--replicas/gzzjr/")
+        # Pass the response text to the Gradio client and get the prediction
         final_result = gradio_client.predict(
             response_text,  
             "I am Tulu, an Expert Plant Doctor, I will exactly summarize the information you provide to me.",
@@ -80,12 +105,16 @@ def ask_openai(question):
         return f"An error occurred: {str(e)}"
 
 iface = gr.Interface(
-    fn=ask_openai,
-    inputs=gr.Textbox(lines=5, placeholder="Hi there, I have a plant that's..."),
+    title=title,
+    description=description,
+    fn=ask_openai, 
+    inputs=[
+        gr.Textbox(lines=5, placeholder="Hi there, I have a plant that's..."),
+        gr.Checkbox(label="Start a new conversation thread"),
+        gr.Dropdown(label="Select previous thread", choices=list(thread_ids.keys()))
+    ], 
     outputs=gr.Markdown(),
-    title="Wecome to Tonic's Bulbi-Tulu Plant Doctor",
-    description="""Welcome to Bulbi using [Tulu](https://huggingface.co/allenai/tulu-2-dpo-70b). Introduce your plant below. Be as descriptive as possible. Respond with additional information when prompted. Save your plants with Tulu Plant Doctor""",
     examples=examples
 )
 
-iface.launch()
+iface.launch()