Update app.py

app.py

@@ -1,4 +1,3 @@
-#
 import streamlit as st
 import os
 import glob
@@ -109,15 +108,15 @@ upload_option = st.radio("Select Upload Type", ["Single Image Upload", "Batch Im
 # Single Image Upload
 if upload_option == "Single Image Upload":
     st.subheader("Upload a Single Vehicle Image")
-    uploaded_image = st.file_uploader("Choose a JPEG image", type="jpeg")
+    uploaded_image = st.file_uploader("Choose an Image (JPEG, PNG, GIF, BMP, etc.)", type=["jpeg", "png", "gif", "bmp", "jpg"])
 
     if uploaded_image is not None:
         # Display the uploaded image
         image = PILImage.open(uploaded_image)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption="Uploaded Image", use_container_width=True)
 
         # Convert the uploaded image to base64
-        image_path = "/tmp/uploaded_image.jpeg"
+        image_path = "/tmp/uploaded_image" + os.path.splitext(uploaded_image.name)[1]
         with open(image_path, "wb") as f:
             f.write(uploaded_image.getbuffer())
 
@@ -130,13 +129,13 @@ if upload_option == "Single Image Upload":
 
         # Optionally, display more vehicle images from the folder
         img_dir = "/content/images"
-        image_paths = glob.glob(os.path.join(img_dir, "*.jpeg"))
+        image_paths = glob.glob(os.path.join(img_dir, "*.*"))
         display_image_grid(image_paths)
 
 # Batch Images Upload
 elif upload_option == "Batch Images Upload":
     st.sidebar.header("Batch Image Upload")
-    batch_images = st.sidebar.file_uploader("Upload Images", type="jpeg", accept_multiple_files=True)
+    batch_images = st.sidebar.file_uploader("Upload Images (JPEG, PNG, GIF, BMP, etc.)", type=["jpeg", "png", "gif", "bmp", "jpg"], accept_multiple_files=True)
 
     if batch_images:
         batch_input = [{"image_path": f"/tmp/{file.name}"} for file in batch_images]