Update app.py

app.py

@@ -42,11 +42,34 @@ def init_driver():
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.add_argument("--disable-gpu")
-    options.binary_location = "/usr/bin/google-chrome"
+    
+    # Try multiple possible Chrome binary locations
+    possible_chrome_paths = [
+        "/usr/bin/google-chrome",
+        "/usr/bin/google-chrome-stable",
+        "/usr/lib/chromium-browser/chrome",
+        "/usr/bin/chromium"
+    ]
+    chrome_path = None
+    for path in possible_chrome_paths:
+        if os.path.exists(path):
+            chrome_path = path
+            break
+    
+    if not chrome_path:
+        logger.error(f"No Chrome binary found in paths: {possible_chrome_paths}")
+        raise Exception(f"No Chrome binary found in paths: {possible_chrome_paths}")
+    
+    options.binary_location = chrome_path
+    logger.info(f"Using Chrome binary: {chrome_path}")
     
     try:
-        # Use explicit ChromeDriver path
-        service = Service(executable_path="/usr/bin/chromedriver")
+        chromedriver_path = "/usr/bin/chromedriver"
+        if not os.path.exists(chromedriver_path):
+            logger.error(f"ChromeDriver not found at {chromedriver_path}")
+            raise Exception(f"ChromeDriver not found at {chromedriver_path}")
+        
+        service = Service(executable_path=chromedriver_path)
         driver = webdriver.Chrome(service=service, options=options)
         logger.info("ChromeDriver initialized successfully")
         return driver
@@ -141,20 +164,30 @@ async def get_transcript(request: VideoRequest):
 @app.get("/health")
 async def health_check():
     paths = {
-        "chrome": "/usr/bin/google-chrome",
-        "chromedriver": "/usr/bin/chromedriver"
+        "chrome": ["/usr/bin/google-chrome", "/usr/bin/google-chrome-stable", "/usr/lib/chromium-browser/chrome", "/usr/bin/chromium"],
+        "chromedriver": ["/usr/bin/chromedriver"]
+    }
+    exists = {
+        "chrome": any(os.path.exists(path) for path in paths["chrome"]),
+        "chromedriver": any(os.path.exists(path) for path in paths["chromedriver"])
+    }
+    found_paths = {
+        "chrome": [path for path in paths["chrome"] if os.path.exists(path)],
+        "chromedriver": [path for path in paths["chromedriver"] if os.path.exists(path)]
     }
-    exists = {name: os.path.exists(path) for name, path in paths.items()}
     
     chrome_version = "Not found"
     chromedriver_version = "Not found"
     
     if exists["chrome"]:
-        try:
-            chrome_version = os.popen("/usr/bin/google-chrome --version").read().strip()
-        except Exception as e:
-            logger.error(f"Failed to get Chrome version: {str(e)}")
-            
+        for path in paths["chrome"]:
+            try:
+                chrome_version = os.popen(f"{path} --version").read().strip()
+                if chrome_version:
+                    break
+            except Exception as e:
+                logger.error(f"Failed to get Chrome version for {path}: {str(e)}")
+    
     if exists["chromedriver"]:
         try:
             chromedriver_version = os.popen("/usr/bin/chromedriver --version").read().strip()
@@ -162,11 +195,12 @@ async def health_check():
             logger.error(f"Failed to get ChromeDriver version: {str(e)}")
     
     status = "OK" if all(exists.values()) else "ERROR"
-    logger.info(f"Health check: Chrome={chrome_version}, ChromeDriver={chromedriver_version}, Paths={exists}")
+    logger.info(f"Health check: Chrome={chrome_version}, ChromeDriver={chromedriver_version}, Paths={found_paths}, Exists={exists}")
     
     return {
         "status": status,
-        "paths": exists,
+        "paths_checked": paths,
+        "paths_found": found_paths,
         "chrome_version": chrome_version,
         "chromedriver_version": chromedriver_version,
         "working": all(exists.values())