backup

app.py

@@ -151,10 +151,14 @@ def load_question_data(question_id):
 
 def load_post_image(post_id):
     if not post_id:
-        return [None] * 31  # source image + 10 triplets of (image, text, notes)
+        return [
+            None
+        ] * 33  # source image + instruction + simplified_instruction + 10 triplets
 
     idx = POST_ID_TO_ID_MAP[post_id]
     source_image = VALID_DATASET[idx]["image"]
+    instruction = VALID_DATASET[idx]["instruction"]
+    simplified_instruction = VALID_DATASET[idx]["simplified_instruction"]
 
     # Load existing responses if any
     post_folder = os.path.join("./data", str(post_id))
@@ -175,15 +179,15 @@ def load_post_image(post_id):
                 responses[idx] = (
                     image_path,
                     response["answer_text"],
-                    response.get("notes", ""),  # Get notes with empty string as default
+                    response.get("notes", ""),
                 )
 
         # Flatten responses for output
         flat_responses = [item for triplet in responses for item in triplet]
-        return [source_image] + flat_responses
+        return [source_image, instruction, simplified_instruction] + flat_responses
 
-    # If no existing responses, return source image and empty responses
-    return [source_image] + [None] * 30
+    # If no existing responses, return source image, instructions and empty responses
+    return [source_image, instruction, simplified_instruction] + [None] * 30
 
 
 def generate_json_files(source_image, responses, post_id):
@@ -255,21 +259,28 @@ with gr.Blocks() as demo:
     gr.Markdown("# Image Response Collector")
 
     # Source image selection at the top
-    with gr.Column():
-        post_id_dropdown = gr.Dropdown(
-            label="Select Post ID to Load Image",
-            choices=VALID_DATASET_POST_IDS,
-            type="value",
-            allow_custom_value=False,
-        )
-        source_image = gr.Image(label="Source Image", type="filepath")
+    with gr.Row():
+        with gr.Column():
+            post_id_dropdown = gr.Dropdown(
+                label="Select Post ID to Load Image",
+                choices=VALID_DATASET_POST_IDS,
+                type="value",
+                allow_custom_value=False,
+            )
+            instruction_text = gr.Textbox(label="Instruction", interactive=False)
+            simplified_instruction_text = gr.Textbox(
+                label="Simplified Instruction", interactive=False
+            )
+        source_image = gr.Image(label="Source Image", type="filepath", height=300)
 
     # Responses in tabs
     with gr.Tabs() as response_tabs:
         responses = []
         for i in range(10):
             with gr.Tab(f"Response {i+1}"):
-                img = gr.Image(label=f"Response Image {i+1}", type="filepath")
+                img = gr.Image(
+                    label=f"Response Image {i+1}", type="filepath", height=300
+                )
                 txt = gr.Textbox(label=f"Model Name {i+1}", lines=2)
                 notes = gr.Textbox(label=f"Miscellaneous Notes {i+1}", lines=3)
                 responses.append((img, txt, notes))
@@ -278,7 +289,9 @@ with gr.Blocks() as demo:
         submit_btn = gr.Button("Submit All Responses")
         clear_btn = gr.Button("Clear Form")
 
-    def submit_responses(source_img, post_id, *response_data):
+    def submit_responses(
+        source_img, post_id, instruction, simplified_instruction, *response_data
+    ):
         if not source_img:
             gr.Warning("Please select a source image first!")
             return
@@ -309,23 +322,29 @@ with gr.Blocks() as demo:
 
     def clear_form():
         outputs = [None] * (
-            1 + 30
-        )  # 1 source image + 10 triplets of (image, text, notes)
+            1 + 2 + 30
+        )  # source image + 2 instruction fields + 10 triplets
         return outputs
 
     # Connect components
     post_id_dropdown.change(
         fn=load_post_image,
         inputs=[post_id_dropdown],
-        outputs=[source_image] + [comp for triplet in responses for comp in triplet],
+        outputs=[source_image, instruction_text, simplified_instruction_text]
+        + [comp for triplet in responses for comp in triplet],
     )
 
-    submit_inputs = [source_image, post_id_dropdown] + [
-        comp for triplet in responses for comp in triplet
-    ]
+    submit_inputs = [
+        source_image,
+        post_id_dropdown,
+        instruction_text,
+        simplified_instruction_text,
+    ] + [comp for triplet in responses for comp in triplet]
     submit_btn.click(fn=submit_responses, inputs=submit_inputs)
 
-    clear_outputs = [source_image] + [comp for triplet in responses for comp in triplet]
+    clear_outputs = [source_image, instruction_text, simplified_instruction_text] + [
+        comp for triplet in responses for comp in triplet
+    ]
     clear_btn.click(fn=clear_form, outputs=clear_outputs)