Update app.py

app.py

@@ -5,6 +5,120 @@ import cv2
 import gradio as gr
 from tensorflow import keras
 from keras.models import load_model
+import folium
+
+
+
+
+# CODE FOR MAPS
+def convert_coordinates(coord_str):
+    # Regular expression pattern to match the coordinate format
+    pattern = r"(\d+)°(\d+)'(\d+(?:\.\d+)?)\"([NS])\s+(\d+)°(\d+)'(\d+(?:\.\d+)?)\"([EW])"
+
+    match = re.match(pattern, coord_str)
+
+    if match:
+        lat_deg = int(match.group(1))
+        lat_min = int(match.group(2))
+        lat_sec = float(match.group(3))
+        lat_dir = match.group(4)
+
+        lon_deg = int(match.group(5))
+        lon_min = int(match.group(6))
+        lon_sec = float(match.group(7))
+        lon_dir = match.group(8)
+
+        # Convert coordinates to decimal format
+        lat_decimal = lat_deg + (lat_min / 60) + (lat_sec / 3600)
+        lon_decimal = lon_deg + (lon_min / 60) + (lon_sec / 3600)
+
+        # Apply sign based on direction
+        if lat_dir == 'S':
+            lat_decimal *= -1
+        if lon_dir == 'W':
+            lon_decimal *= -1
+
+        return (lat_decimal, lon_decimal)
+    else:
+        return None
+
+# Example usage
+coordinates = "30°01'48.7\"N 30°58'35.1\"E"
+latitude, longitude = convert_coordinates(coordinates)
+print(f"Latitude: {latitude}, Longitude: {longitude}")
+
+
+mineral_locations = {
+    'biotite': [
+        convert_coordinates("45°26'54.0\"N 75°41'24.0\"W"),
+        convert_coordinates("51°03'07.2\"N 114°04'28.8\"W"),
+        convert_coordinates("37°47'24.0\"N 122°28'12.0\"W")
+    ],
+    'granite': [
+        convert_coordinates("40°44'54.0\"N 73°59'08.0\"W"),
+        convert_coordinates("34°03'08.0\"S 151°06'24.0\"E"),
+        convert_coordinates("55°45'00.0\"N 37°37'00.0\"E")
+    ],
+    'olivine': [
+        convert_coordinates("19°28'48.0\"N 155°36'00.0\"W"),
+        convert_coordinates("38°41'24.0\"N 9°08'24.0\"W"),
+        convert_coordinates("46°28'12.0\"N 7°57'36.0\"E")
+    ],
+    'plagioclase': [
+        convert_coordinates("48°51'24.0\"N 2°21'05.0\"E"),
+        convert_coordinates("35°40'48.0\"N 139°41'24.0\"E"),
+        convert_coordinates("41°53'24.0\"N 12°29'24.0\"E")
+    ],
+    'staurolite': [
+        convert_coordinates("43°04'12.0\"N 89°22'48.0\"W"),
+        convert_coordinates("47°36'00.0\"N 7°36'00.0\"E"),
+        convert_coordinates("51°30'00.0\"N 0°07'48.0\"W")
+    ]
+}
+
+
+
+import folium
+
+def plot_mineral_locations(mineral_locations):
+    map = folium.Map(location=[0, 0], zoom_start=2)
+
+    for mineral, locations in mineral_locations.items():
+        for location in locations:
+            lat, lon = location
+            folium.Marker(
+                location=[lat, lon],
+                popup=mineral,
+                icon=folium.Icon(color='green')
+            ).add_to(map)
+
+    return map
+
+
+def generate_mineral_map(mineral_name):
+    # Create a map object
+    mineral_map = folium.Map(location=[0, 0], zoom_start=2)
+
+    # Get the locations for the specified mineral
+    locations = mineral_locations.get(mineral_name, [])
+
+    # Add markers for the mineral locations
+    for location in locations:
+        lat, lon = location
+        folium.Marker(
+            location=[lat, lon],
+            popup=mineral_name,
+            icon=folium.Icon(color='green')
+        ).add_to(mineral_map)
+
+    return mineral_map
+
+
+
+
+
+
+
 
 # Load the CNN feature extractor model
 from tensorflow.keras.models import load_model