revert visual document (streamlit deprications)

src/ui_components/visual_documents.py

@@ -1,6 +1,5 @@
 """
 Visual Document Display Components
-
 UI components for displaying visual search results with enhanced metadata.
 Includes saliency map visualization for tile-aware ColPali embeddings.
 """
@@ -185,6 +184,27 @@ def display_visual_document_details(
     """
     st.markdown("### 📄 Document Details")
     
+    # CSS to stabilize layout and prevent reflow loops on HF Spaces
+    # The key issues: proportional columns + responsive images = infinite resize loop
+    st.markdown("""
+    <style>
+    /* Stabilize image containers - fixed aspect ratio prevents reflow */
+    [data-testid="stImage"] {
+        min-height: 200px;
+        max-height: 600px;
+    }
+    /* Prevent column width oscillation */
+    [data-testid="column"] {
+        min-width: 0;
+        overflow: hidden;
+    }
+    /* Stabilize expander content */
+    [data-testid="stExpander"] > div {
+        overflow: hidden;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+    
     # Compute max score for normalization (display only)
     all_scores = [getattr(doc, 'score', 0.0) for doc in sources]
     max_score = max(all_scores) if all_scores else 1.0
@@ -366,7 +386,7 @@ def display_visual_document_details(
                             st.markdown("**📄 Original**")
                             if image_url and isinstance(image_url, str) and image_url.startswith('http'):
                                 try:
-                                    st.image(image_url, use_container_width=True, caption=f"Page {page_number}")
+                                    st.image(image_url, width=599, caption=f"Page {page_number}")
                                 except Exception as e:
                                     st.error(f"Failed to load image: {e}")
                             else:
@@ -478,7 +498,7 @@ def display_visual_document_details(
                                             pass
 
                                     if saliency_img:
-                                        st.image(saliency_img, use_container_width=True, caption=f"Relevance heatmap")
+                                        st.image(saliency_img, width=599, caption=f"Relevance heatmap")
                                         logger.info(f"✅ Saliency map displayed for doc {i+1}")
                                     else:
                                         logger.warning(f"Saliency generation returned None for doc {i+1}")
@@ -492,7 +512,7 @@ def display_visual_document_details(
                         # Can't generate saliency - just show original image (full width)
                         if image_url and isinstance(image_url, str) and image_url.startswith('http'):
                             try:
-                                st.image(image_url, width=850, caption=f"Page {page_number}")
+                                st.image(image_url, width=599, caption=f"Page {page_number}")
                             except Exception as e:
                                 st.error(f"Failed to load image: {e}")
                         
@@ -505,7 +525,7 @@ def display_visual_document_details(
                 elif show_images:
                     if image_url and isinstance(image_url, str) and image_url.startswith('http'):
                         try:
-                            st.image(image_url, width=850, caption=f"Page {page_number}")
+                            st.image(image_url, width=599, caption=f"Page {page_number}")
                         except Exception as e:
                             st.error(f"Failed to load image: {e}")
                     else:
@@ -590,4 +610,3 @@ def display_visual_search_results(
     
     if len(sources) > max_display:
         st.info(f"💡 {len(sources) - max_display} more results not shown")
-