Update app.py

app.py

@@ -2,8 +2,7 @@ import streamlit as st
 import os
 from PIL import Image
 
-
-# Define the path to the folder
+# Define the path to the folder where images are stored
 FIRE_SHOT_FOLDER = os.path.join(os.getcwd(), 'FireShot')
 
 # Ensure the folder exists
@@ -12,32 +11,18 @@ if not os.path.exists(FIRE_SHOT_FOLDER):
 else:
     st.success(f"FireShot folder found at {FIRE_SHOT_FOLDER}")
 
-# Display a file uploader
-uploaded_file = st.file_uploader("Browse images from FireShot", type=["jpg", "png", "jpeg"])
-
-# If the user uploads a file, save it in the FireShot folder
-if uploaded_file is not None:
-
-    img = Image.open(uploaded_file)
-    st.image(img, caption="Uploaded Image", use_column_width=True)
-    
-    # Save the image to FireShot folder
-    file_path = os.path.join(FIRE_SHOT_FOLDER, uploaded_file.name)
-    img.save(file_path)
-    st.success(f"Image saved to {file_path}")
+# Get list of images from the FireShot folder
+image_files = [f for f in os.listdir(FIRE_SHOT_FOLDER) if f.endswith(('jpg', 'jpeg', 'png'))]
 
+if image_files:
+    # Create a selectbox for the user to select an image
+    selected_image = st.selectbox("Select an image from the FireShot folder", image_files)
 
+    # Load and display the selected image
+    if selected_image:
+        img_path = os.path.join(FIRE_SHOT_FOLDER, selected_image)
+        img = Image.open(img_path)
+        st.image(img, caption=f"Selected Image: {selected_image}", use_column_width=True)
 
-
-
-# Optionally, display all images from the FireShot folder
-if st.button("Show all images in FireShot folder"):
-    image_files = [f for f in os.listdir(FIRE_SHOT_FOLDER) if f.endswith(('jpg', 'jpeg', 'png'))]
-    
-    if image_files:
-        for image_file in image_files:
-            img_path = os.path.join(FIRE_SHOT_FOLDER, image_file)
-            img = Image.open(img_path)
-            st.image(img, caption=image_file, use_column_width=True)
-    else:
-        st.info("No images found in FireShot folder.")
+else:
+    st.info("No images found in FireShot folder.")