Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,11 +1,7 @@
-
-
 import streamlit as st
+import requests
 from PIL import Image
 
-
-
-
 st.image("src/RezbinLogo.jpg", width=150)
 st.title("Rezbin AI 📷")
 st.write("Upload an image of your waste before throwing it.")
@@ -13,6 +9,20 @@ 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:
+    # Show preview
     img = Image.open(uploaded_file)
     st.image(img, caption="Uploaded Image", use_container_width=True)
-    st.markdown('<div class="orange-background">This is a can. You earn 0.50</div>', unsafe_allow_html=True)
+
+    # Send to AI API
+    files = {"file": uploaded_file.getvalue()}
+    try:
+        response = requests.post("https://your-api-url.com/predict", files=files)
+        if response.status_code == 200:
+            result = response.json()
+            label = result.get("label", "Unknown item")
+            reward = result.get("reward", 0)
+            st.markdown(f'<div class="orange-background">This is a {label}. You earn {reward:.2f}</div>', unsafe_allow_html=True)
+        else:
+            st.error(f"Error: {response.status_code} - {response.text}")
+    except Exception as e:
+        st.error(f"Failed to connect to model: {e}")