add proxies

app.py

@@ -7,6 +7,21 @@ import re, shutil, subprocess, textwrap, os, tempfile
 from pathlib import Path
 from typing import List, Optional
 import gradio as gr
+import random
+
+# --- Oxylabs Proxy Configuration ---
+PROXY_USERNAME = "ujwal"
+PROXY_PASSWORD = "xJv4DChht5P6y+u"
+PROXY_COUNTRY = "US"
+
+# List of proxy endpoints (Oxylabs DC Proxies)
+PROXY_PORTS = [8001, 8002, 8003, 8004, 8005]
+PROXY_HOST = "dc.oxylabs.io"
+
+def get_random_proxy():
+    port = random.choice(PROXY_PORTS)
+    return f"http://user-{PROXY_USERNAME}-country-{PROXY_COUNTRY}:{PROXY_PASSWORD}@{PROXY_HOST}:{port}"
+
 
 # ---------------- CONFIG ----------------
 TITLE_DUR         = 3                               # seconds
@@ -35,23 +50,70 @@ def run_cmd(cmd: list[str], timeout: int = 120) -> bool:
         print(f"Command failed:\n{' '.join(cmd)}\nError:\n{e.stderr if hasattr(e, 'stderr') else str(e)}")
         return False
 
+# def yt_urls(query: str, max_results: int) -> List[str]:
+#     """Get YouTube URLs for search query"""
+#     try:
+#         from youtube_search import YoutubeSearch
+#         results = YoutubeSearch(query, max_results=max_results).to_dict()
+#         return ["https://www.youtube.com" + r["url_suffix"] for r in results]
+#     except Exception as e:
+#         print(f"YouTube search failed: {e}")
+#         return []
+
 def yt_urls(query: str, max_results: int) -> List[str]:
-    """Get YouTube URLs for search query"""
+    """Get YouTube URLs for search query via proxy"""
     try:
+        import requests
         from youtube_search import YoutubeSearch
+
+        proxy_url = get_random_proxy()
+        proxies = {
+            "http": proxy_url,
+            "https": proxy_url,
+        }
+
+        # Monkey-patch YoutubeSearch to use proxy
+        original_get = requests.get
+
+        def proxied_get(*args, **kwargs):
+            kwargs["proxies"] = proxies
+            kwargs["verify"] = False
+            return original_get(*args, **kwargs)
+
+        requests.get = proxied_get
         results = YoutubeSearch(query, max_results=max_results).to_dict()
+        requests.get = original_get  # Restore
+
         return ["https://www.youtube.com" + r["url_suffix"] for r in results]
     except Exception as e:
         print(f"YouTube search failed: {e}")
         return []
 
+
 def safe_filename(name: str) -> str:
     """Create safe filename"""
     return re.sub(r"[^\w\-\.]", "_", name)[:50]  # Limit length
 
+# def dl_video(url: str, out: Path) -> bool:
+#     """Download video with length limit"""
+#     out.parent.mkdir(exist_ok=True)
+#     cmd = [
+#         "yt-dlp",
+#         "--match-filter", f"duration<{MAX_VIDEO_LENGTH}",
+#         "-f", "mp4",
+#         "--merge-output-format", "mp4",
+#         "-o", str(out),
+#         "--no-playlist",
+#         "--no-check-certificate",
+#         url,
+#     ]
+#     return run_cmd(cmd, timeout=60)
+
 def dl_video(url: str, out: Path) -> bool:
-    """Download video with length limit"""
+    """Download video with length limit using rotating proxy"""
     out.parent.mkdir(exist_ok=True)
+    proxy = get_random_proxy()
+
     cmd = [
         "yt-dlp",
         "--match-filter", f"duration<{MAX_VIDEO_LENGTH}",
@@ -60,10 +122,12 @@ def dl_video(url: str, out: Path) -> bool:
         "-o", str(out),
         "--no-playlist",
         "--no-check-certificate",
+        "--proxy", proxy,
         url,
     ]
     return run_cmd(cmd, timeout=60)
 
+
 def make_card(text: str, out: Path, dur: int = TITLE_DUR) -> bool:
     """Create title card with text"""
     # Wrap text for better display