Add rating and description parameters to flag_img_input function in app.py, enhancing user feedback capabilities and updating the UI to accommodate new inputs.

app.py

@@ -70,12 +70,13 @@ def inference(image):
 
 def flag_img_input(
     image: gr.Image,
-    name: str = "misdetection",
+    name: str = "anonymous",
     email: str = "[email protected]",
     subscribe: bool = False,
+    rating: int = 0,
+    description: str = "",
 ):
     """Wrapper for flagging"""
-    logger.info("Flagging image - name: %s, email: %s, subscribe: %s", name, email, subscribe)
 
     # Decode blob data if necessary
     if is_blob_data(image):
@@ -85,6 +86,8 @@ def flag_img_input(
         "name": name,
         "email": email,
         "subscribe": subscribe,
+        "rating": rating,
+        "description": description,
     }
 
     hf_writer.flag([image], metadata=metadata)
@@ -162,6 +165,8 @@ with gr.Blocks(theme=theme, css=css, title="SEA.AI Vision Demo") as demo:
     name_ = gr.Textbox(label="name", visible=False, value="anonymous")
     email_ = gr.Textbox(label="email", visible=False, value="[email protected]")
     subscribe_ = gr.Checkbox(label="subscribe", value=False, visible=False)
+    rating_ = gr.Slider(label="rating", value=0, minimum=0, maximum=5, step=1, visible=False)
+    description_ = gr.Textbox(label="description", visible=False, value="")
 
     # This needs to be called prior to the first call to callback.flag()
     hf_writer.setup([img_input], "flagged")
@@ -174,7 +179,7 @@ with gr.Blocks(theme=theme, css=css, title="SEA.AI Vision Demo") as demo:
         show_api=False,
     ).then(
         flag_img_input,
-        [img_input, name_, email_, subscribe_],
+        [img_input, name_, email_, subscribe_, rating_, description_],
         [],
         preprocess=False,
         show_api=True,