Update app.py

app.py

@@ -37,27 +37,31 @@ 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")
-    options.add_argument("--window-size=1280,1696")
-    
-    # CORRECT PATH FOR HUGGING FACE
-    options.binary_location = "/usr/bin/google-chrome-stable"  # Changed from google-chrome
     
-    service = Service(
-        executable_path="/usr/bin/chromedriver",
-        service_args=["--verbose", "--log-path=/tmp/chromedriver.log"]
-    )
+    # 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)
         return driver
     except Exception as e:
-        logger.error(f"Driver initialization failed: {str(e)}")
-        raise
+        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)}")
 
 
 @app.post("/transcript", response_model=TranscriptResponse)
@@ -146,16 +150,35 @@ async def get_transcript(request: VideoRequest):
 
 @app.get("/health")
 async def health_check():
+    import os
     paths = {
-        "chrome": "/usr/bin/google-chrome-stable",  # Consistent with binary_location
+        "chrome": "/usr/bin/google-chrome",
         "chromedriver": "/usr/bin/chromedriver"
     }
     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
+    
     return {
         "status": "OK",
         "paths": exists,
-        "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"
+        "chrome_version": chrome_version,
+        "chromedriver_version": chromedriver_version,
+        "working": all(exists.values())
     }
     
 if __name__ == "__main__":