finish

app.py

@@ -29,13 +29,14 @@ LINKS = {
 NUM_QUESTIONS = 10
 MODEL_KEYS = "-".join(SUBMISSIONS.keys())
 SUBMISSION_ORG = f"results-{MODEL_KEYS}"
-PROMPT_FORMAT = "Pick the picture that best matches the prompt: **{}**"
+PROMPT_FORMAT = " Pick the picture that best matches the prompt:"
 
 submission_names = list(SUBMISSIONS.keys())
 num_images = len(SUBMISSIONS[submission_names[0]])
 
 
 def load_submissions():
+    return {}
     all_datasets = list_datasets(author=SUBMISSION_ORG)
     relevant_ids = [d.id for d in all_datasets]
     
@@ -125,7 +126,7 @@ def start():
 
 def process(dataframe, row_number=0):
     if row_number == NUM_QUESTIONS:
-        return None, "", ""
+        return None, None, None, None, "", ""
 
     image_id = dataframe.iloc[row_number]["id"]
     choices = [
@@ -135,8 +136,8 @@ 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 = PROMPT_FORMAT.format(prompt)
-    counter = f"{row_number + 1}/{NUM_QUESTIONS}"
+    prompt = f'# "{prompt}"'
+    counter = f"***{row_number + 1}/{NUM_QUESTIONS} {PROMPT_FORMAT}***"
 
     return *images, prompt, counter
 
@@ -199,10 +200,9 @@ EXPLANATION = """\n\n
 GALLERY_COLUMN_NUM = len(SUBMISSIONS)
 
 with gr.Blocks() as demo:
-    gr.Markdown(TITLE)
-    gr.Markdown(DESCRIPTION)
-
     with gr.Column(visible=True) as intro_view:
+        gr.Markdown(TITLE)
+        gr.Markdown(DESCRIPTION)
         gr.Markdown(EXPLANATION)
         start_button = gr.Button("Start").style(full_width=False)
 
@@ -237,20 +237,22 @@ with gr.Blocks() as demo:
         selected_image = gr.Number(label="Selected index", value=-1, precision=0)
 
     with gr.Column(visible=False) as gallery_view:
-        counter = gr.Markdown(f" ### 1/{NUM_QUESTIONS}")
-        prompt = gr.Markdown(PROMPT_FORMAT.format(""))
+        with gr.Row():
+            counter = gr.Markdown(f"***1/{NUM_QUESTIONS} {PROMPT_FORMAT}***")
+        with gr.Row():
+            prompt = gr.Markdown("")
         with gr.Blocks():
             with gr.Row():
-                with gr.Column():
+                with gr.Column() as c1:
                     image_1 = gr.Image(interactive=False)
                     image_1_button = gr.Button("Select 1").style(full_width=True)
-                with gr.Column():
+                with gr.Column() as c2:
                     image_2 = gr.Image(interactive=False)
                     image_2_button = gr.Button("Select 2").style(full_width=True)
-                with gr.Column():
+                with gr.Column() as c3:
                     image_3 = gr.Image(interactive=False)
                     image_3_button = gr.Button("Select 3").style(full_width=True)
-                with gr.Column():
+                with gr.Column() as c4:
                     image_4 = gr.Image(interactive=False)
                     image_4_button = gr.Button("Select 4").style(full_width=True)         
 
@@ -271,10 +273,15 @@ with gr.Blocks() as demo:
         fn=process, inputs=[dataframe], outputs=[image_1, image_2, image_3, image_4, prompt, counter]
     )
 
-    for i, button in enumerate([image_1_button, image_2_button, image_3_button, image_4_button]):
+    images = [image_1_button, image_2_button, image_3_button, image_4_button]
+    for i, button in enumerate(images):
         button.click(
+            fn=lambda: i,
+            inputs=[],
+            outputs=[selected_image],
+        ).then(
             fn=write_result,
-            inputs=[i, row_number, dataframe],
+            inputs=[selected_image, row_number, dataframe],
             outputs=[row_number, dataframe],
         ).then(
             fn=change_view,