Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
| 47 |
-
|
| 48 |
-
# Chrome binary location in Hugging Face Spaces
|
| 49 |
options.binary_location = "/usr/bin/google-chrome"
|
| 50 |
|
| 51 |
-
# ChromeDriver
|
| 52 |
-
service = Service(
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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):
|