Update app.py

app.py

@@ -20,58 +20,35 @@ except Exception as e:
     print(f"Error reading 'preferences.csv': {e}")
     sys.exit(1)
 
-# Function to display image pairs and metadata
+# Function to display image pairs
 def show_image_pair(pair_id):
     try:
         record = df[df['pair_id'] == pair_id].iloc[0]
     except IndexError:
-        return {
-            "images": [None, None],
-            "prompt": "Invalid Pair ID.",
-            "labels": {
-                "Label 1": {"name": "", "score": 0.0},
-                "Label 2": {"name": "", "score": 0.0},
-                "Label 3": {"name": "", "score": 0.0},
-            }
-        }
+        return None, None, "Invalid Pair ID.", "", 0.0, "", 0.0, "", 0.0
     
     img1_path = os.path.join(current_dir, record['image1'])
     img2_path = os.path.join(current_dir, record['image2'])
     
-    # Initialize default response
-    response = {
-        "images": [None, None],
-        "prompt": record.get('prompt', ''),
-        "labels": {
-            "Label 1": {"name": record.get('label1', ''), "score": record.get('label1_score', 0.0)},
-            "Label 2": {"name": record.get('label2', ''), "score": record.get('label2_score', 0.0)},
-            "Label 3": {"name": record.get('label3', ''), "score": record.get('label3_score', 0.0)},
-        }
-    }
+    # Check if images exist
+    if not os.path.isfile(img1_path):
+        return None, None, "Image 1 not found.", "", 0.0, "", 0.0, "", 0.0
+    if not os.path.isfile(img2_path):
+        return None, None, "Image 2 not found.", "", 0.0, "", 0.0, "", 0.0
     
-    # Load Image 1
-    if os.path.isfile(img1_path):
-        try:
-            img1 = Image.open(img1_path)
-            response["images"][0] = img1
-        except Exception as e:
-            print(f"Error opening Image 1: {e}")
-            response["images"][0] = None
-    else:
-        response["prompt"] = "Image 1 not found."
+    try:
+        img1 = Image.open(img1_path)
+    except Exception as e:
+        img1 = None
+        print(f"Error opening Image 1: {e}")
     
-    # Load Image 2
-    if os.path.isfile(img2_path):
-        try:
-            img2 = Image.open(img2_path)
-            response["images"][1] = img2
-        except Exception as e:
-            print(f"Error opening Image 2: {e}")
-            response["images"][1] = None
-    else:
-        response["prompt"] = "Image 2 not found."
+    try:
+        img2 = Image.open(img2_path)
+    except Exception as e:
+        img2 = None
+        print(f"Error opening Image 2: {e}")
     
-    return response
+    return img1, img2, record['prompt'], record['label1'], record['label1_score'], record['label2'], record['label2_score'], record['label3'], record['label3_score']
 
 # Create dropdown options
 pair_ids = df['pair_id'].tolist()
@@ -81,87 +58,34 @@ with gr.Blocks() as demo:
     gr.Markdown("# MID-Space Dataset Viewer")
     
     with gr.Row():
-        pair_id_input = gr.Dropdown(
-            label="Select Pair ID",
-            choices=pair_ids,
-            value=pair_ids[0],
-            interactive=True
-        )
-        show_button = gr.Button("Show Image Pair")
+        with gr.Column():
+            pair_id_input = gr.Dropdown(label="Select Pair ID", choices=pair_ids, value=pair_ids[0])
+            show_button = gr.Button("Show Image Pair")
+        
+        # New Row for side-by-side images
+        with gr.Row():
+            img1 = gr.Image(label="Image 1")
+            img2 = gr.Image(label="Image 2")
     
     with gr.Row():
-        img1 = gr.Image(
-            label="Image 1",
-            interactive=False
-        )
-        img2 = gr.Image(
-            label="Image 2",
-            interactive=False
-        )
+        prompt = gr.Textbox(label="Prompt", interactive=False)
     
     with gr.Row():
-        prompt = gr.Textbox(
-            label="Prompt",
-            interactive=False,
-            lines=2,
-            placeholder="Select a Pair ID to view details."
-        )
+        label1 = gr.Textbox(label="Label 1", interactive=False)
+        label1_score = gr.Number(label="Label 1 Score", interactive=False)
     
     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
-        )
+        label2 = gr.Textbox(label="Label 2", interactive=False)
+        label2_score = gr.Number(label="Label 2 Score", interactive=False)
     
     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):
-        return {
-            img1: response["images"][0],
-            img2: response["images"][1],
-            prompt: response["prompt"],
-            label1: response["labels"]["Label 1"]["name"],
-            label1_score: response["labels"]["Label 1"]["score"],
-            label2: response["labels"]["Label 2"]["name"],
-            label2_score: response["labels"]["Label 2"]["score"],
-            label3: response["labels"]["Label 3"]["name"],
-            label3_score: response["labels"]["Label 3"]["score"],
-        }
+        label3 = gr.Textbox(label="Label 3", interactive=False)
+        label3_score = gr.Number(label="Label 3 Score", interactive=False)
     
     show_button.click(
-        show_image_pair,
-        inputs=[pair_id_input],
-        outputs=[img1, img2, prompt, label1, label1_score, label2, label2_score, label3, label3_score],
-        postprocess=update_viewer
+        show_image_pair, 
+        inputs=[pair_id_input], 
+        outputs=[img1, img2, prompt, label1, label1_score, label2, label2_score, label3, label3_score]
     )
 
 if __name__ == "__main__":