Adding of multi word hashtags/designers

app.py

@@ -105,6 +105,43 @@ def submit_form(
     return int(prediction)
 
 
+def add_hashtag(hashtags, new_hashtag):
+    if not new_hashtag:
+        return hashtags
+    current = hashtags.split() if hashtags else []
+    if new_hashtag not in current:
+        current.append(new_hashtag)
+    return " ".join(current)
+
+def remove_hashtag(hashtags, hashtag_to_remove):
+    if not hashtags:
+        return ""
+    current = hashtags.split()
+    try:
+        current.remove(hashtag_to_remove.strip())
+    except ValueError:
+        pass
+    return " ".join(current)
+
+def add_designer(designers, new_designer):
+    if not new_designer:
+        return designers
+    current = designers.split() if designers else []
+    if new_designer not in current:
+        current.append(new_designer)
+    return " ".join(current)
+
+def remove_designer(designers, designer_to_remove):
+    if not designers:
+        return ""
+    current = designers.split()
+    try:
+        current.remove(designer_to_remove.strip())
+    except ValueError:
+        pass
+    return " ".join(current)
+
+
 with gr.Blocks(theme="argilla/argilla-theme", title="Grailed Price Predictor") as demo:
     global size_text_box
     gr.HTML(
@@ -188,24 +225,82 @@ with gr.Blocks(theme="argilla/argilla-theme", title="Grailed Price Predictor") a
 
         with gr.Row():
             with gr.Column():
-                hashtags_text_box = gr.Textbox(label="Hashtags")
+                hashtags_state = gr.State("")
+                new_hashtag_input = gr.Textbox(label="Add Hashtag")
+                with gr.Row():
+                    add_hashtag_btn = gr.Button("Add Hashtag")
+                    clear_hashtags_btn = gr.Button("Clear All Hashtags")
+                hashtags_display = gr.Dataframe(
+                    headers=["Hashtag"],
+                    interactive=False,
+                    label="Current Hashtags"
+                )
+                
+                def update_hashtags_display(hashtags):
+                    if not hashtags:
+                        return []
+                    return [[tag] for tag in hashtags.split()]
 
-                @gr.render(inputs=hashtags_text_box)
-                def show_hashtags(textbox):
-                    if not textbox:
-                        return
-                    for i, hashtag in enumerate(textbox.split(" ")):
-                        gr.Textbox(label=f"Hashtag #{i+1}", value=hashtag)
+                add_hashtag_btn.click(
+                    add_hashtag,
+                    inputs=[hashtags_state, new_hashtag_input],
+                    outputs=hashtags_state
+                ).then(
+                    update_hashtags_display,
+                    inputs=[hashtags_state],
+                    outputs=hashtags_display
+                ).then(
+                    lambda: "", 
+                    outputs=new_hashtag_input
+                )
+
+                # Add clear functionality
+                clear_hashtags_btn.click(
+                    lambda: "",  # Clear the state
+                    outputs=hashtags_state
+                ).then(
+                    lambda: [],  # Clear the display
+                    outputs=hashtags_display
+                )
 
             with gr.Column():
-                designer_names_text_box = gr.Textbox(label="Designer Names")
+                designers_state = gr.State("")
+                new_designer_input = gr.Textbox(label="Add Designer")
+                with gr.Row():
+                    add_designer_btn = gr.Button("Add Designer")
+                    clear_designers_btn = gr.Button("Clear All Designers")
+                designers_display = gr.Dataframe(
+                    headers=["Designer"],
+                    interactive=False,
+                    label="Current Designers"
+                )
 
-                @gr.render(inputs=designer_names_text_box)
-                def show_designer_names(textbox):
-                    if not textbox:
-                        return
-                    for i, designer_name in enumerate(textbox.split(" ")):
-                        gr.Textbox(label=f"Designer Name #{i+1}", value=designer_name)
+                def update_designers_display(designers):
+                    if not designers:
+                        return []
+                    return [[designer] for designer in designers.split()]
+
+                add_designer_btn.click(
+                    add_designer,
+                    inputs=[designers_state, new_designer_input],
+                    outputs=designers_state
+                ).then(
+                    update_designers_display,
+                    inputs=[designers_state],
+                    outputs=designers_display
+                ).then(
+                    lambda: "",
+                    outputs=new_designer_input
+                )
+
+                # Add clear functionality
+                clear_designers_btn.click(
+                    lambda: "",  # Clear the state
+                    outputs=designers_state
+                ).then(
+                    lambda: [],  # Clear the display
+                    outputs=designers_display
+                )
 
         with gr.Row():
             followernoNumber = gr.Number(label="Number of Followers")
@@ -227,8 +322,8 @@ with gr.Blocks(theme="argilla/argilla-theme", title="Grailed Price Predictor") a
                 sub_sub_category,
                 size_text_box,
                 color_text_box,
-                hashtags_text_box,
-                designer_names_text_box,
+                hashtags_state,
+                designers_state,
                 followernoNumber,
                 # userScoreNumber,
             ],