Fix PORT environment variable issue

Dockerfile

@@ -46,6 +46,7 @@ COPY --chown=user . .
 ENV PATH="/home/user/.local/bin:${PATH}"
 ENV PYTHONPATH="/home/user/.local/lib/python3.9/site-packages:${PYTHONPATH}"
 ENV CHROME_BINARY_LOCATION="/usr/bin/google-chrome"
+ENV PORT=7860
 
 # Run the application
 CMD ["python", "main.py"]

main.py

@@ -29,6 +29,8 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max file size
 
+PORT = int(os.getenv('PORT', 7860))
+
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
@@ -329,58 +331,27 @@ def capture_screenshot():
                 height=height
             )
             
-            # Set the bounds
-            bounds = map_state['bounds']
-            m.fit_bounds([[bounds['south'], bounds['west']], 
-                         [bounds['north'], bounds['east']]])
-            
-            # Add custom JavaScript to ensure correct zoom
-            m.get_root().html.add_child(folium.Element(f"""
-                <script>
-                    document.addEventListener('DOMContentLoaded', function() {{
-                        setTimeout(function() {{
-                            var map = document.querySelector('#map');
-                            if (map && map._leaflet_map) {{
-                                map._leaflet_map.setView([{center['lat']}, {center['lng']}], {zoom});
-                            }}
-                        }}, 1000);
-                    }});
-                </script>
-            """))
-            
-            # Save the map
             map_path = os.path.join(app.static_folder, 'temp_map.html')
             m.save(map_path)
         
-        # Increase wait time to ensure map loads completely
         time.sleep(1)
         
         driver = setup_webdriver()
         try:
-            driver.set_window_size(width + 50, height + 50)  # Add padding to prevent scrollbars
-            map_url = f"http://localhost:{app.config['PORT']}/static/temp_map.html"
+            driver.set_window_size(width + 50, height + 50)
+            map_url = f"http://localhost:{PORT}/static/temp_map.html"
             driver.get(map_url)
-            
-            # Wait for map to load and settle
             time.sleep(3)
-            
-            # Take screenshot
             driver.save_screenshot(filepath)
             
             if polygon_points and len(polygon_points) >= 3:
-                # Create polygon cutout
                 img = Image.open(filepath)
                 mask = create_polygon_mask(img.size, polygon_points)
-                
-                # Create cutout image
                 cutout = Image.new('RGBA', img.size, (0, 0, 0, 0))
                 cutout.paste(img, mask=mask)
-                
-                # Save cutout
                 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)
-                
                 return jsonify({
                     'success': True,
                     'screenshot_path': f'/static/screenshots/{filename}',
@@ -469,4 +440,4 @@ def upload_file():
     return jsonify({'error': 'Invalid file type'}), 400
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860) 
+    app.run(host='0.0.0.0', port=PORT)