up

app.py

@@ -19,9 +19,17 @@ SUBMISSIONS = {
     "karlo": load_dataset(os.path.join(ORG, "if-v-1.0"))["train"],
     # "Kadinsky":
 }
+
+LINKS = {
+    "sd-v1-5": "https://huggingface.co/runwayml/stable-diffusion-v1-5",
+    "sd-v2-1": "https://huggingface.co/stabilityai/stable-diffusion-2-1",
+    "if-v1-0": "https://huggingface.co/DeepFloyd/IF-I-XL-v1.0",
+    "karlo": "https://huggingface.co/kakaobrain/karlo-v1-alpha",
+}
 NUM_QUESTIONS = 10
 MODEL_KEYS = "-".join(SUBMISSIONS.keys())
 SUBMISSION_ORG = f"results-{MODEL_KEYS}"
+PROMPT_FORMAT = "Pick the picture that best matches the prompt: **{}**"
 
 
 submission_names = list(SUBMISSIONS.keys())
@@ -106,7 +114,7 @@ def start():
 
 def process(dataframe, row_number=0):
     if row_number == NUM_QUESTIONS:
-        return None, ""
+        return None, "", ""
 
     image_id = dataframe.iloc[row_number]["id"]
     choices = [
@@ -116,12 +124,13 @@ def process(dataframe, row_number=0):
     images = [SUBMISSIONS[c][int(image_id)]["images"] for c in choices]
 
     prompt = SUBMISSIONS[choices[0]][int(image_id)]["Prompt"]
-    prompt = f"Prompt {row_number + 1}/{NUM_QUESTIONS}: '{prompt}'"
+    prompt = PROMPT_FORMAT.format(prompt)
+    counter = f"{row_number + 1}/{NUM_QUESTIONS}"
 
-    return images, prompt
+    return images, prompt, counter
 
 
-def write_result(user_choice, row_number, dataframe, prompt):
+def write_result(user_choice, row_number, dataframe):
     if row_number == NUM_QUESTIONS:
         return row_number, dataframe
 
@@ -147,31 +156,46 @@ def change_view(row_number, dataframe):
 
         dataset.push_to_hub(repo_id, token=os.getenv("HF_TOKEN"))
         return {
-            result: f"You are of type: {favorite_model}!",
+            intro_view: gr.update(visible=True),
             result_view: gr.update(visible=True),
             gallery_view: gr.update(visible=False),
+            result: f"You are of type: [**{favorite_model}**]({LINKS[favorite_model]}) 🔥",
         }
     else:
         return {
-            result: "",
+            intro_view: gr.update(visible=False),
             result_view: gr.update(visible=False),
             gallery_view: gr.update(visible=True),
+            result: "",
         }
 
 
 if True:
-    TITLE = "Community Parti Prompts - Who is your open-source genAI model?"
+    TITLE = "# Community Parti Prompts - Who is your open-source genAI model?"
     DESCRIPTION = """
-    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. 
-    You choices will go into the public community [genAI leaderboard](TODO)
+    *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
+    """
+
     GALLERY_COLUMN_NUM = len(SUBMISSIONS)
 
     with gr.Blocks(css="style.css") as demo:
         gr.Markdown(TITLE)
         gr.Markdown(DESCRIPTION)
-        start_button = gr.Button("Start").style(full_width=False)
+
+        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))
@@ -200,13 +224,12 @@ if True:
             )
             gr.Markdown("Click on start to play again!")
 
-        with gr.Column(visible=True) as gallery_view:
-            gr.Markdown("Pick your the photo that best corresponds to the prompt.")
-            prompt = gr.Markdown(f"Prompt 1/{NUM_QUESTIONS}: ")
+        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):
@@ -215,7 +238,8 @@ if True:
         start_button.click(
             fn=start,
             inputs=[],
-            outputs=dataframe
+            outputs=dataframe,
+            show_progress=True
         ).then(
             fn=lambda x: 0 if x == NUM_QUESTIONS else x,
             inputs=[row_number],
@@ -223,9 +247,9 @@ if True:
         ).then(
             fn=change_view,
             inputs=[row_number, dataframe],
-            outputs=[result_view, gallery_view, result]
+            outputs=[intro_view, result_view, gallery_view, result]
         ).then(
-            fn=process, inputs=[dataframe], outputs=[gallery, prompt]
+            fn=process, inputs=[dataframe], outputs=[gallery, prompt, counter]
         )
 
         gallery.select(
@@ -236,16 +260,16 @@ if True:
 
         next_button.click(
             fn=write_result,
-            inputs=[selected_image, row_number, dataframe, prompt],
+            inputs=[selected_image, row_number, dataframe],
             outputs=[row_number, dataframe],
         ).then(
             fn=change_view,
             inputs=[row_number, dataframe],
-            outputs=[result_view, gallery_view, result]
+            outputs=[intro_view, result_view, gallery_view, result]
         ).then(
             fn=process,
             inputs=[dataframe, row_number],
-            outputs=[gallery, prompt]
+            outputs=[gallery, prompt, counter]
         ).then(
             fn=lambda x: 0 if x == NUM_QUESTIONS else x,
             inputs=[row_number],