Update app.py

app.py

@@ -837,7 +837,25 @@ if __name__ == "__main__":
     print(f"Low resolution folder:  {LOW_RES_FOLDER} [{'✓' if os.path.isdir(LOW_RES_FOLDER) else '✗'}]")
     print("=" * 60)
     
-    with gr.Blocks(title=STUDY_TITLE, theme=gr.themes.Soft()) as demo:
+    custom_css = f"""
+    .cand-a img {{
+        height: {CANDIDATE_IMAGE_HEIGHT_STEP_A}px !important;
+        width: auto !important;
+        object-fit: contain;
+    }}
+    .cand-b img {{
+        height: {CANDIDATE_IMAGE_HEIGHT_STEP_B}px !important;
+        width: auto !important;
+        object-fit: contain;
+    }}
+    .ref-a img {{
+        height: {REFERENCE_IMAGE_HEIGHT}px !important;
+        width: auto !important;
+        object-fit: contain;
+    }}
+    """
+
+    with gr.Blocks(title=STUDY_TITLE, theme=gr.themes.Soft(), css=custom_css) as demo:
 
         # Hidden state - Added user_assigned_images
         state_uid = gr.State("")
@@ -897,7 +915,8 @@ if __name__ == "__main__":
                                 show_download_button=False,
                                 type="pil",
                                 image_mode="RGB",
-                                height=CANDIDATE_IMAGE_HEIGHT_STEP_A
+                                height= None, #CANDIDATE_IMAGE_HEIGHT_STEP_A,
+                                elem_classes=["cand-a"],
                             )
                             a_imgs.append(img)
                 
@@ -915,7 +934,8 @@ if __name__ == "__main__":
                         show_download_button=False,
                         type="pil",
                         image_mode="RGB",
-                        height=REFERENCE_IMAGE_HEIGHT
+                        height= None, #REFERENCE_IMAGE_HEIGHT,
+                        elem_classes=["ref-a"]
                     )
                 
                 # Create image components dynamically in rows
@@ -931,7 +951,8 @@ if __name__ == "__main__":
                                 show_download_button=False,
                                 type="pil",
                                 image_mode="RGB",
-                                height=CANDIDATE_IMAGE_HEIGHT_STEP_B
+                                height= None, #CANDIDATE_IMAGE_HEIGHT_STEP_B
+                                elem_classes=["cand-b"]
                             )
                             b_imgs.append(img)