Update app.py

app.py

@@ -9,6 +9,33 @@ from keras.layers import BatchNormalization, DepthwiseConv2D, TFSMLayer
 import os
 from io import BytesIO
 import base64
+st.markdown(
+    """
+    <style>
+    .flex-row {
+        display: flex;
+        gap: 2rem;
+        align-items: stretch; /* ensures equal height */
+    }
+    .flex-row > div {
+        flex: 1;
+        display: flex;
+        flex-direction: column;
+    }
+    .overlay {
+        background-color: rgba(255, 255, 255, 0.85);
+        padding: 1rem;
+        border-radius: 10px;
+        overflow-y: auto;
+        color: #333;
+        font-size: 16px;
+        line-height: 1.5;
+        flex-grow: 1;  /* fills the column height */
+    }
+    </style>
+    """,
+    unsafe_allow_html=True,
+)
 
 # --- Fix deserialization issues ---
 original_bn = BatchNormalization.from_config
@@ -220,10 +247,13 @@ def show_lime(img, model, pred_idx, pred_label, all_probs):
         buf.seek(0)
         lime_data = buf.getvalue()
 
+        # Wrap columns inside a flex container for equal height
+        st.markdown('<div class="flex-row">', unsafe_allow_html=True)
+
         col1, col2 = st.columns(2)
         with col1:
             st.markdown("### 📍 LIME Explanation")
-            st.image(lime_data, width=224, output_format="PNG")  # 👈 Small LIME image
+            st.image(lime_data, width=224, output_format="PNG")
             st.download_button(
                 "📥 Download LIME Image",
                 lime_data,
@@ -232,10 +262,13 @@ def show_lime(img, model, pred_idx, pred_label, all_probs):
             )
         with col2:
             st.markdown(
-                f"<div class='overlay'>{explanation_text.get(pred_label, 'No explanation available.')}</div>",
+                f'<div class="overlay">{explanation_text.get(pred_label, "No explanation available.")}</div>',
                 unsafe_allow_html=True
             )
 
+        st.markdown('</div>', unsafe_allow_html=True)
+
+
 # --- Streamlit App UI ---
 st.set_page_config(page_title="👁 Retina Classifier with LIME", layout="wide")
 st.title("👁 Retina Disease Classifier with LIME Explanation")