add submission selection

competitions/__init__.py

@@ -3,7 +3,7 @@ import os
 from .info import CompetitionInfo
 
 
-__version__ = "0.0.7"
+__version__ = "0.0.8"
 
 MOONLANDING_URL = os.getenv("MOONLANDING_URL")
 COMPETITION_ID = os.getenv("COMPETITION_ID")

competitions/competitions.py

@@ -34,21 +34,35 @@ def _my_submissions(user_token):
             gr.Markdown.update(visible=True, value=NO_SUBMISSIONS),
             gr.DataFrame.update(visible=False),
             gr.TextArea.update(visible=False),
+            gr.Button.update(visible=False),
         ]
-    selected_submission_ids = df[df["selected"] is True]["submission_id"].values.tolist()
+    selected_submission_ids = df[df["selected"] == True]["submission_id"].values.tolist()
     if len(selected_submission_ids) > 0:
         return [
             gr.Markdown.update(visible=True),
-            gr.DataFrame.update(visible=True, data=df),
-            gr.TextArea.update(visible=False, value="\n".join(selected_submission_ids)),
+            gr.DataFrame.update(visible=True, value=df),
+            gr.TextArea.update(visible=True, value="\n".join(selected_submission_ids), interactive=True),
+            gr.Button.update(visible=True),
         ]
     return [
         gr.Markdown.update(visible=False),
         gr.DataFrame.update(visible=True, value=df),
-        gr.TextArea.update(visible=True),
+        gr.TextArea.update(visible=True, interactive=True),
+        gr.Button.update(visible=True),
     ]
 
 
+def _update_selected_submissions(user_token, submission_ids):
+    submission_ids = submission_ids.split("\n")
+    submission_ids = [sid.strip() for sid in submission_ids]
+    if len(submission_ids) > competition_info.selection_limit:
+        raise ValueError(
+            f"You can select only {competition_info.selection_limit} submissions. You selected {len(submission_ids)} submissions."
+        )
+    submissions.update_selected_submissions(user_token, submission_ids)
+    return _my_submissions(user_token)
+
+
 with gr.Blocks() as demo:
     with gr.Tabs() as tab_container:
         with gr.TabItem("Overview", id="overview"):
@@ -84,12 +98,23 @@ with gr.Blocks() as demo:
             )
             output_text = gr.Markdown(visible=True, show_label=False)
             output_df = gr.DataFrame(visible=False)
-            selected_submissions = gr.TextArea(visible=False, label="Selected Submissions")
+            selected_submissions = gr.TextArea(
+                visible=False,
+                label="Selected Submissions (one submission id per line)",
+                max_lines=competition_info.selection_limit,
+                lines=competition_info.selection_limit,
+            )
+            update_selected_submissions = gr.Button("Update Selected Submissions", visible=False)
             my_subs_button = gr.Button("Fetch Submissions")
             my_subs_button.click(
                 fn=_my_submissions,
                 inputs=[user_token],
-                outputs=[output_text, output_df, selected_submissions],
+                outputs=[output_text, output_df, selected_submissions, update_selected_submissions],
+            )
+            update_selected_submissions.click(
+                fn=_update_selected_submissions,
+                inputs=[user_token, selected_submissions],
+                outputs=[output_text, output_df, selected_submissions, update_selected_submissions],
             )
 
         fetch_lb_partial = partial(leaderboard.fetch, private=False)

competitions/submissions.py

@@ -156,6 +156,38 @@ class Submissions:
             user_submission_info = json.load(f)
         return user_submission_info["submissions"]
 
+    def update_selected_submissions(self, user_token, selected_submission_ids):
+        user_info = self._get_user_info(user_token)
+        user_id = user_info["id"]
+
+        user_fname = hf_hub_download(
+            repo_id=self.competition_id,
+            filename=f"{user_id}.json",
+            use_auth_token=self.autotrain_token,
+            repo_type="dataset",
+        )
+        with open(user_fname, "r") as f:
+            user_submission_info = json.load(f)
+
+        for sub in user_submission_info["submissions"]:
+            if sub["submission_id"] in selected_submission_ids:
+                sub["selected"] = True
+            else:
+                sub["selected"] = False
+
+        # convert user_submission_info to BufferedIOBase file object
+        user_submission_info_json = json.dumps(user_submission_info)
+        user_submission_info_json_bytes = user_submission_info_json.encode("utf-8")
+        user_submission_info_json_buffer = io.BytesIO(user_submission_info_json_bytes)
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=user_submission_info_json_buffer,
+            path_in_repo=f"{user_id}.json",
+            repo_id=self.competition_id,
+            repo_type="dataset",
+            token=self.autotrain_token,
+        )
+
     def _get_user_subs(self, user_info, private=False):
         # get user submissions
         user_id = user_info["id"]