backup

app.py

@@ -151,7 +151,7 @@ def load_question_data(question_id):
 
 def load_post_image(post_id):
     if not post_id:
-        return [None] * 21  # source image + 10 pairs of (image, text)
+        return [None] * 31  # source image + 10 triplets of (image, text, notes)
 
     idx = POST_ID_TO_ID_MAP[post_id]
     source_image = VALID_DATASET[idx]["image"]
@@ -165,21 +165,25 @@ def load_post_image(post_id):
             metadata = json.load(f)
 
         # Initialize response data
-        responses = [(None, "")] * 10
+        responses = [(None, "", "")] * 10  # Initialize with empty notes
 
         # Fill in existing responses
         for response in metadata["responses"]:
             idx = response["response_id"]
             if idx < 10:  # Ensure we don't exceed our UI limit
                 image_path = os.path.join(post_folder, response["image_path"])
-                responses[idx] = (image_path, response["answer_text"])
+                responses[idx] = (
+                    image_path,
+                    response["answer_text"],
+                    response.get("notes", ""),  # Get notes with empty string as default
+                )
 
         # Flatten responses for output
-        flat_responses = [item for pair in responses for item in pair]
+        flat_responses = [item for triplet in responses for item in triplet]
         return [source_image] + flat_responses
 
     # If no existing responses, return source image and empty responses
-    return [source_image] + [None] * 20
+    return [source_image] + [None] * 30
 
 
 def generate_json_files(source_image, responses, post_id):
@@ -188,7 +192,7 @@ def generate_json_files(source_image, responses, post_id):
 
     Args:
         source_image: Path to the source image
-        responses: List of (image, answer) tuples
+        responses: List of (image, answer, notes) tuples
         post_id: The post ID from the dataset
     """
     # Create parent data folder if it doesn't exist
@@ -210,7 +214,7 @@ def generate_json_files(source_image, responses, post_id):
 
     # Create responses data
     responses_data = []
-    for idx, (response_image, answer_text) in enumerate(responses):
+    for idx, (response_image, answer_text, notes) in enumerate(responses):
         if response_image and answer_text:  # Only process if both image and text exist
             response_folder = os.path.join(post_folder, f"response_{idx}")
             os.makedirs(response_folder)
@@ -227,6 +231,7 @@ def generate_json_files(source_image, responses, post_id):
                 {
                     "response_id": idx,
                     "answer_text": answer_text,
+                    "notes": notes,
                     "image_path": f"response_{idx}/response_image.png",
                 }
             )
@@ -266,7 +271,8 @@ with gr.Blocks() as demo:
             with gr.Tab(f"Response {i+1}"):
                 img = gr.Image(label=f"Response Image {i+1}", type="filepath")
                 txt = gr.Textbox(label=f"Model Name {i+1}", lines=2)
-                responses.append((img, txt))
+                notes = gr.Textbox(label=f"Miscellaneous Notes {i+1}", lines=3)
+                responses.append((img, txt, notes))
 
     with gr.Row():
         submit_btn = gr.Button("Submit All Responses")
@@ -281,38 +287,45 @@ with gr.Blocks() as demo:
             gr.Warning("Please select a post ID first!")
             return
 
-        # Convert flat response_data into pairs of (image, text)
-        response_pairs = list(zip(response_data[::2], response_data[1::2]))
+        # Convert flat response_data into triplets of (image, text, notes)
+        response_triplets = list(
+            zip(response_data[::3], response_data[1::3], response_data[2::3])
+        )
 
-        # Filter out empty responses
+        # Filter out empty responses (only checking image and model name, notes are optional)
         valid_responses = [
-            (img, txt) for img, txt in response_pairs if img is not None and txt
+            (img, txt, notes)
+            for img, txt, notes in response_triplets
+            if img is not None and txt
         ]
 
         if not valid_responses:
             gr.Warning("Please provide at least one response (image + text)!")
             return
 
+        # Modify generate_json_files call to handle notes
         generate_json_files(source_img, valid_responses, post_id)
         gr.Info("Responses saved successfully! 🎉")
 
     def clear_form():
-        outputs = [None] * (1 + 20)  # 1 source image + 10 pairs of (image, text)
+        outputs = [None] * (
+            1 + 30
+        )  # 1 source image + 10 triplets of (image, text, notes)
         return outputs
 
     # Connect components
     post_id_dropdown.change(
         fn=load_post_image,
         inputs=[post_id_dropdown],
-        outputs=[source_image] + [comp for pair in responses for comp in pair],
+        outputs=[source_image] + [comp for triplet in responses for comp in triplet],
     )
 
     submit_inputs = [source_image, post_id_dropdown] + [
-        comp for pair in responses for comp in pair
+        comp for triplet in responses for comp in triplet
     ]
     submit_btn.click(fn=submit_responses, inputs=submit_inputs)
 
-    clear_outputs = [source_image] + [comp for pair in responses for comp in pair]
+    clear_outputs = [source_image] + [comp for triplet in responses for comp in triplet]
     clear_btn.click(fn=clear_form, outputs=clear_outputs)