update dropdown

README.md

@@ -10,4 +10,8 @@ pinned: false
 license: apache-2.0
 ---
 
+```
+export no_proxy="localhost, 127.0.0.1"
+```
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -4,15 +4,16 @@ import json
 from pymongo import MongoClient
 import os
 
-
 # Initialize MongoDB client
-client = MongoClient(os.environ['DB_URL'])
+# client = MongoClient(os.environ['DB_URL'])
+client = MongoClient("mongodb+srv://gradio-test:jiheng@cluster0.dvxynte.mongodb.net/?retryWrites=true&w=majority")
+
 db = client['test']
 collection = db['gradio']
 
 def get_saved_data():
-    saved_data = collection.find({}, {"_id": 0, "question": 1, "system_message": 1})
-    options = [json.dumps(item) for item in saved_data]
+    saved_data = collection.find({}, {"_id": 0, "name": 1})
+    options = [item['name'] for item in saved_data]
     return options
 
 def chat_with_gpt(question, api_key, temperature, system_message):
@@ -28,18 +29,24 @@ def chat_with_gpt(question, api_key, temperature, system_message):
     assistant_reply = response['choices'][0]['message']['content']
     return f"{assistant_reply}"
 
-def save_to_mongodb(question, system_message):
-    collection.insert_one({"question": question, "system_message": system_message})
-    return "Saved to MongoDB."
+def update_dropdown_choices():
+    new_choices = get_saved_data()
+    return gr.Dropdown.update(choices=new_choices)
+
+def save_to_mongodb(name, question, system_message):
+    if not name.strip():  # Check if name is empty or just whitespace
+        return "Please enter a name.", None  # Return a message and None to indicate no update for the dropdown
+    collection.insert_one({"name": name, "question": question, "system_message": system_message})
+    return "Saved to MongoDB.", update_dropdown_choices()
+
+def update_textboxes(selected_name):
+    selected_data = collection.find_one({"name": selected_name}, {"_id": 0})
+    return selected_data['question'], selected_data['system_message'], selected_data['name']
 
-def update_textboxes(selected_data):
-    print(selected_data)
-    select_json = json.loads(selected_data)
-    print(select_json['question'], select_json['system_message'])
-    return select_json['question'], select_json['system_message']
 
 with gr.Blocks() as app:
     saved_data_dropdown = gr.Dropdown(get_saved_data(), label="Select Saved Data")
+    name = gr.Textbox(lines=1, placeholder="Name", label="Name")
     question = gr.Textbox(lines=2, placeholder="What's your question?", label="Question")
     api_key = gr.Textbox(lines=1, placeholder="Your OpenAI API Key", label="API Key")
     temperature = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.7, label="Temperature")
@@ -52,7 +59,7 @@ with gr.Blocks() as app:
     output = gr.Textbox(label="Result", interactive=False)
 
     chat_btn.click(chat_with_gpt, inputs=[question, api_key, temperature, system_message], outputs=output)
-    save_btn.click(save_to_mongodb, inputs=[question, system_message], outputs=output)
-    saved_data_dropdown.select(update_textboxes, inputs=[saved_data_dropdown], outputs=[question, system_message])
+    save_btn.click(save_to_mongodb, inputs=[name, question, system_message], outputs=[output, saved_data_dropdown])
+    saved_data_dropdown.select(update_textboxes, inputs=[saved_data_dropdown], outputs=[question, system_message, name])
 
 app.launch()