renamed to app.py

chatbot-demo.py → app.py

@@ -12,12 +12,9 @@ def get_model_response(chat_txt, user_chats, full_chat_history):
     return responses, full_chat_history, user_chats
 
 
-def clear_chat():
-    return [], [], [], []
-
-
-def clear_dropdown():
-    return gr.update(choices=[], value="")
+def clear_chat(index: int):
+    print(index)
+    return [], [], [], [], gr.update(choices=[], value="")
 
 
 def add_to_dropdown(txt, drop_down_options):
@@ -26,12 +23,12 @@ def add_to_dropdown(txt, drop_down_options):
 
 
 drop_down = gr.Dropdown(
-    label="Clear Till", choices=[], interactive=True, multiselect=False
+    label="Clear Till", choices=[], interactive=True, multiselect=False, type="index"
 )
 with gr.Blocks() as demo:
     user_chats = gr.State(value=[])
     full_chat_history = gr.State(value=[])
-    clear_dropdown_options = gr.State(value=[])
+    cleartill_dropdown_choices = gr.State(value=[])
 
     chatbox = gr.Chatbot()
     with gr.Row():
@@ -47,22 +44,23 @@ with gr.Blocks() as demo:
             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],
+                inputs=[user_chat_txt, cleartill_dropdown_choices],
+                outputs=[drop_down, cleartill_dropdown_choices],
             )
         with gr.Column(scale=1, min_width=250):
             drop_down.render()
             clear_btn = gr.Button("Clear", label="Clear chat history")
             clear_btn.click(
                 clear_chat,
+                inputs=[drop_down],
                 outputs=[
                     full_chat_history,
                     chatbox,
                     user_chats,
-                    clear_dropdown_options,
+                    cleartill_dropdown_choices,
+                    drop_down,
                 ],
             )
-            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*")