added a dropbox that selects elements correctly

chatbot-demo.py

@@ -1,37 +1,68 @@
 import gradio as gr
 
 
-def get_model_response(user_input, history, user_chats):
-    print(user_input, history, user_chats)
-    response = f'Hello, you said "{user_input}"'
-    history += [user_input, response]
-    user_chats.append(user_input)
-    responses = [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)]
-    return responses, history, user_chats
+def get_model_response(chat_txt, user_chats, full_chat_history):
+    response = f'Hello, you said "{chat_txt}"'
+    full_chat_history = full_chat_history + [chat_txt, response]
+    user_chats.append(chat_txt)
+    responses = [
+        (full_chat_history[i], full_chat_history[i + 1])
+        for i in range(0, len(full_chat_history) - 1, 2)
+    ]
+    return responses, full_chat_history, user_chats
 
 
 def clear_chat():
-    return [], []
+    return [], [], [], []
 
 
+def clear_dropdown():
+    return gr.update(choices=[], value="")
+
+
+def add_to_dropdown(txt, drop_down_options):
+    drop_down_options = drop_down_options + [txt]
+    return gr.update(choices=drop_down_options), drop_down_options
+
+
+drop_down = gr.Dropdown(
+    label="Clear Till", choices=[], interactive=True, multiselect=False
+)
 with gr.Blocks() as demo:
-    state = gr.State([])
     user_chats = gr.State(value=[])
+    full_chat_history = gr.State(value=[])
+    clear_dropdown_options = gr.State(value=[])
 
-    chat = gr.Chatbot()
+    chatbox = gr.Chatbot()
     with gr.Row():
         with gr.Column(scale=5):
-            txt = gr.Textbox(placeholder="Say Hi!", label="press Enter to submit")
-            txt.submit(
+            user_chat_txt = gr.Textbox(
+                placeholder="Say Hi!", label="press Enter to submit"
+            )
+            user_chat_txt.submit(
                 get_model_response,
-                [txt, state, user_chats],
-                [chat, state, user_chats],
+                inputs=[user_chat_txt, user_chats, full_chat_history],
+                outputs=[chatbox, full_chat_history, user_chats],
+            )
+            user_chat_txt.submit(lambda: gr.update(value=""), [], user_chat_txt)
+            user_chat_txt.submit(
+                add_to_dropdown,
+                inputs=[user_chat_txt, clear_dropdown_options],
+                outputs=[drop_down, clear_dropdown_options],
             )
         with gr.Column(scale=1, min_width=250):
-            choices = ["test"]
-            drop_down = gr.Dropdown(label="Clear Till", choices=user_chats.value)
+            drop_down.render()
             clear_btn = gr.Button("Clear", label="Clear chat history")
-            clear_btn.click(clear_chat, outputs=[state, chat])
+            clear_btn.click(
+                clear_chat,
+                outputs=[
+                    full_chat_history,
+                    chatbox,
+                    user_chats,
+                    clear_dropdown_options,
+                ],
+            )
+            clear_btn.click(clear_dropdown, outputs=drop_down)
 
     with gr.Accordion(label="Red-Teaming", open=True):
         gr.Markdown("*instructions on what to do if you break the model*")