update FileURL

app.py

@@ -4,7 +4,9 @@ import requests
 import io
 import re
 from fastapi import FastAPI
-import kml2geojson  # Using this library for robust KML parsing
+import kml2geojson
+
+# All helper functions from the previous version remain the same.
 
 def get_gdf_from_file_url(file_url: str):
     """
@@ -17,9 +19,7 @@ def get_gdf_from_file_url(file_url: str):
     Returns:
         A GeoDataFrame containing the data from the file.
     """
-    # --- Handle different Google Drive URL formats ---
     if "drive.google.com" in file_url:
-        # Use regex to extract the file ID robustly
         match = re.search(r'/d/([a-zA-Z0-9_-]+)', file_url) or re.search(r'id=([a-zA-Z0-9_-]+)', file_url)
         if match:
             file_id = match.group(1)
@@ -29,38 +29,26 @@ def get_gdf_from_file_url(file_url: str):
     else:
         download_url = file_url
 
-    # --- Download the file content ---
     response = requests.get(download_url, timeout=30)
     response.raise_for_status()
     
     bytes_data = io.BytesIO(response.content)
-    # Decode for sniffing, handle potential encoding issues
     try:
         string_data = response.content.decode('utf-8')
     except UnicodeDecodeError:
         string_data = response.content.decode('latin-1')
 
-
-    # --- Determine file type and parse ---
-    # If it looks like XML, it's likely KML.
     if string_data.strip().startswith("<?xml"):
-        # kml2geojson requires the file path or a stream of the file
-        bytes_data.seek(0) # Reset stream position
+        bytes_data.seek(0)
         geojson_data = kml2geojson.convert(bytes_data)
-        
-        # Extract features from the converted GeoJSON
-        # The output of kml2geojson is a list of geojson objects
         all_features = []
         for gj in geojson_data:
             all_features.extend(gj.get('features', []))
-
         if not all_features:
             raise ValueError("KML file parsed, but no features were found.")
-            
         input_gdf = gpd.GeoDataFrame.from_features(all_features, crs="EPSG:4326")
     else:
-        # Assume it's GeoJSON or another format GeoPandas can read directly
-        bytes_data.seek(0) # Reset stream position
+        bytes_data.seek(0)
         input_gdf = gpd.read_file(bytes_data)
 
     return input_gdf
@@ -68,53 +56,36 @@ def get_gdf_from_file_url(file_url: str):
 
 def calculate_geometry(file_url: str) -> tuple[str, str]:
     """
-    Main function for the Gradio interface. It orchestrates the downloading,
-    parsing, validation, and calculation of area and perimeter.
-    
-    Args:
-        file_url: The URL of the KML or GeoJSON file.
-
-    Returns:
-        A tuple containing the formatted area and perimeter strings.
+    Main calculation logic.
     """
     if not file_url or not file_url.strip():
-        return "Error: Please provide a URL.", ""
+        return "Error: No URL provided.", ""
 
     try:
-        # 1. Get GeoDataFrame from the URL
         input_gdf = get_gdf_from_file_url(file_url)
-
         if input_gdf.empty:
             return "Error: Could not find any geometric features in the file.", ""
         
-        # 2. Find the first valid polygon in the GeoDataFrame
         geometry_gdf = None
         for i, row in input_gdf.iterrows():
             if row.geometry.geom_type in ['Polygon', 'MultiPolygon']:
-                # Create a new GDF with just this valid geometry
                 geometry_gdf = gpd.GeoDataFrame([row], crs=input_gdf.crs)
                 break
         
         if geometry_gdf is None:
             return "Error: No valid Polygon or MultiPolygon geometry found in the file.", ""
 
-        # 3. Reproject for accurate measurement
-        # Ensure the initial CRS is set correctly (WGS84) before reprojecting
         if geometry_gdf.crs is None or geometry_gdf.crs.to_epsg() != 4326:
              geometry_gdf = geometry_gdf.set_crs(epsg=4326, allow_override=True)
 
-        # Determine the correct UTM zone for accurate measurements in meters
         centroid = geometry_gdf.geometry.iloc[0].centroid
         utm_zone = int((centroid.x + 180) / 6) + 1
         epsg_code = 32600 + utm_zone if centroid.y >= 0 else 32700 + utm_zone
-        
         gdf_proj = geometry_gdf.to_crs(epsg=epsg_code)
 
-        # 4. Calculate area and perimeter
         area_m2 = gdf_proj.geometry.iloc[0].area
         perimeter_m = gdf_proj.geometry.iloc[0].length
 
-        # 5. Format output
         area_km2 = area_m2 / 1_000_000
         perimeter_km = perimeter_m / 1_000
 
@@ -122,32 +93,68 @@ def calculate_geometry(file_url: str) -> tuple[str, str]:
         perimeter_str = f"{perimeter_m:,.2f} meters ({perimeter_km:,.2f} km)"
 
         return area_str, perimeter_str
-
     except Exception as e:
         return f"An error occurred: {e}", ""
 
-# Create the Gradio user interface
-gradio_interface = gr.Interface(
-    fn=calculate_geometry,
-    inputs=gr.Textbox(
-        label="Google Drive KML/GeoJSON File URL",
-        placeholder="Paste the URL to your file here..."
-    ),
-    outputs=[
-        gr.Textbox(label="Calculated Area", interactive=False),
-        gr.Textbox(label="Calculated Perimeter", interactive=False)
-    ],
-    title="Polygon Area & Perimeter Calculator 🗺️",
-    description=(
-        "Enter the public Google Drive URL of a KML or GeoJSON file that contains a polygon. "
-        "The tool will find the first valid polygon and calculate its area and perimeter."
-    ),
-    examples=[
-        ["https://drive.google.com/file/d/123KCak3o1VUcrYQO6v-HxVPYDrYLw4Ft/view?usp=drivesdk"]
-    ],
-    allow_flagging="never"
-)
+
+# --- New Gradio Interface using Blocks for more control ---
+with gr.Blocks() as demo:
+    gr.Markdown(
+        "# Polygon Area & Perimeter Calculator 🗺️\n"
+        "Enter the public Google Drive URL of a KML or GeoJSON file, or load this page with "
+        "`?file_url=<your_url>` to process automatically."
+    )
+
+    with gr.Column():
+        url_input = gr.Textbox(
+            label="Google Drive KML/GeoJSON File URL",
+            placeholder="Paste URL here or load from page URL..."
+        )
+        submit_button = gr.Button("Calculate", variant="primary")
+
+    with gr.Row():
+        area_output = gr.Textbox(label="Calculated Area", interactive=False)
+        perimeter_output = gr.Textbox(label="Calculated Perimeter", interactive=False)
+    
+    gr.Examples(
+        examples=[
+            ["https://drive.google.com/file/d/123KCak3o1VUcrYQO6v-HxVPYDrYLw4Ft/view?usp=drivesdk"]
+        ],
+        inputs=url_input
+    )
+
+    # This function runs when the page loads to check for a URL query parameter
+    def process_url_on_load(request: gr.Request):
+        file_url = request.query_params.get("file_url")
+        if file_url:
+            area, perimeter = calculate_geometry(file_url)
+            # Return a dictionary to update the components
+            return {
+                url_input: file_url,
+                area_output: area,
+                perimeter_output: perimeter
+            }
+        # If no URL, return empty values
+        return {
+            url_input: "",
+            area_output: "",
+            perimeter_output: ""
+        }
+
+    # Define the event handlers
+    submit_button.click(
+        fn=calculate_geometry,
+        inputs=url_input,
+        outputs=[area_output, perimeter_output]
+    )
+    
+    demo.load(
+        fn=process_url_on_load,
+        inputs=None, # The 'load' event doesn't take direct component inputs
+        outputs=[url_input, area_output, perimeter_output]
+    )
 
 # Initialize the FastAPI app and mount the Gradio interface
 app = FastAPI()
-app = gr.mount_gradio_app(app, gradio_interface, path="/")
+app = gr.mount_gradio_app(app, demo, path="/")
+