Update app.py

app.py

@@ -340,34 +340,25 @@ with tab2:
 # Tab 3: User Input and Results
 with tab3:
 
-    # Check if 'image_paths' is in session state
-    if 'image_paths' in st.session_state:
-        # Grid Layout Section
-        st.header("Image Grid Layout")
-        for i in range(0, len(st.session_state['image_paths']), 4):
-            cols = st.columns(4)
-            for j in range(4):
-                idx = i + j
-                if idx < len(st.session_state['image_paths']):
-                    with cols[j]:
-                        st.image(st.session_state['image_paths'][idx], use_column_width=True)
-                        if st.button(f"Details {idx}", key=f"button_{idx}"):
-                            st.write(f"You clicked on image {idx}")
+
+    if 'image_paths' in st.session_state and st.session_state['image_paths']:
+            # Create a slider for image selection
+            image_index = st.select_slider(
+                "Choose an image",
+                options=list(range(len(st.session_state['image_paths']))),
+                format_func=lambda x: f"Image {x + 1}"
+            )
     
-        # Slider-Based Single Image View Section
-        st.header("Slider-Based Image Viewer")
-        # Create a slider
-        image_idx = st.slider("Select an Image", 0, len(st.session_state['image_paths']) - 1, 0)
-        # Display the selected image
-        st.image(st.session_state['image_paths'][image_idx], use_column_width=True)
-        # Optional: Add a details button for the selected image
-        if st.button(f"Details for Image {image_idx}", key=f"button_details_{image_idx}"):
-            st.write(f"You clicked on image {image_idx}")
+            # Display the selected image
+            st.image(st.session_state['image_paths'][image_index], use_column_width=True)
     
+            # Button for actions related to the selected image
+            if st.button("Details", key=f"details_{image_index}"):
+                st.write(f"You clicked on image {image_index + 1}")
+                # Add more actions for the click event here
 
 
-'''
-    
+'''    
     if 'image_paths' in st.session_state:
         for i in range(0, len(st.session_state['image_paths']), 4):
             cols = st.columns(4)
@@ -379,4 +370,5 @@ with tab3:
                         if st.button(f"Details {idx}", key=f"button_{idx}"):
                             st.write(f"You clicked on image {idx}")
                             # Add more actions for the click event here
-'''
+
+'''