Update app.py

app.py

@@ -44,10 +44,9 @@ def init_driver():
     options.add_argument("--disable-gpu")
     options.add_argument("--window-size=1280,1696")
     
-    # For Hugging Face Spaces
-    options.binary_location = "/usr/bin/google-chrome"
+    # CORRECT PATH FOR HUGGING FACE
+    options.binary_location = "/usr/bin/google-chrome-stable"  # Changed from google-chrome
     
-    # Configure ChromeDriver
     service = Service(
         executable_path="/usr/bin/chromedriver",
         service_args=["--verbose", "--log-path=/tmp/chromedriver.log"]
@@ -56,15 +55,10 @@ def init_driver():
     try:
         driver = webdriver.Chrome(service=service, options=options)
         return driver
-    except WebDriverException as e:
+    except Exception as e:
         logger.error(f"Driver initialization failed: {str(e)}")
-        # Try alternative approach
-        try:
-            driver = webdriver.Chrome(options=options)
-            return driver
-        except Exception as fallback_e:
-            logger.error(f"Fallback driver initialization also failed: {str(fallback_e)}")
-            raise
+        raise
+
 
 @app.post("/transcript", response_model=TranscriptResponse)
 async def get_transcript(request: VideoRequest):
@@ -153,7 +147,7 @@ async def get_transcript(request: VideoRequest):
 @app.get("/health")
 async def health_check():
     paths = {
-        "chrome": "/usr/bin/google-chrome-stable",
+        "chrome": "/usr/bin/google-chrome-stable",  # Consistent with binary_location
         "chromedriver": "/usr/bin/chromedriver"
     }
     exists = {name: os.path.exists(path) for name, path in paths.items()}
@@ -163,6 +157,7 @@ async def health_check():
         "chrome_version": os.popen("/usr/bin/google-chrome-stable --version").read().strip() if exists["chrome"] else "Not found",
         "chromedriver_version": os.popen("/usr/bin/chromedriver --version").read().strip() if exists["chromedriver"] else "Not found"
     }
+    
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))