StarrySkyWorld commited on
Commit
2f7aeb5
·
1 Parent(s): 6201250

fix: fix browser not include image

Browse files
Files changed (1) hide show
  1. registration/browser.py +6 -0
registration/browser.py CHANGED
@@ -192,6 +192,10 @@ class BrowserAutomation:
192
  # headless можно переопределить параметром
193
  if headless is None:
194
  headless = browser_settings.get('headless', False)
 
 
 
 
195
 
196
  self.settings = settings
197
  self.headless = headless
@@ -221,6 +225,8 @@ class BrowserAutomation:
221
 
222
  if headless:
223
  co.headless()
 
 
224
  # Дополнительные аргументы для стабильного headless
225
  co.set_argument('--disable-gpu')
226
  co.set_argument('--no-sandbox')
 
192
  # headless можно переопределить параметром
193
  if headless is None:
194
  headless = browser_settings.get('headless', False)
195
+
196
+ # Force headless when no display is available (e.g., HF Spaces)
197
+ if not os.environ.get('DISPLAY') and not os.environ.get('WAYLAND_DISPLAY'):
198
+ headless = True
199
 
200
  self.settings = settings
201
  self.headless = headless
 
225
 
226
  if headless:
227
  co.headless()
228
+ # Explicit headless flag for Chromium in headless Linux
229
+ co.set_argument('--headless=new')
230
  # Дополнительные аргументы для стабильного headless
231
  co.set_argument('--disable-gpu')
232
  co.set_argument('--no-sandbox')