replace yt-dlp with cobalt.tools

app.py

@@ -503,12 +503,46 @@ def get_youtube_stream_url(youtube_url):
                 f"Error: Unable to process the YouTube URL. {str(e)}")
             return None
 
+# Function to get video URL using Cobalt API
+
+
+def get_cobalt_video_url(youtube_url):
+    cobalt_api_url = "https://api.cobalt.tools/api/json"
+    headers = {
+        "Accept": "application/json",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "url": youtube_url,
+        "vCodec": "h264",
+        "vQuality": "720",
+        "aFormat": "mp3",
+        "isAudioOnly": False
+    }
+
+    try:
+        response = requests.post(cobalt_api_url, headers=headers, json=payload)
+        response.raise_for_status()
+        data = response.json()
+
+        if data['status'] == 'stream':
+            return data['url']
+        elif data['status'] == 'redirect':
+            return data['url']
+        else:
+            yt_error.error(f"Error: {data['text']}")
+            return None
+    except requests.exceptions.RequestException as e:
+        yt_error.error(f"Error: Unable to process the YouTube URL. {str(e)}")
+        return None
+
 
 # If a YouTube URL is provided, process the video
 if youtube_url:
     analysis_init()  # Initialize the analysis UI
 
-    stream_url = get_youtube_stream_url(youtube_url)
+    stream_url = get_cobalt_video_url(youtube_url)
+    # stream_url = get_youtube_stream_url(youtube_url)
 
     if stream_url:
         process_video(stream_url)  # Process the video

requirements.txt

@@ -6,5 +6,4 @@ setuptools
 streamlit_webrtc
 twilio
 ultralytics
-yt-dlp
 watchdog