Update app.py

app.py

@@ -2,18 +2,76 @@ import gradio as gr
 import pandas as pd
 from PIL import Image
 import os
+import sys
+
+# Determine the absolute path to preferences.csv
+current_dir = os.path.dirname(os.path.abspath(__file__))
+csv_path = os.path.join(current_dir, 'preferences.csv')
+
+# Check if preferences.csv exists
+if not os.path.isfile(csv_path):
+    print(f"Error: 'preferences.csv' not found at {csv_path}")
+    sys.exit(1)
 
 # Load metadata
-df = pd.read_csv('preferences.csv')
+try:
+    df = pd.read_csv(csv_path)
+except Exception as e:
+    print(f"Error reading 'preferences.csv': {e}")
+    sys.exit(1)
 
-# Function to display image pairs
+# Function to display image pairs and metadata
 def show_image_pair(pair_id):
-    record = df[df['pair_id'] == pair_id].iloc[0]
-    img1_path = record['image1']
-    img2_path = record['image2']
-    img1 = Image.open(img1_path)
-    img2 = Image.open(img2_path)
-    return img1, img2, record['prompt'], record['label1'], record['label1_score'], record['label2'], record['label2_score'], record['label3'], record['label3_score']
+    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},
+            }
+        }
+    
+    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)},
+        }
+    }
+    
+    # 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."
+    
+    # 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."
+    
+    return response
 
 # Create dropdown options
 pair_ids = df['pair_id'].tolist()
@@ -21,29 +79,95 @@ pair_ids = df['pair_id'].tolist()
 # Define Gradio Interface
 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.Row():
         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")
+            img1 = gr.Image(
+                label="Image 1",
+                interactive=False,
+                elem_id="image1"
+            )
         with gr.Column():
-            img1 = gr.Image(label="Image 1")
-            img2 = gr.Image(label="Image 2")
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt", interactive=False)
-    with gr.Row():
-        label1 = gr.Textbox(label="Label 1", interactive=False)
-        label1_score = gr.Number(label="Label 1 Score", interactive=False)
-    with gr.Row():
-        label2 = gr.Textbox(label="Label 2", interactive=False)
-        label2_score = gr.Number(label="Label 2 Score", interactive=False)
+            img2 = gr.Image(
+                label="Image 2",
+                interactive=False,
+                elem_id="image2"
+            )
+    
     with gr.Row():
-        label3 = gr.Textbox(label="Label 3", interactive=False)
-        label3_score = gr.Number(label="Label 3 Score", interactive=False)
+        prompt = gr.Textbox(
+            label="Prompt",
+            interactive=False,
+            lines=2,
+            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,
+                placeholder="0.0"
+            )
+        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,
+                placeholder="0.0"
+            )
+        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,
+                placeholder="0.0"
+            )
+    
+    # Define the callback function for the button
+    def update_viewer(response):
+        return {
+            img1: response["images"][0],
+            img2: response["images"][1],
+            prompt: response["prompt"],
+            label1_name: response["labels"]["Label 1"]["name"],
+            label1_score: response["labels"]["Label 1"]["score"],
+            label2_name: response["labels"]["Label 2"]["name"],
+            label2_score: response["labels"]["Label 2"]["score"],
+            label3_name: 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, label1_score, label2, label2_score, label3, label3_score]
+        show_image_pair,
+        inputs=[pair_id_input],
+        outputs=[img1, img2, prompt, label1_name, label1_score, label2_name, label2_score, label3_name, label3_score],
+        postprocess=update_viewer
     )
 
 if __name__ == "__main__":