Update app.py

app.py

@@ -90,18 +90,14 @@ with gr.Blocks() as demo:
         show_button = gr.Button("Show Image Pair")
     
     with gr.Row():
-        with gr.Column():
-            img1 = gr.Image(
-                label="Image 1",
-                interactive=False,
-                elem_id="image1"
-            )
-        with gr.Column():
-            img2 = gr.Image(
-                label="Image 2",
-                interactive=False,
-                elem_id="image2"
-            )
+        img1 = gr.Image(
+            label="Image 1",
+            interactive=False
+        )
+        img2 = gr.Image(
+            label="Image 2",
+            interactive=False
+        )
     
     with gr.Row():
         prompt = gr.Textbox(
@@ -111,43 +107,41 @@ with gr.Blocks() as demo:
             placeholder="Select a Pair ID to view details."
         )
     
-    with gr.Accordion("Labels", open=False):
-        with gr.Row():
-            label1_name = gr.Textbox(
-                label="Label 1",
-                interactive=False,
-                placeholder="Label 1"
-            )
-            label1_score = gr.Number(
-                label="Label 1 Score",
-                interactive=False,
-                precision=2
-                # Removed placeholder
-            )
-        with gr.Row():
-            label2_name = gr.Textbox(
-                label="Label 2",
-                interactive=False,
-                placeholder="Label 2"
-            )
-            label2_score = gr.Number(
-                label="Label 2 Score",
-                interactive=False,
-                precision=2
-                # Removed placeholder
-            )
-        with gr.Row():
-            label3_name = gr.Textbox(
-                label="Label 3",
-                interactive=False,
-                placeholder="Label 3"
-            )
-            label3_score = gr.Number(
-                label="Label 3 Score",
-                interactive=False,
-                precision=2
-                # Removed placeholder
-            )
+    with gr.Row():
+        label1 = gr.Textbox(
+            label="Label 1",
+            interactive=False
+        )
+        label1_score = gr.Number(
+            label="Label 1 Score",
+            interactive=False,
+            precision=2,
+            value=0.0  # ✅ Correct usage
+        )
+    
+    with gr.Row():
+        label2 = gr.Textbox(
+            label="Label 2",
+            interactive=False
+        )
+        label2_score = gr.Number(
+            label="Label 2 Score",
+            interactive=False,
+            precision=2,
+            value=0.0  # ✅ Correct usage
+        )
+    
+    with gr.Row():
+        label3 = gr.Textbox(
+            label="Label 3",
+            interactive=False
+        )
+        label3_score = gr.Number(
+            label="Label 3 Score",
+            interactive=False,
+            precision=2,
+            value=0.0  # ✅ Correct usage
+        )
     
     # Define the callback function for the button
     def update_viewer(response):
@@ -155,18 +149,18 @@ with gr.Blocks() as demo:
             img1: response["images"][0],
             img2: response["images"][1],
             prompt: response["prompt"],
-            label1_name: response["labels"]["Label 1"]["name"],
+            label1: response["labels"]["Label 1"]["name"],
             label1_score: response["labels"]["Label 1"]["score"],
-            label2_name: response["labels"]["Label 2"]["name"],
+            label2: response["labels"]["Label 2"]["name"],
             label2_score: response["labels"]["Label 2"]["score"],
-            label3_name: response["labels"]["Label 3"]["name"],
+            label3: response["labels"]["Label 3"]["name"],
             label3_score: response["labels"]["Label 3"]["score"],
         }
     
     show_button.click(
         show_image_pair,
         inputs=[pair_id_input],
-        outputs=[img1, img2, prompt, label1_name, label1_score, label2_name, label2_score, label3_name, label3_score],
+        outputs=[img1, img2, prompt, label1, label1_score, label2, label2_score, label3, label3_score],
         postprocess=update_viewer
     )