clear till index

app.py

@@ -12,9 +12,17 @@ def get_model_response(chat_txt, user_chats, full_chat_history):
     return responses, full_chat_history, user_chats
 
 
-def clear_chat(index: int):
-    print(index)
-    return [], [], [], [], gr.update(choices=[], value="")
+def clear_chat(
+    full_chat_history, chatbox, user_chats, cleartill_dropdown_choices, drop_down
+):
+    index = user_chats.index(drop_down)
+    return (
+        full_chat_history[:index],
+        chatbox[:index],
+        user_chats[:index],
+        cleartill_dropdown_choices[:index],
+        gr.update(choices=cleartill_dropdown_choices[:index], value=""),
+    )
 
 
 def add_to_dropdown(txt, drop_down_options):
@@ -23,7 +31,10 @@ def add_to_dropdown(txt, drop_down_options):
 
 
 drop_down = gr.Dropdown(
-    label="Clear Till", choices=[], interactive=True, multiselect=False, type="index"
+    label="Clear From (including)",
+    choices=[],
+    interactive=True,
+    multiselect=False,
 )
 with gr.Blocks() as demo:
     user_chats = gr.State(value=[])
@@ -52,7 +63,13 @@ with gr.Blocks() as demo:
             clear_btn = gr.Button("Clear", label="Clear chat history")
             clear_btn.click(
                 clear_chat,
-                inputs=[drop_down],
+                inputs=[
+                    full_chat_history,
+                    chatbox,
+                    user_chats,
+                    cleartill_dropdown_choices,
+                    drop_down,
+                ],
                 outputs=[
                     full_chat_history,
                     chatbox,