a

main.py

@@ -253,8 +253,15 @@ def setup_webdriver():
     chrome_options.add_argument('--headless')
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
-    chrome_options.binary_location = os.getenv('CHROME_BINARY_LOCATION', '/usr/bin/google-chrome')
-    return webdriver.Chrome(options=chrome_options)
+    
+    # Check if running on Windows or Linux
+    if os.name == 'nt':  # Windows
+        # Let Selenium Manager handle driver installation
+        chrome_options.binary_location = None  # Use default Chrome installation
+        return webdriver.Chrome(options=chrome_options)
+    else:  # Linux
+        chrome_options.binary_location = os.getenv('CHROME_BINARY_LOCATION', '/usr/bin/google-chrome')
+        return webdriver.Chrome(options=chrome_options)
 
 def create_polygon_mask(image_size, points):
     """Create a mask image from polygon points"""
@@ -336,14 +343,34 @@ def capture_screenshot():
         
         time.sleep(1)
         
-        driver = setup_webdriver()
+        try:
+            driver = setup_webdriver()
+        except Exception as e:
+            logging.error(f"Webdriver setup error: {str(e)}")
+            error_msg = str(e)
+            if "chromedriver" in error_msg.lower():
+                return jsonify({
+                    'error': 'ChromeDriver setup failed. Please ensure Chrome is installed.'
+                }), 500
+            return jsonify({
+                'error': 'Failed to initialize screenshot capture. Please try again.'
+            }), 500
+            
         try:
             driver.set_window_size(width + 50, height + 50)
             map_url = f"http://localhost:{PORT}/static/temp_map.html"
             driver.get(map_url)
             time.sleep(3)
+            
+            # Check if the map loaded properly
+            if not os.path.exists(map_path):
+                raise Exception("Map file not generated")
+                
             driver.save_screenshot(filepath)
             
+            if not os.path.exists(filepath):
+                raise Exception("Screenshot not saved")
+            
             if polygon_points and len(polygon_points) >= 3:
                 img = Image.open(filepath)
                 mask = create_polygon_mask(img.size, polygon_points)
@@ -352,6 +379,10 @@ def capture_screenshot():
                 cutout_filename = f"cutout_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
                 cutout_filepath = os.path.join(SCREENSHOT_DIR, cutout_filename)
                 cutout.save(cutout_filepath)
+                
+                if not os.path.exists(cutout_filepath):
+                    raise Exception("Cutout not saved")
+                    
                 return jsonify({
                     'success': True,
                     'screenshot_path': f'/static/screenshots/{filename}',
@@ -363,12 +394,27 @@ def capture_screenshot():
                 'screenshot_path': f'/static/screenshots/{filename}'
             })
             
+        except Exception as e:
+            logging.error(f"Screenshot capture error: {str(e)}")
+            error_msg = str(e)
+            if "timeout" in error_msg.lower():
+                return jsonify({
+                    'error': 'Map loading timed out. Please try again.'
+                }), 500
+            return jsonify({
+                'error': 'Failed to capture screenshot. Please try again.'
+            }), 500
         finally:
-            driver.quit()
+            try:
+                driver.quit()
+            except:
+                pass
             
     except Exception as e:
         logging.error(f"Screenshot error: {str(e)}")
-        return jsonify({'error': str(e)}), 500
+        return jsonify({
+            'error': 'An unexpected error occurred. Please try again.'
+        }), 500
 
 @app.route('/analyze')
 def analyze():

static/css/styles.css

@@ -1,44 +0,0 @@
-#map {
-    position: relative;
-    width: 100%;
-    height: 500px;
-    overflow: hidden;
-}
-
-.draw-button {
-    position: absolute;
-    top: 10px;
-    right: 10px;
-    z-index: 1000;
-    padding: 8px 16px;
-    background-color: #fff;
-    border: 2px solid #ccc;
-    border-radius: 4px;
-    cursor: pointer;
-}
-
-.draw-button:hover {
-    background-color: #f0f0f0;
-}
-
-/* Make sure the SVG overlay is visible and blocks map interaction during drawing */
-svg {
-    position: absolute;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    z-index: 1000;
-}
-
-/* Style for the polygon being drawn */
-polygon {
-    fill: rgba(255, 0, 0, 0.2);
-    stroke: red;
-    stroke-width: 2;
-}
-
-/* Style for the map container during drawing */
-#map.drawing-mode iframe {
-    pointer-events: none;
-}

static/temp_map.html

@@ -0,0 +1,72 @@
+<!DOCTYPE html>
+<html>
+<head>
+    
+    <meta http-equiv="content-type" content="text/html; charset=UTF-8" />
+    
+        <script>
+            L_NO_TOUCH = false;
+            L_DISABLE_3D = false;
+        </script>
+    
+    <style>html, body {width: 100%;height: 100%;margin: 0;padding: 0;}</style>
+    <style>#map {position:absolute;top:0;bottom:0;right:0;left:0;}</style>
+    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/leaflet.js"></script>
+    <script src="https://code.jquery.com/jquery-3.7.1.min.js"></script>
+    <script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js"></script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/Leaflet.awesome-markers/2.0.2/leaflet.awesome-markers.js"></script>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/leaflet.css"/>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css"/>
+    <link rel="stylesheet" href="https://netdna.bootstrapcdn.com/bootstrap/3.0.0/css/bootstrap-glyphicons.css"/>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@fortawesome/[email protected]/css/all.min.css"/>
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/Leaflet.awesome-markers/2.0.2/leaflet.awesome-markers.css"/>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/gh/python-visualization/folium/folium/templates/leaflet.awesome.rotate.min.css"/>
+    
+            <meta name="viewport" content="width=device-width,
+                initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
+            <style>
+                #map_523d0a34c54581e390ad502b0b1160a9 {
+                    position: relative;
+                    width: 969.0px;
+                    height: 500.0px;
+                    left: 0.0%;
+                    top: 0.0%;
+                }
+                .leaflet-container { font-size: 1rem; }
+            </style>
+        
+</head>
+<body>
+    
+    
+            <div class="folium-map" id="map_523d0a34c54581e390ad502b0b1160a9" ></div>
+        
+</body>
+<script>
+    
+    
+            var map_523d0a34c54581e390ad502b0b1160a9 = L.map(
+                "map_523d0a34c54581e390ad502b0b1160a9",
+                {
+                    center: [18.463836119599378, 73.86806580367184],
+                    crs: L.CRS.EPSG3857,
+                    zoom: 18,
+                    zoomControl: true,
+                    preferCanvas: false,
+                }
+            );
+
+            
+
+        
+    
+            var tile_layer_ab287d9f0977d3570f0f8ea43ca86459 = L.tileLayer(
+                "https://server.arcgisonline.com/ArcGIS/rest/services/World_Imagery/MapServer/tile/{z}/{y}/{x}",
+                {"attribution": "Esri", "detectRetina": false, "maxZoom": 18, "minZoom": 0, "noWrap": false, "opacity": 1, "subdomains": "abc", "tms": false}
+            );
+        
+    
+            tile_layer_ab287d9f0977d3570f0f8ea43ca86459.addTo(map_523d0a34c54581e390ad502b0b1160a9);
+        
+</script>
+</html>

templates/index.html

@@ -1,5 +1,5 @@
 <!DOCTYPE html>
-<html>
+<html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">