Update app.py

app.py

@@ -89,44 +89,6 @@ st.markdown("""
         box-shadow: 0 5px 15px rgba(102, 126, 234, 0.4);
     }
     
-    /* Navigation buttons styling */
-    div[data-testid="column"] .stButton > button {
-        padding: 0.5rem 1rem;
-        font-size: 0.9rem;
-        min-width: 50px;
-    }
-    
-    /* Page number buttons */
-    .stButton > button[kind="primary"] {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        font-weight: 700;
-    }
-    
-    .stButton > button[kind="secondary"] {
-        background: white;
-        color: #667eea;
-        border: 2px solid #667eea;
-        font-weight: 500;
-    }
-    
-    /* Example image containers */
-    .example-image-container {
-        aspect-ratio: 1 / 1;
-        overflow: hidden;
-        border-radius: 10px;
-        margin-bottom: 0.5rem;
-        background: #f0f2f6;
-        display: flex;
-        align-items: center;
-        justify-content: center;
-    }
-    
-    .example-image-container img {
-        width: 100%;
-        height: 100%;
-        object-fit: cover;
-    }
-    
     /* Image container */
     .uploaded-image {
         border-radius: 15px;
@@ -177,7 +139,7 @@ def load_model():
         learner = load_learner(MODEL_PATH)
         return learner
     except Exception as e:
-        st.error(f"⚠️ Error loading model:\n\n{e}")
+        st.error(f⚠️ Error loading model:\n\n{e}")
         return None
 
 def predict(learner, img_bytes: bytes):
@@ -210,72 +172,15 @@ def main():
         example_images = list(examples_path.glob("*.jpg")) + list(examples_path.glob("*.png")) + list(examples_path.glob("*.jpeg"))
         
         if example_images:
-            # Initialize page number in session state
-            if 'example_page' not in st.session_state:
-                st.session_state.example_page = 0
+            # Display examples in a grid
+            cols = st.columns(min(5, len(example_images)))
             
-            # Number of images per page
-            images_per_page = 5
-            total_pages = (len(example_images) + images_per_page - 1) // images_per_page
-            
-            # Get current page images
-            start_idx = st.session_state.example_page * images_per_page
-            end_idx = min(start_idx + images_per_page, len(example_images))
-            current_page_images = example_images[start_idx:end_idx]
-            
-            # Display examples in a grid (always 5 columns for consistency)
-            cols = st.columns(5)
-            
-            for idx in range(5):
+            for idx, img_path in enumerate(example_images[:5]):  # Show max 5 examples
                 with cols[idx]:
-                    if idx < len(current_page_images):
-                        img_path = current_page_images[idx]
-                        # Use HTML container for consistent sizing
-                        st.markdown(
-                            f'<div class="example-image-container">',
-                            unsafe_allow_html=True
-                        )
-                        st.image(str(img_path), use_container_width=True)
-                        st.markdown('</div>', unsafe_allow_html=True)
-                        st.caption(img_path.stem)
-                        if st.button(f"Use", key=f"example_{start_idx + idx}"):
-                            # Store the selected example in session state
-                            st.session_state.example_image = img_path
-                            st.rerun()
-                    else:
-                        # Empty placeholder to maintain grid consistency
-                        st.markdown('<div style="height: 200px;"></div>', unsafe_allow_html=True)
-            
-            # Pagination controls
-            if total_pages > 1:
-                st.markdown("<br>", unsafe_allow_html=True)
-                page_cols = st.columns([1, 6, 1])
-                
-                with page_cols[0]:
-                    if st.button("⬅️", disabled=(st.session_state.example_page == 0), help="Previous page"):
-                        st.session_state.example_page -= 1
-                        st.rerun()
-                
-                with page_cols[1]:
-                    # Page number buttons centered
-                    center_cols = st.columns([1, 2, 1])
-                    with center_cols[1]:
-                        page_buttons = st.columns(total_pages)
-                        for page_num in range(total_pages):
-                            with page_buttons[page_num]:
-                                if st.button(
-                                    f"{page_num + 1}", 
-                                    key=f"page_{page_num}",
-                                    type="primary" if page_num == st.session_state.example_page else "secondary",
-                                    use_container_width=True
-                                ):
-                                    st.session_state.example_page = page_num
-                                    st.rerun()
-                
-                with page_cols[2]:
-                    if st.button("➡️", disabled=(st.session_state.example_page == total_pages - 1), help="Next page"):
-                        st.session_state.example_page += 1
-                        st.rerun()
+                    st.image(str(img_path), use_container_width=True, caption=img_path.stem)
+                    if st.button(f"Use", key=f"example_{idx}"):
+                        # Store the selected example in session state
+                        st.session_state.example_image = img_path
         else:
             st.info("No example images found in the 'examples' folder.")
     else: