Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,16 @@
 import streamlit as st
 from PIL import Image
+import io
 
-
-
-
+# Show logo — make sure RezbinLogo.jpg is in the 'src/' folder in your repo
 st.image("src/RezbinLogo.jpg", width=150)
+
 st.title("Rezbin AI 📷")
 st.write("Upload an image of your waste before throwing it.")
 
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
-    with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
-        tmp_file.write(uploaded_file.getbuffer())
-        tmp_file_path = tmp_file.name
-    img = Image.open(tmp_file_path)
-    st.image(img, caption="Uploaded Image", use_container_width=True)
+    # Open image directly from uploaded file buffer (no temp file needed)
+    img = Image.open(io.BytesIO(uploaded_file.read()))
+    st.image(img, caption="Uploaded Image", use_container_width=True)