Update app.py

app.py

@@ -8,6 +8,7 @@ from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException, WebDriverException
+from webdriver_manager.chrome import ChromeDriverManager
 import time
 import logging
 import os
@@ -37,32 +38,19 @@ class TranscriptResponse(BaseModel):
     processing_time: float
 
 def init_driver():
-    from selenium import webdriver
-    from selenium.webdriver.chrome.service import Service
-    from selenium.webdriver.chrome.options import Options
-    
     options = Options()
     options.add_argument("--headless=new")
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.add_argument("--disable-gpu")
-    
-    # For Hugging Face Spaces
     options.binary_location = "/usr/bin/google-chrome"
     
     try:
-        # Try with service first
-        service = Service(executable_path="/usr/bin/chromedriver")
-        driver = webdriver.Chrome(service=service, options=options)
+        # Use WebDriver Manager to automatically download ChromeDriver
+        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
         return driver
     except Exception as e:
-        try:
-            # Fallback without service
-            driver = webdriver.Chrome(options=options)
-            return driver
-        except Exception as e:
-            raise Exception(f"Both driver initialization methods failed: {str(e)}")
-
+        raise Exception(f"Driver initialization failed: {str(e)}")
 
 @app.post("/transcript", response_model=TranscriptResponse)
 async def get_transcript(request: VideoRequest):
@@ -150,37 +138,42 @@ async def get_transcript(request: VideoRequest):
 
 @app.get("/health")
 async def health_check():
-    import os
     paths = {
         "chrome": "/usr/bin/google-chrome",
-        "chromedriver": "/usr/bin/chromedriver"
+        "chromedriver": "/usr/bin/chromedriver"  # May not exist with WebDriver Manager
     }
     exists = {name: os.path.exists(path) for name, path in paths.items()}
     
-    # Try to get versions
     chrome_version = "Not found"
     chromedriver_version = "Not found"
     
     if exists["chrome"]:
         try:
             chrome_version = os.popen("/usr/bin/google-chrome --version").read().strip()
-        except:
-            pass
-            
-    if exists["chromedriver"]:
-        try:
-            chromedriver_version = os.popen("/usr/bin/chromedriver --version").read().strip()
-        except:
-            pass
+        except Exception as e:
+            logger.error(f"Failed to get Chrome version: {str(e)}")
+    
+    # Check ChromeDriver version (may be managed by WebDriver Manager)
+    try:
+        chromedriver_version = os.popen("/usr/bin/chromedriver --version").read().strip() or "Managed by WebDriver Manager"
+    except Exception as e:
+        logger.info("ChromeDriver version check skipped (likely managed by WebDriver Manager)")
+
+    status = "OK" if exists["chrome"] else "ERROR"
+    logger.info(f"Health check: Chrome={chrome_version}, ChromeDriver={chromedriver_version}, Paths={exists}")
     
     return {
-        "status": "OK",
+        "status": status,
         "paths": exists,
         "chrome_version": chrome_version,
         "chromedriver_version": chromedriver_version,
-        "working": all(exists.values())
+        "working": exists["chrome"]  # Chrome is critical, ChromeDriver may be managed
     }
-    
+
+@app.get("/")
+async def root():
+    return {"message": "Welcome to YouTube Transcript API"}
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))