Update app.py

app.py

@@ -7,14 +7,13 @@ from selenium.webdriver.chrome.options import Options
 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 selenium.common.exceptions import TimeoutException
 import time
 import logging
 import os
 import shutil
 from pathlib import Path
 
-
 app = FastAPI()
 
 # Configure CORS
@@ -30,6 +29,7 @@ app.add_middleware(
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Pydantic models
 class VideoRequest(BaseModel):
     url: str
 
@@ -39,6 +39,8 @@ class TranscriptResponse(BaseModel):
     error: str | None
     processing_time: float
 
+# Driver init and route handlers here...
+
 def init_driver():
     options = Options()
     options.add_argument("--headless=new")
@@ -163,11 +165,7 @@ async def get_transcript(request: VideoRequest):
     finally:
         if driver:
             driver.quit()
-
-
-
-app = FastAPI()
-
+ 
 @app.get("/health")
 def health_check():
     chrome_path = shutil.which("google-chrome")
@@ -179,11 +177,11 @@ def health_check():
         "ChromeDriverExists": Path(chromedriver_path or "").exists()
     }
 
-
 @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)))
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))
+