Update app.py

app.py

@@ -31,17 +31,6 @@ MODEL_KEYS = "-".join(SUBMISSIONS.keys())
 SUBMISSION_ORG = f"results-{MODEL_KEYS}"
 PROMPT_FORMAT = "Pick the picture that best matches the prompt: **{}**"
 
-
-all_datasets = list_datasets(author=SUBMISSION_ORG)
-relevant_ids = [d.id for d in all_datasets]
-
-submitted_ids = []
-for _id in relevant_ids:
-    ds = load_dataset(_id)["train"]
-    submitted_ids += ds["id"]
-
-submitted_ids = Counter(submitted_ids)
-
 submission_names = list(SUBMISSIONS.keys())
 num_images = len(SUBMISSIONS[submission_names[0]])
 
@@ -65,10 +54,30 @@ def generate_random_hash(length=8):
     
     return random_hash
 
+def load_submissions():
+    all_datasets = list_datasets(author=SUBMISSION_ORG)
+    relevant_ids = [d.id for d in all_datasets]
+    
+    submitted_ids = []
+    for _id in relevant_ids:
+        ds = load_dataset(_id)["train"]
+        submitted_ids += ds["id"]
+    
+    submitted_ids = Counter(submitted_ids)
+    return submitted_ids
 
-def start():
-    ids = {id: 0 for id in range(num_images)}
+def refresh(row_number, dataframe):
+    if row_number == NUM_QUESTIONS:
+        submitted_ids = load_submissions()
+        return start(submitted_ids)
+    else:
+        return dataframe
 
+def start(submitted_ids=None):
+    if submitted_ids is None:
+        submitted_ids = load_submissions()
+        
+    ids = {id: 0 for id in range(num_images)}
     ids = {**ids, **submitted_ids}
 
     # sort by count
@@ -170,110 +179,113 @@ def change_view(row_number, dataframe):
         }
 
 
-if True:
-    TITLE = "# Community Parti Prompts - Who is your open-source genAI model?"
-    DESCRIPTION = """
-    *This is an interactive game in which you click through pre-generated images from SD-v1-5, SD-v2.1, Karlo, and IF 
-    using [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) prompts.* \n
-    *You choices will go into the public community [genAI leaderboard](TODO).*
-    """
-    EXPLANATION = """\n\n
-    ## How it works 📖 \n\n
-
-    1. Click on 'Start'
-    2. A prompt and 4 different images are displayed
-    3. Select your favorite image
-    4. Click on 'Select'
-    5. After 10 rounds your favorite diffusion model is displayed
-    """
+TITLE = "# Community Parti Prompts - Who is your open-source genAI model?"
+DESCRIPTION = """
+*This is an interactive game in which you click through pre-generated images from SD-v1-5, SD-v2.1, Karlo, and IF 
+using [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) prompts.* \n
+*You choices will go into the public community [genAI leaderboard](TODO).*
+"""
+EXPLANATION = """\n\n
+## How it works 📖 \n\n
 
-    GALLERY_COLUMN_NUM = len(SUBMISSIONS)
-
-    with gr.Blocks(css="style.css") as demo:
-        gr.Markdown(TITLE)
-        gr.Markdown(DESCRIPTION)
-
-        with gr.Column(visible=True) as intro_view:
-            gr.Markdown(EXPLANATION)
-            start_button = gr.Button("Start").style(full_width=False)
-
-        headers = ["prompt", "result", "id", "Challenge", "Category", "Note"] + [
-            f"choice_{i}" for i in range(len(SUBMISSIONS))
-        ]
-        datatype = ["str", "str", "number", "str", "str", "str"] + len(SUBMISSIONS) * [
-            "number"
-        ]
-
-        with gr.Column(visible=False):
-            row_number = gr.Number(
-                label="Current row selection index",
-                value=0,
-                precision=0,
-                interactive=False,
-            )
-
-        # Create Data Frame
-        with gr.Column(visible=False) as result_view:
-            result = gr.Markdown("")
-            dataframe = gr.Dataframe(
-                headers=headers,
-                datatype=datatype,
-                row_count=NUM_QUESTIONS,
-                col_count=(6 + len(SUBMISSIONS), "fixed"),
-                interactive=False,
-            )
-            gr.Markdown("Click on start to play again!")
-
-        with gr.Column(visible=False) as gallery_view:
-            counter = gr.Markdown(f" ### 1/{NUM_QUESTIONS}")
-            prompt = gr.Markdown(PROMPT_FORMAT.format(""))
-            gallery = gr.Gallery(
-                label="All images", show_label=False, elem_id="gallery"
-            ).style(columns=GALLERY_COLUMN_NUM, object_fit="contain")
-            next_button = gr.Button("Select").style(full_width=False)
-
-        with gr.Column(visible=False):
-            selected_image = gr.Number(label="Selected index", value=-1, precision=0)
-
-        start_button.click(
-            fn=start,
-            inputs=[],
-            outputs=dataframe,
-            show_progress=True
-        ).then(
-            fn=lambda x: 0 if x == NUM_QUESTIONS else x,
-            inputs=[row_number],
-            outputs=[row_number],
-        ).then(
-            fn=change_view,
-            inputs=[row_number, dataframe],
-            outputs=[intro_view, result_view, gallery_view, result]
-        ).then(
-            fn=process, inputs=[dataframe], outputs=[gallery, prompt, counter]
-        )
+1. Click on 'Start'
+2. A prompt and 4 different images are displayed
+3. Select your favorite image
+4. Click on 'Select'
+5. After 10 rounds your favorite diffusion model is displayed
+"""
 
-        gallery.select(
-            fn=get_index,
-            outputs=selected_image,
-            queue=False,
-        )
+GALLERY_COLUMN_NUM = len(SUBMISSIONS)
+
+with gr.Blocks() as demo:
+    gr.Markdown(TITLE)
+    gr.Markdown(DESCRIPTION)
 
-        next_button.click(
-            fn=write_result,
-            inputs=[selected_image, row_number, dataframe],
-            outputs=[row_number, dataframe],
-        ).then(
-            fn=change_view,
-            inputs=[row_number, dataframe],
-            outputs=[intro_view, result_view, gallery_view, result]
-        ).then(
-            fn=process,
-            inputs=[dataframe, row_number],
-            outputs=[gallery, prompt, counter]
-        ).then(
-            fn=lambda x: 0 if x == NUM_QUESTIONS else x,
-            inputs=[row_number],
-            outputs=[row_number],
+    with gr.Column(visible=True) as intro_view:
+        gr.Markdown(EXPLANATION)
+        start_button = gr.Button("Start").style(full_width=False)
+
+    headers = ["prompt", "result", "id", "Challenge", "Category", "Note"] + [
+        f"choice_{i}" for i in range(len(SUBMISSIONS))
+    ]
+    datatype = ["str", "str", "number", "str", "str", "str"] + len(SUBMISSIONS) * [
+        "number"
+    ]
+
+    with gr.Column(visible=False):
+        row_number = gr.Number(
+            label="Current row selection index",
+            value=0,
+            precision=0,
+            interactive=False,
         )
 
-    demo.launch()
+    # Create Data Frame
+    with gr.Column(visible=False) as result_view:
+        result = gr.Markdown("")
+        dataframe = gr.Dataframe(
+            headers=headers,
+            datatype=datatype,
+            row_count=NUM_QUESTIONS,
+            col_count=(6 + len(SUBMISSIONS), "fixed"),
+            interactive=False,
+        )
+        gr.Markdown("Click on start to play again!")
+
+    with gr.Column(visible=False) as gallery_view:
+        counter = gr.Markdown(f" ### 1/{NUM_QUESTIONS}")
+        prompt = gr.Markdown(PROMPT_FORMAT.format(""))
+        gallery = gr.Gallery(
+            label="All images", show_label=False, elem_id="gallery"
+        ).style(columns=GALLERY_COLUMN_NUM, object_fit="contain")
+        next_button = gr.Button("Select").style(full_width=False)
+
+    with gr.Column(visible=False):
+        selected_image = gr.Number(label="Selected index", value=-1, precision=0)
+
+    start_button.click(
+        fn=start,
+        inputs=[],
+        outputs=dataframe,
+        show_progress=True
+    ).then(
+        fn=lambda x: 0 if x == NUM_QUESTIONS else x,
+        inputs=[row_number],
+        outputs=[row_number],
+    ).then(
+        fn=change_view,
+        inputs=[row_number, dataframe],
+        outputs=[intro_view, result_view, gallery_view, result]
+    ).then(
+        fn=process, inputs=[dataframe], outputs=[gallery, prompt, counter]
+    )
+
+    gallery.select(
+        fn=get_index,
+        outputs=selected_image,
+        queue=False,
+    )
+
+    next_button.click(
+        fn=write_result,
+        inputs=[selected_image, row_number, dataframe],
+        outputs=[row_number, dataframe],
+    ).then(
+        fn=change_view,
+        inputs=[row_number, dataframe],
+        outputs=[intro_view, result_view, gallery_view, result]
+    ).then(
+        fn=process,
+        inputs=[dataframe, row_number],
+        outputs=[gallery, prompt, counter]
+    ).then(
+        fn=lambda x: 0 if x == NUM_QUESTIONS else x,
+        inputs=[row_number],
+        outputs=[row_number],
+    ).then(
+        fn=refresh,
+        inputs=[row_number, dataframe],
+        outputs=[dataframe],
+    )
+
+demo.launch()