Update app.py

app.py

@@ -1,17 +1,16 @@
+
 import streamlit as st
-import requests
 import firebase_admin
-from firebase_admin import credentials, db, auth
+from firebase_admin import credentials, db
 from PIL import Image
 import numpy as np
 from geopy.geocoders import Nominatim
 from tensorflow.keras.applications import MobileNetV2
 from tensorflow.keras.applications.mobilenet_v2 import decode_predictions, preprocess_input
-import json
 import base64
 from io import BytesIO
 
-# Initialize Firebase (Check if already initialized)
+# Initialize Firebase
 if not firebase_admin._apps:
     cred = credentials.Certificate("firebase_credentials.json")  
     firebase_admin.initialize_app(cred, {
@@ -35,57 +34,86 @@ def classify_image_with_mobilenet(image):
         st.error(f"Error during image classification: {e}")
         return {}
 
-# Function to get user's location
-def get_user_location():
-    st.write("Fetching location, please allow location access in your browser.")
-    geolocator = Nominatim(user_agent="binsight")
-    try:
-        ip_info = requests.get("https://ipinfo.io/json").json()
-        loc = ip_info.get("loc", "").split(",")
-        latitude, longitude = loc[0], loc[1] if len(loc) == 2 else (None, None)
-        if latitude and longitude:
-            address = geolocator.reverse(f"{latitude}, {longitude}").address
-            return latitude, longitude, address
-    except Exception as e:
-        st.error(f"Error retrieving location: {e}")
-    return None, None, None
-
 # Function to convert image to Base64
 def convert_image_to_base64(image):
     buffered = BytesIO()
-    image.save(buffered, format="PNG")  # Convert to PNG format
-    img_str = base64.b64encode(buffered.getvalue()).decode()  # Encode as Base64
+    image.save(buffered, format="PNG")  
+    img_str = base64.b64encode(buffered.getvalue()).decode()
     return img_str
 
-# User Login
-st.sidebar.header("User Login")
-user_email = st.sidebar.text_input("Enter your email")
-login_button = st.sidebar.button("Login")
-
-if login_button:
-    if user_email:
-        st.session_state["user_email"] = user_email
-        st.sidebar.success(f"Logged in as {user_email}")
+# Function to get detailed location info from latitude & longitude
+def get_location_details(lat, lon):
+    try:
+        geolocator = Nominatim(user_agent="binsight")
+        location = geolocator.reverse((lat, lon), exactly_one=True)
+        if location:
+            address = location.raw.get('address', {})
+            return {
+                "road": address.get("road", "Unknown"),
+                "city": address.get("city", address.get("town", "Unknown")),
+                "state": address.get("state", "Unknown"),
+                "country": address.get("country", "Unknown"),
+                "full_address": location.address
+            }
+    except Exception as e:
+        st.error(f"Error retrieving location details: {e}")
+    return {}
+
+# JavaScript to get live location and send it to Streamlit
+get_location_js = """
+<script>
+function sendLocation() {
+    navigator.geolocation.getCurrentPosition(
+        (position) => {
+            const latitude = position.coords.latitude;
+            const longitude = position.coords.longitude;
+            const locationData = latitude + "," + longitude;
+            fetch('/_stcore/', {
+                method: 'POST',
+                headers: {'Content-Type': 'application/json'},
+                body: JSON.stringify({latlon: locationData})
+            }).then(response => response.json())
+              .then(data => console.log("Location sent:", data));
+        },
+        (error) => {
+            console.log("Error getting location:", error);
+        }
+    );
+}
+sendLocation();
+</script>
+"""
 
-if "user_email" not in st.session_state:
-    st.warning("Please log in first.")
-    st.stop()
+# Run JavaScript in Streamlit
+st.components.v1.html(get_location_js, height=0)
 
-# Get user location
-latitude, longitude, address = get_user_location()
-if latitude and longitude:
-    st.success(f"Location detected: {address}")
-else:
-    st.warning("Unable to fetch location, please enable location access.")
-    st.stop()
+# Capture location from Streamlit session state
+if "latlon" not in st.session_state:
+    st.session_state.latlon = None
 
 # Streamlit App
 st.title("BinSight: Upload Dustbin Image")
 
+# Fetch live location
+if st.session_state.latlon:
+    try:
+        latitude, longitude = map(float, st.session_state.latlon.split(","))
+        location_details = get_location_details(latitude, longitude)
+        st.success(f"Detected Location: {location_details['full_address']}")
+        st.write(f"**State:** {location_details['state']}")
+        st.write(f"**City:** {location_details['city']}")
+        st.write(f"**Road:** {location_details['road']}")
+        st.write(f"**Country:** {location_details['country']}")
+    except Exception as e:
+        st.error(f"Error processing location: {e}")
+else:
+    st.warning("Fetching live location... Please allow location access.")
+
+# File uploader for dustbin images
 uploaded_file = st.file_uploader("Upload an image of the dustbin", type=["jpg", "jpeg", "png"])
 submit_button = st.button("Analyze and Upload")
 
-if submit_button and uploaded_file:
+if submit_button and uploaded_file and st.session_state.latlon:
     image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image", use_container_width=True)
 
@@ -98,19 +126,15 @@ if submit_button and uploaded_file:
     if classification_results:
         db_ref = db.reference("dustbins")
         dustbin_data = {
-            "user_email": st.session_state["user_email"],
             "latitude": latitude,
             "longitude": longitude,
-            "address": address,
+            "location_details": location_details,
             "classification": classification_results,
-            "allocated_truck": None,
             "status": "Pending",
             "image": image_base64  # Store image as Base64 string
         }
         db_ref.push(dustbin_data)
         st.success("Dustbin data uploaded successfully!")
-        st.write(f"**Location:** {address}")
-        st.write(f"**Latitude:** {latitude}, **Longitude:** {longitude}")
     else:
         st.error("Missing classification details. Cannot upload.")
 
@@ -118,9 +142,6 @@ if submit_button and uploaded_file:
 
 
 
-
-
-
 # best without image
 
 # import streamlit as st