hamza2923 commited on
Commit
4bdb267
·
verified ·
1 Parent(s): bf38a8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -43,20 +43,28 @@ def init_driver():
43
  options.add_argument("--disable-dev-shm-usage")
44
  options.add_argument("--disable-gpu")
45
  options.add_argument("--window-size=1280,1696")
46
- options.add_argument("--remote-debugging-port=9222")
47
- options.add_argument("--disable-blink-features=AutomationControlled")
48
- # Chrome binary location in Hugging Face Spaces
49
  options.binary_location = "/usr/bin/google-chrome"
50
 
51
- # ChromeDriver path
52
- service = Service(executable_path="/usr/bin/chromedriver")
 
 
 
53
 
54
  try:
55
  driver = webdriver.Chrome(service=service, options=options)
56
  return driver
57
  except WebDriverException as e:
58
  logger.error(f"Driver initialization failed: {str(e)}")
59
- raise
 
 
 
 
 
 
60
 
61
  @app.post("/transcript", response_model=TranscriptResponse)
62
  async def get_transcript(request: VideoRequest):
 
43
  options.add_argument("--disable-dev-shm-usage")
44
  options.add_argument("--disable-gpu")
45
  options.add_argument("--window-size=1280,1696")
46
+
47
+ # For Hugging Face Spaces
 
48
  options.binary_location = "/usr/bin/google-chrome"
49
 
50
+ # Configure ChromeDriver
51
+ service = Service(
52
+ executable_path="/usr/bin/chromedriver",
53
+ service_args=["--verbose", "--log-path=/tmp/chromedriver.log"]
54
+ )
55
 
56
  try:
57
  driver = webdriver.Chrome(service=service, options=options)
58
  return driver
59
  except WebDriverException as e:
60
  logger.error(f"Driver initialization failed: {str(e)}")
61
+ # Try alternative approach
62
+ try:
63
+ driver = webdriver.Chrome(options=options)
64
+ return driver
65
+ except Exception as fallback_e:
66
+ logger.error(f"Fallback driver initialization also failed: {str(fallback_e)}")
67
+ raise
68
 
69
  @app.post("/transcript", response_model=TranscriptResponse)
70
  async def get_transcript(request: VideoRequest):