Add loading of image for traffic analysis

AirbnbMapVisualiser.py

@@ -27,7 +27,7 @@ class AirbnbMapVisualiser:
         try:
             self.neighborhoods = self.get_all_neighborhoods()
             self.cached_listings = {}
-            self.cached_listings["Sha Tin"] = self.get_neighborhood_listings("Sha Tin")
+            self.cached_listings["Southern"] = self.get_neighborhood_listings("Southern")
         except Exception as e:
             print(f"Initialization error: {str(e)}")
             self.neighborhoods = []

TrafficSpot.py

@@ -1,23 +1,62 @@
-import oracledb
+import logging
+from datetime import datetime
 from html import escape
 import folium
+import oracledb
+from datasets import load_dataset
+import base64  # Add this import for base64 encoding
+
 
 class TrafficSpot:
-    def __init__(self, key, latitude, longitude):
+    def __init__(self, key, latitude, longitude, dataset_rows=None):
         self.key = key
         self.latitude = float(latitude) if latitude is not None else None
         self.longitude = float(longitude) if longitude is not None else None
+        self.dataset_rows = dataset_rows or []  # List of matching dataset rows (up to 5)
 
     def is_valid(self):
         return self.latitude is not None and self.longitude is not None
 
     def create_popup_content(self):
-        return f"""
+        html = f"""
             <div style='min-width: 150px; padding: 10px;'>
-                <p style='margin: 5px 0;'><strong>Traffic Camera ID:</strong> {escape(str(self.key))}</p>
-            </div>
+                <p style='margin: 5px 0;'><strong>Location ID:</strong> {escape(str(self.key))}</p>
         """
 
+        if self.dataset_rows:
+            html += "<h4>Recent Records:</h4>"
+            for row in self.dataset_rows:
+                # Convert binary processed_image to base64
+                image_data = row.get('processed_image')
+                image_html = ""
+                if image_data:
+                    try:
+                        # Encode binary data to base64
+                        base64_encoded = base64.b64encode(image_data).decode('utf-8')
+                        # Create img tag with base64 data
+                        image_html = f"""
+                            <img src='data:image/jpeg;base64,{base64_encoded}' 
+                                 style='max-width: 100px; max-height: 100px; margin: 5px 0;' 
+                                 alt='Processed Image'>
+                        """
+                    except Exception as e:
+                        logging.error(f"Error encoding image for {self.key}: {str(e)}")
+                        image_html = "<p>Image load failed</p>"
+
+                html += f"""
+                    <div style='border-top: 1px solid #ccc; padding: 5px 0;'>
+                        <p style='margin: 2px 0;'><strong>Time:</strong> {escape(str(row['capture_time']))}</p>
+                        <p style='margin: 2px 0;'><strong>Vehicles:</strong> {escape(str(row['vehicle_count']))}</p>
+                        <p style='margin: 2px 0;'><strong>Confidence:</strong> {escape(str(row['confidence_score']))}</p>
+                        {image_html}
+                    </div>
+                """
+        else:
+            html += "<p>No recent records available</p>"
+
+        html += "</div>"
+        return html
+
     def add_to_map(self, folium_map):
         if self.is_valid():
             folium.Marker(
@@ -35,21 +74,49 @@ class TrafficSpotManager:
 
     def load_traffic_spots(self):
         try:
+            dataset = load_dataset("slliac/traffic-analysis", split="train")
+            dataset_list = [row for row in dataset]
+            dataset_list.sort(key=lambda x: x['capture_time'], reverse=True)
+
+            dataset_dict = {}
+            for row in dataset_list:
+                loc_id = row['location_id']
+                if loc_id not in dataset_dict:
+                    dataset_dict[loc_id] = []
+                if len(dataset_dict[loc_id]) < 2:
+                    dataset_dict[loc_id].append(row)
+
+            unique_locations = list(dataset_dict.keys())
+            location_ids = tuple(unique_locations) if unique_locations else ('',)
+
             with oracledb.connect(**self.connection_params) as conn:
                 cursor = conn.cursor()
-                cursor.execute("""
+                query = """
                     SELECT KEY, LATITUDE, LONGITUDE 
                     FROM TD_TRAFFIC_CAMERA_LOCATION 
-                    WHERE LATITUDE IS NOT NULL 
+                    WHERE KEY IN ({})
+                    AND LATITUDE IS NOT NULL 
                     AND LONGITUDE IS NOT NULL
-                """)
+                """.format(','.join([':' + str(i + 1) for i in range(len(location_ids))]))
+
+                cursor.execute(query, location_ids)
                 spots = cursor.fetchall()
+
                 self.traffic_spots = [
-                    TrafficSpot(spot[0], spot[1], spot[2])
+                    TrafficSpot(
+                        spot[0],
+                        spot[1],
+                        spot[2],
+                        dataset_dict.get(spot[0])
+                    )
                     for spot in spots
                 ]
+
+                conn.commit()
+                logging.info(f"Loaded {len(self.traffic_spots)} traffic spots")
+
         except Exception as e:
-            print(f"Database error loading traffic spots: {str(e)}")
+            logging.error(f"Error loading traffic spots: {str(e)}")
             self.traffic_spots = []
 
     def add_spots_to_map(self, folium_map):

app.py

@@ -129,7 +129,7 @@ def main():
         neighborhood = st.selectbox(
             "Select Neighborhood",
             options=visualizer.neighborhoods,
-            index=visualizer.neighborhoods.index("Sha Tin") if "Sha Tin" in visualizer.neighborhoods else 0
+            index=visualizer.neighborhoods.index("Southern") if "Southern" in visualizer.neighborhoods else 0
         )
         show_traffic = st.checkbox("Show Traffic Cameras", value=True)
 

requirements.txt

@@ -15,4 +15,5 @@ requests~=2.32.3
 oracledb~=2.5.1
 pillow~=10.4.0
 timm~=1.0.14
-streamlit_folium~=0.24.0
+streamlit_folium~=0.24.0
+watchdog