Update app.py

app.py

@@ -8,7 +8,7 @@ import sys
 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)
@@ -29,8 +29,6 @@ def show_image_pair(pair_id):
     
     img1_path = os.path.join(current_dir, record['image1'])
     img2_path = os.path.join(current_dir, record['image2'])
-
-
     
     # Check if images exist
     if not os.path.isfile(img1_path):
@@ -59,33 +57,39 @@ pair_ids = df['pair_id'].tolist()
 with gr.Blocks() as demo:
     gr.Markdown("# MID-Space Dataset Viewer")
     
+    # Pair Selection Row
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
             pair_id_input = gr.Dropdown(label="Select Pair ID", choices=pair_ids, value=pair_ids[0])
+        with gr.Column(scale=1):
             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")
-            img1 = gr.Image(label="Image 1", elem_id="img1", width=400)
-            img2 = gr.Image(label="Image 2", elem_id="img2", width=400)
     
+    # Images Row
     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.Column(scale=1):
+            img1 = gr.Image(label="Image 1")
+        with gr.Column(scale=1):
+            img2 = gr.Image(label="Image 2")
     
+    # Scores Row
     with gr.Row():
-        label2 = gr.Textbox(label="Label 2", interactive=False)
-        label2_score = gr.Number(label="Label 2 Score", interactive=False)
+        # Left Scores
+        with gr.Column(scale=1):
+            label1 = gr.Textbox(label="Label 1", interactive=False)
+            label1_score = gr.Number(label="Label 1 Score", interactive=False)
+            label2 = gr.Textbox(label="Label 2", interactive=False)
+            label2_score = gr.Number(label="Label 2 Score", interactive=False)
+        
+        # Right Scores
+        with gr.Column(scale=1):
+            label3 = gr.Textbox(label="Label 3", interactive=False)
+            label3_score = gr.Number(label="Label 3 Score", interactive=False)
     
+    # Prompt Row
     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)
     
+    # Define Button Click Action
     show_button.click(
         show_image_pair, 
         inputs=[pair_id_input],