Update app.py

app.py

@@ -182,7 +182,7 @@ def main():
     # Define page navigation using the sidebar
     page = st.sidebar.selectbox(
         "Choose your action:",
-        ("Home", "Upload Images", "Social Media URL", "Web Page URL"),
+        ("Home", "Images", "Social Media", "Web Pages"),
         index=0  # Default to Home
     )
 
@@ -235,33 +235,34 @@ def social_media_page():
     st.header("Social Media Analyser")
     social_media_url = st.text_input("Enter a social media URL to analyse:", key='social_media_url_input')
     if social_media_url:
-        media_data = get_media(social_media_url)  # Assume this function is defined elsewhere
+        media_data = get_media(social_media_url)
         if media_data:
-            # Display the full text of the first media found
             full_text = media_data[0][1]
             st.subheader("Full Text")
             st.write(full_text)
-            most_used_location = most_frequent_locations(full_text)
+            most_used_location, most_common_locations = most_frequent_locations(full_text)
             st.subheader("Most Frequent Location")
             st.write(most_used_location)
 
-            # Process and display each image found in the media data
             for idx, (media_url, _) in enumerate(media_data, start=1):
                 st.subheader(f"Image {idx}")
                 response = requests.get(media_url)
                 if response.status_code == 200:
                     image = Image.open(BytesIO(response.content)).convert('RGB')
                     st.image(image, caption=f'Image from URL: {media_url}', use_column_width=True)
-                    model = load_geoloc_model()  # Assume this function is defined elsewhere
+                    model = load_geoloc_model()
                     if model:
-                        result = predict_location(image, model)  # Assume this function is defined elsewhere
+                        result = predict_location(image, model)
                         if result:
                             gps_degrees, location_query, city_geojson, processing_time = result
-                            st.write(
-                                f"City: {location_query['name']}, Region: {location_query['admin1']}, Country: {location_query['cc']}")
+                            location_name = f"{location_query['name']}, {location_query['admin1']}, {location_query['cc']}"
+                            st.write(f"City: {location_query['name']}, Region: {location_query['admin1']}, Country: {location_query['cc']}")
                             if city_geojson:
                                 display_map(city_geojson, gps_degrees)
                                 st.write(f"Processing Time (seconds): {processing_time}")
+                            # Check for match and notify
+                            if location_query['name'] in most_common_locations:
+                                st.success(f"The predicted location {location_query['name']} matches the most frequently mentioned location!")
                 else:
                     st.error(f"Failed to fetch image at URL {media_url}: HTTP {response.status_code}")