update: return only polygons

server/main.py

@@ -7,8 +7,6 @@ from APIs.get_weather import get_weather
 import ssl
 import keyboard
 import sqlite3
-import geopandas as gpd
-import matplotlib.pyplot as plt
 
 
 app = Flask(__name__)
@@ -67,19 +65,6 @@ def on_key_press(server, event, cursor, conn):
 # Start a server on port 8443 using self defined HTTP class
 if __name__ == "__main__":
 
-    # Load GeoJSON data using geopandas
-    geo_data = gpd.read_file("server/map/mjosa.geojson")
-
-    # Filter only polygons
-    polygon_data = geo_data[geo_data['geometry'].geom_type == 'Polygon']
-
-    # Plot polygons
-    polygon_data.plot()
-    plt.title('Polygons in Mjøsa GeoJSON')
-    plt.xlabel('Longitude')
-    plt.ylabel('Latitude')
-    plt.show()
-
     try:
         # Initialize database connection
         conn = sqlite3.connect('server/sql_db/icedb')

server/map/get_relation.py

 import string
 import json
+import geopandas as gpd
+import matplotlib.pyplot as plt
 
 
 # get_relation returns the geojson data for a selected body of water
 def get_relation(self, body_of_water: string):
-    # Read geojson file
-    with open("server/map/" + body_of_water.lower() + ".geojson") as f:
-        data = json.load(f)
+    # Load GeoJSON data using geopandas
+    geo_data = gpd.read_file("server/map/mjosa.geojson")
+
+    # Filter only polygons
+    polygon_data = geo_data[geo_data['geometry'].geom_type == 'Polygon']
+
+    # Plot polygons
+    polygon_data.plot()
+    plt.title('Polygons in Mjøsa GeoJSON')
+    plt.xlabel('Longitude')
+    plt.ylabel('Latitude')
+    plt.show()
+
+    # Convert GeoDataFrame to GeoJSON-like dictionary
+    geojson_dict = json.loads(polygon_data.to_json())
 
     # Convert response data to JSON string
-    response_json = json.dumps(data)
+    response_json = json.dumps(geojson_dict)
 
     # Set headers
     self.send_response(200)

server/map/map_test.py

-import geopandas as gpd
-import matplotlib.pyplot as plt
-
-def map_test():
-    # Load GeoJSON data using geopandas
-    geo_data = gpd.read_file("mjosa.geojson")
-
-    # Plot choropleth map
-    geo_data.plot()
-    plt.title('Choropleth Map of Mjøsa Lake')
-    plt.xlabel('Longitude')
-    plt.ylabel('Latitude')
-    plt.show()
\ No newline at end of file