Update app.py

app.py

@@ -1,30 +1,38 @@
 import streamlit as st
+import os
 from PIL import Image
-import numpy as np
 
-st.title("Tree Counter App")
+# Define the path to the folder
+FIRE_SHOT_FOLDER = os.path.join(os.getcwd(), 'FireShot')
 
-# Upload image via Streamlit
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+# Ensure the folder exists
+if not os.path.exists(FIRE_SHOT_FOLDER):
+    st.error("FireShot folder does not exist!")
+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:
-    # Open the image
     img = Image.open(uploaded_file)
-    st.image(img, caption="Uploaded Image.", use_column_width=True)
+    st.image(img, caption="Uploaded Image", use_column_width=True)
     
-    # Check image dimensions
-    width, height = img.size
-    if width > 1024 or height > 1024:
-        st.error("Image is too large. Please crop it.")
-    else:
-        # Simulate tree detection using random number
-        trees = np.random.randint(0, 100)
-        st.success(f"Number of Trees: {trees}")
+    # 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}")
 
-# Image cropping functionality
-if uploaded_file is not None and st.button("Crop Image to 1024x1024"):
-    cropped_img = img.crop((0, 0, min(1024, img.width), min(1024, img.height)))
-    st.image(cropped_img, caption="Cropped Image.", use_column_width=True)
-    cropped_img.save("cropped_image.jpg")
-    st.success("Image cropped successfully.")
+# 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.")