Update app.py

app.py

@@ -11,6 +11,9 @@ from selenium.common.exceptions import TimeoutException, WebDriverException
 import time
 import logging
 import os
+import shutil
+from pathlib import Path
+
 
 app = FastAPI()
 
@@ -161,51 +164,22 @@ async def get_transcript(request: VideoRequest):
         if driver:
             driver.quit()
 
+
+
+app = FastAPI()
+
 @app.get("/health")
-async def health_check():
-    paths = {
-        "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)]
-    }
-    
-    chrome_version = "Not found"
-    chromedriver_version = "Not found"
-    
-    if exists["chrome"]:
-        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()
-        except Exception as e:
-            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={found_paths}, Exists={exists}")
-    
+def health_check():
+    chrome_path = shutil.which("google-chrome")
+    chromedriver_path = shutil.which("chromedriver")
     return {
-        "status": status,
-        "paths_checked": paths,
-        "paths_found": found_paths,
-        "chrome_version": chrome_version,
-        "chromedriver_version": chromedriver_version,
-        "working": all(exists.values())
+        "ChromePath": chrome_path,
+        "ChromeDriverPath": chromedriver_path,
+        "ChromeExists": Path(chrome_path or "").exists(),
+        "ChromeDriverExists": Path(chromedriver_path or "").exists()
     }
 
+
 @app.get("/")
 async def root():
     return {"message": "Welcome to YouTube Transcript API"}