Update app.py

app.py

@@ -1,12 +1,23 @@
 from pytubefix import YouTube
 from pytubefix.cli import on_progress
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+import logging as log
+import subprocess
+import json
+from typing import Tuple
 
+# --- Logging Setup ---
+log.basicConfig(
+    level=log.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s",
+)
+
+# --- FastAPI Setup ---
 app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Adjust as needed
+    allow_origins=["*"],  # Adjust in production
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -16,14 +27,73 @@ app.add_middleware(
 def index():
     return {"message": "Hello, World!"}
 
+
+# --- Shell Command Runner ---
+def cmd(command, check=True, shell=True, capture_output=True, text=True):
+    log.info(f"Executing command: {command}")
+    try:
+        result = subprocess.run(
+            command,
+            check=check,
+            shell=shell,
+            capture_output=capture_output,
+            text=text
+        )
+        log.info("Command executed successfully.")
+        return result
+    except subprocess.CalledProcessError as error:
+        log.error(f"Command failed: {error}")
+        raise HTTPException(status_code=500, detail="Token generation failed.")
+
+
+# --- Token Generator ---
+def generate_youtube_token() -> dict:
+    result = cmd("node test.js")
+    if not result or not result.stdout:
+        log.error("No output received from test.js")
+        raise HTTPException(status_code=500, detail="Invalid token response")
+    
+    try:
+        data = json.loads(result.stdout)
+        log.info(f"Token data received: {data}")
+        return data
+    except json.JSONDecodeError as e:
+        log.error(f"Failed to decode token JSON: {e}")
+        raise HTTPException(status_code=500, detail="Invalid JSON from token generator")
+
+
+# --- Token Verifier Function ---
+def po_token_verifier() -> Tuple[str, str]:
+    token_object = generate_youtube_token()
+    return token_object["visitorData"], token_object["poToken"]
+
+
+# --- Video Fetch Endpoint ---
 @app.get("/api/video/{id}")
 def video_id(id: str):
     url = f"https://www.youtube.com/watch?v={id}"
-    yt = YouTube(url, on_progress_callback=on_progress)
-    ys = yt.streams.get_highest_resolution()
-    return {
-        "title": yt.title,
-        "url": ys.url,
-        "thumbnail": yt.thumbnail_url,
-        "description": yt.description
-    }
+    log.info(f"Fetching YouTube video for ID: {id}")
+    
+    try:
+        yt = YouTube(
+            url,
+            use_po_token=True,
+            po_token_verifier=po_token_verifier
+        )
+        ys = yt.streams.get_highest_resolution()
+
+        if not ys:
+            log.warning("No suitable video stream found.")
+            raise HTTPException(status_code=404, detail="No suitable video stream found")
+
+        log.info(f"Video fetched: {yt.title}")
+        return {
+            "title": yt.title,
+            "url": ys.url,
+            "thumbnail": yt.thumbnail_url,
+            "description": yt.description
+        }
+
+    except Exception as e:
+        log.error(f"Failed to fetch video info: {e}")
+        raise HTTPException(status_code=500, detail=str(e))