Update app.py

app.py

@@ -1,13 +1,14 @@
 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
+from tensorflow.keras import backend as K
+from PIL import ExifTags
 
 # Initialize Firebase
 if not firebase_admin._apps:
@@ -16,6 +17,9 @@ if not firebase_admin._apps:
         'databaseURL': 'https://binsight-beda0-default-rtdb.asia-southeast1.firebasedatabase.app/'
     })
 
+# Clear previous TensorFlow session to avoid session-related errors
+K.clear_session()
+
 # Load MobileNetV2 pre-trained model
 mobilenet_model = MobileNetV2(weights="imagenet")
 
@@ -48,6 +52,22 @@ def get_user_location():
         st.error(f"Error retrieving location: {e}")
     return None, None, None
 
+# Function to extract GPS location from EXIF data (if available)
+def get_image_location(image):
+    try:
+        exif_data = image._getexif()
+        if exif_data is not None:
+            for tag, value in exif_data.items():
+                if tag == 34853:  # GPS tag
+                    gps_info = value
+                    latitude = gps_info[2]
+                    longitude = gps_info[4]
+                    return latitude, longitude
+        return None, None
+    except Exception as e:
+        st.error(f"Error retrieving EXIF data: {e}")
+    return None, None
+
 # User Login
 st.sidebar.header("User Login")
 user_email = st.sidebar.text_input("Enter your email")
@@ -65,7 +85,7 @@ if "user_email" not in st.session_state:
 # Get user location and display details
 latitude, longitude, address = get_user_location()
 if latitude and longitude:
-    st.success(f"Location detected: {address}")
+    st.success(f"User's live location detected: {address}")
 else:
     st.warning("Unable to fetch location, please ensure location access is enabled.")
     st.stop()
@@ -80,6 +100,16 @@ if submit_button and uploaded_file:
     image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image", use_container_width=True)
 
+    # Get the GPS data from the image's EXIF
+    img_lat, img_lon = get_image_location(image)
+    if img_lat and img_lon:
+        st.write(f"**Image Location (from EXIF):** Latitude: {img_lat}, Longitude: {img_lon}")
+        geolocator = Nominatim(user_agent="binsight")
+        img_address = geolocator.reverse(f"{img_lat}, {img_lon}").address
+        st.write(f"**Image Address (from EXIF):** {img_address}")
+    else:
+        st.warning("No GPS information found in the image EXIF.")
+
     classification_results = classify_image_with_mobilenet(image)
     
     if classification_results:
@@ -95,7 +125,7 @@ if submit_button and uploaded_file:
         }
         db_ref.push(dustbin_data)
         st.success("Dustbin data uploaded successfully!")
-        st.write(f"**Location:** {address}")
+        st.write(f"**User Location:** {address}")
         st.write(f"**Latitude:** {latitude}, **Longitude:** {longitude}")
     else:
         st.error("Missing classification details. Cannot upload.")
@@ -104,8 +134,6 @@ if submit_button and uploaded_file:
 
 
 
-
-
 # best with firebase but below code is not giving correct location of user.
 
 # import streamlit as st