Update main.py

main.py

@@ -13,8 +13,8 @@ import logging
 import os
 import shutil
 from pathlib import Path
-import tempfile
 from tenacity import retry, stop_after_attempt, wait_fixed
+import subprocess
 
 app = FastAPI()
 
@@ -41,19 +41,24 @@ class TranscriptResponse(BaseModel):
     error: str | None
     processing_time: float
 
-@retry(stop=stop_after_attempt(3), wait=wait_fixed(2))
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(3))
 def init_driver():
+    # Clean up any lingering Chrome processes
+    try:
+        subprocess.run(["pkill", "-f", "chrome"], check=False)
+        logger.info("Terminated any existing Chrome processes")
+    except Exception as e:
+        logger.warning(f"Failed to terminate Chrome processes: {str(e)}")
+
     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("--disable-extensions")
-
-    # Create a unique temporary user data directory
-    user_data_dir = tempfile.mkdtemp()
-    options.add_argument(f"--user-data-dir={user_data_dir}")
-    logger.info(f"Using temporary user data directory: {user_data_dir}")
+    # Removed --user-data-dir to avoid conflicts
+    options.add_argument("--disable-setuid-sandbox")
+    options.add_argument("--remote-debugging-port=9222")
 
     possible_chrome_paths = [
         "/usr/bin/google-chrome",
@@ -83,26 +88,22 @@ def init_driver():
         chrome_version = driver.capabilities["browserVersion"]
         chromedriver_version = driver.capabilities["chrome"]["chromedriverVersion"].split()[0]
         logger.info(f"Chrome version: {chrome_version}, ChromeDriver version: {chromedriver_version}")
-        return driver, user_data_dir
+        return driver
     except Exception as e:
         logger.error(f"Driver initialization failed: {str(e)}")
-        # Clean up the temporary directory in case of failure
-        if Path(user_data_dir).exists():
-            shutil.rmtree(user_data_dir, ignore_errors=True)
         raise Exception(f"Driver initialization failed: {str(e)}")
 
 @app.post("/transcript", response_model=TranscriptResponse)
 async def get_transcript(request: VideoRequest):
     start_time = time.time()
     driver = None
-    user_data_dir = None
 
     try:
         video_url = request.url
         if not ("youtube.com" in video_url or "youtu.be" in video_url):
             raise HTTPException(status_code=400, detail="Invalid YouTube URL")
 
-        driver, user_data_dir = init_driver()
+        driver = init_driver()
         logger.info(f"Processing URL: {video_url}")
         driver.get(video_url)
 
@@ -174,19 +175,25 @@ async def get_transcript(request: VideoRequest):
     finally:
         if driver:
             driver.quit()
-        if user_data_dir and Path(user_data_dir).exists():
-            shutil.rmtree(user_data_dir, ignore_errors=True)
-            logger.info(f"Cleaned up temporary user data directory: {user_data_dir}")
+            logger.info("Driver closed")
 
 @app.get("/health")
 def health_check():
     chrome_path = shutil.which("google-chrome")
     chromedriver_path = shutil.which("chromedriver")
+    try:
+        # Check for running Chrome processes
+        result = subprocess.run(["ps", "aux"], capture_output=True, text=True)
+        chrome_processes = [line for line in result.stdout.splitlines() if "chrome" in line.lower()]
+        chrome_process_count = len(chrome_processes)
+    except Exception as e:
+        chrome_process_count = f"Error checking processes: {str(e)}"
     return {
         "ChromePath": chrome_path,
         "ChromeDriverPath": chromedriver_path,
         "ChromeExists": Path(chrome_path or "").exists(),
-        "ChromeDriverExists": Path(chromedriver_path or "").exists()
+        "ChromeDriverExists": Path(chromedriver_path or "").exists(),
+        "ChromeProcessCount": chrome_process_count
     }
 
 @app.get("/")