headless selenium update

app.py

@@ -13,6 +13,9 @@ from gradio_folium import Folium
 from diffusers import ControlNetModel, StableDiffusionControlNetInpaintPipeline
 import spaces
 
+from selenium.webdriver.firefox.options import Options
+from selenium import webdriver
+
 # Initialize APIs
 openai_client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
 geolocator = Nominatim(user_agent="geoapi")
@@ -141,7 +144,14 @@ def save_map_image(geojson_data):
     geo_layer.add_to(m)
     bounds = get_bounds(geojson_data)
     m.fit_bounds(bounds)
-    img_data = m._to_png(5)
+
+    # Configure Selenium for headless operation
+    options = Options()
+    options.add_argument("--headless")  # Enable headless mode
+    driver = webdriver.Firefox(options=options)  # Ensure GeckoDriver is properly installed and in PATH
+    img_data = m._to_png(5, driver=driver)
+    driver.quit()
+
     img = Image.open(io.BytesIO(img_data))
     img.save('map_image.png')
     return 'map_image.png'