Update app.py

app.py

@@ -12,9 +12,9 @@ import streamlit as st
 
 # Function to parse KML
 def parse_kml(kml_file):
-    with open(kml_file, 'rt') as f:
-        doc = f.read()
-    k = ET.ElementTree(ET.fromstring(doc))
+    # kml_file will be in bytes format, so we decode it to a string
+    kml_content = kml_file.decode('utf-8')  # Decode bytes to string
+    k = ET.ElementTree(ET.fromstring(kml_content))
     root = k.getroot()
     ns = {'kml': 'http://www.opengis.net/kml/2.2'}
     
@@ -44,14 +44,20 @@ def extract_kmz(kmz_file):
     with zipfile.ZipFile(kmz_file, 'r') as zip_ref:
         zip_ref.extractall('temp_kml')
     kml_file = [f for f in os.listdir('temp_kml') if f.endswith('.kml')][0]
-    return parse_kml(os.path.join('temp_kml', kml_file))
+    with open(os.path.join('temp_kml', kml_file), 'rb') as f:
+        return parse_kml(f.read())
 
 # Function to handle KML/KMZ file input
 def handle_kml_upload(uploaded_file):
+    # Read the uploaded file as bytes
+    file_bytes = uploaded_file.read()
+
+    # If the file is a KMZ, first extract the KML file from the KMZ archive
     if uploaded_file.name.endswith('.kmz'):
-        kml_shapes = extract_kmz(uploaded_file)
+        kml_shapes = extract_kmz(file_bytes)
     else:
-        kml_shapes = parse_kml(uploaded_file)
+        # Otherwise, it's a direct KML file
+        kml_shapes = parse_kml(file_bytes)
     
     return kml_shapes
 
@@ -62,9 +68,9 @@ uploaded_file = st.file_uploader("Upload KML/KMZ file", type=['kml', 'kmz'])
 
 if uploaded_file is not None:
     # Step 1: Read in the shapefiles
-    shapefile_1 = gpd.read_file("K_FLD_HAZ_AR.shp")
-    shapefile_2 = gpd.read_file("N_FLD_HAZ_AR.shp")
-    shapefile_3 = gpd.read_file("S_FLD_HAZ_AR.shp")
+    shapefile_1 = gpd.read_file("Kent/S_FLD_HAZ_AR.shp")
+    shapefile_2 = gpd.read_file("NC/S_FLD_HAZ_AR.shp")
+    shapefile_3 = gpd.read_file("Sus/S_FLD_HAZ_AR.shp")
 
     # Step 2: Concatenate them into a single GeoDataFrame
     merged_gdf = gpd.GeoDataFrame(pd.concat([shapefile_1, shapefile_2, shapefile_3], ignore_index=True))