Update app.py

app.py

@@ -1,3 +1,4 @@
+import requests
 import gradio as gr
 import os
 import time
@@ -11,10 +12,26 @@ asr = pipeline("automatic-speech-recognition", model="distil-whisper/distil-smal
 # Summarization model
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
+# Function to fetch PO Token from the Node.js server
+def get_po_token():
+    try:
+        response = requests.get("http://localhost:3001/get-token")  # Call Node.js API
+        response.raise_for_status()
+        return response.json().get("poToken")  # Extract token
+    except Exception as e:
+        print(f"Error fetching PO token: {e}")
+        return None  # Return None if the request fails
+
 def process_youtube_link(youtube_url):
     try:
-        print(youtube_url)
-        yt = YouTube(youtube_url, use_po_token=True)
+        print(f"Received YouTube URL: {youtube_url}")  # Debugging
+
+        po_token = get_po_token()
+        if not po_token:
+            return "Error: Unable to fetch PO token.", ""
+
+        # Use the token with pytubefix
+        yt = YouTube(youtube_url, po_token=po_token)
         title = yt.title
         print(f"Downloading: {title}")
 
@@ -57,4 +74,3 @@ iface = gr.Interface(
 )
 
 iface.launch()
-