Update app.py

app.py

@@ -19,26 +19,27 @@ OpenAI doesnt let you use Agents without paying for it, so I made you an interfa
 Big thanks to 🤗Huggingface Organisation for the🫂Community Grant"""
 
 examples = [
-    ["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."]
+    ["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')
 client = openai.OpenAI(api_key=openai.api_key)
+thread_ids = {}
 
-def ask_openai(question):
-    global current_thread_id
+def ask_openai(question, start_new_thread=True, selected_thread_id=None):
+    global thread_ids
 
     try:
-
-        if current_thread_id is None:
+        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:
-            thread = client.beta.threads.retrieve(thread_id=current_thread_id)
+            current_thread_id = thread_ids[selected_thread_id]
 
         client.beta.threads.messages.create(
             thread_id=current_thread_id,
@@ -92,7 +93,13 @@ iface = gr.Interface(
     title=title,
     description=description,
     fn=ask_openai, 
-    inputs=gr.Textbox(lines=5, placeholder="Hi there, I have a plant that's..."), 
+    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(),
     examples=examples
-)
+)
+
+iface.launch()