fix youtube downloader v1.1

requirements.txt

@@ -27,3 +27,4 @@ google-api-python-client==2.115.0
 pydantic-core==2.41.5
 ffmpeg-python
 firebase-admin==6.5.0
+dnspython

src/api/main.py

@@ -68,6 +68,54 @@ app.include_router(auth_router)
 app.include_router(notes_router)
 
 
+@app.get("/health")
+async def health_check():
+    import socket
+    import httpx
+    try:
+        import dns.resolver
+        has_dnspython = True
+    except ImportError:
+        has_dnspython = False
+    
+    dns_results = {}
+    for domain in ["www.youtube.com", "google.com"]:
+        # System DNS
+        try:
+            dns_results[f"{domain}_sys"] = socket.gethostbyname(domain)
+        except Exception as e:
+            dns_results[f"{domain}_sys"] = f"Error: {e}"
+            
+        # dnspython (External DNS 8.8.8.8)
+        if has_dnspython:
+            try:
+                resolver = dns.resolver.Resolver()
+                resolver.nameservers = ['8.8.8.8']
+                answer = resolver.resolve(domain, 'A')
+                dns_results[f"{domain}_ext"] = [str(rdata) for rdata in answer]
+            except Exception as e:
+                dns_results[f"{domain}_ext"] = f"Error: {repr(e)}"
+        else:
+            dns_results[f"{domain}_ext"] = "dnspython not installed"
+            
+    connectivity = {}
+    async with httpx.AsyncClient(timeout=5.0) as client:
+        for url in ["https://www.youtube.com"]:
+            try:
+                resp = await client.get(url, follow_redirects=True)
+                connectivity[url] = f"OK ({resp.status_code})"
+            except Exception as e:
+                connectivity[url] = f"Failed: {repr(e)}"
+                
+    return {
+        "status": "v6-online",
+        "dnspython": has_dnspython,
+        "dns": dns_results,
+        "connectivity": connectivity,
+        "timestamp": datetime.now()
+    }
+
+
 @app.post("/generate", response_model=TaskResponse)
 async def generate(
     request: GenerateNotesRequest,
@@ -152,35 +200,6 @@ async def process_video_and_save(
             downloader.cleanup(audio_file)
 
 
-@app.get("/health")
-async def health_check():
-    import socket
-    import httpx
-    
-    dns_results = {}
-    for domain in ["www.youtube.com", "google.com", "huggingface.co"]:
-        try:
-            dns_results[domain] = socket.gethostbyname(domain)
-        except Exception as e:
-            dns_results[domain] = f"Error: {e}"
-            
-    connectivity = {}
-    async with httpx.AsyncClient(timeout=5.0) as client:
-        for url in ["https://www.google.com", "https://www.youtube.com"]:
-            try:
-                resp = await client.get(url, follow_redirects=True)
-                connectivity[url] = f"OK ({resp.status_code})"
-            except Exception as e:
-                connectivity[url] = f"Failed: {e}"
-                
-    return {
-        "status": "online",
-        "dns": dns_results,
-        "connectivity": connectivity,
-        "timestamp": datetime.now()
-    }
-
-
 @app.get("/status/{task_id}")
 async def get_task_status(task_id: str):
     if task_id not in tasks:

src/transcription/audio_downloader.py

@@ -4,10 +4,44 @@ Uses yt-dlp for robust YouTube video handling.
 """
 
 import re
+import socket
 from pathlib import Path
 from typing import Dict, Optional
 import yt_dlp
 
+try:
+    import dns.resolver
+    HAS_DNSPYTHON = True
+except ImportError:
+    HAS_DNSPYTHON = False
+
+# --- DNS Monkeypatch (v6) ---
+# Some cloud hosts (Hugging Face) have broken DNS resolvers.
+# This patch forces a fallback to Google DNS (8.8.8.8) if the system resolver fails.
+_orig_getaddrinfo = socket.getaddrinfo
+
+def _custom_getaddrinfo(host, port, family=0, type=0, proto=0, flags=0):
+    try:
+        return _orig_getaddrinfo(host, port, family, type, proto, flags)
+    except socket.gaierror as e:
+        if HAS_DNSPYTHON and host and not host.startswith('127.'):
+            # logger isn't initialized yet, but we'll try to log if possible later
+            # For now, just try to resolve
+            try:
+                resolver = dns.resolver.Resolver()
+                resolver.nameservers = ['8.8.8.8', '8.8.4.4']
+                answer = resolver.resolve(host, 'A')
+                if answer:
+                    ip = str(answer[0])
+                    # Re-run original with the IP
+                    return _orig_getaddrinfo(ip, port, family, type, proto, flags)
+            except Exception:
+                pass
+        raise e
+
+socket.getaddrinfo = _custom_getaddrinfo
+# ----------------------------
+
 from src.utils.logger import setup_logger
 from src.utils.config import settings
 
@@ -26,6 +60,33 @@ class YouTubeDownloader:
         """
         self.output_dir = output_dir or settings.temp_dir
         self.output_dir.mkdir(parents=True, exist_ok=True)
+
+    def _resolve_hostname(self, hostname: str) -> Optional[str]:
+        """
+        Attempt to resolve hostname using system DNS, then fallback to manual DNS (v6).
+        """
+        # 1. Try system DNS
+        try:
+            return socket.gethostbyname(hostname)
+        except Exception as e:
+            logger.warning(f"[v6] System DNS failed for {hostname}: {e}")
+            
+        # 2. Fallback to manual DNS query to 8.8.8.8
+        if HAS_DNSPYTHON:
+            try:
+                resolver = dns.resolver.Resolver()
+                resolver.nameservers = ['8.8.8.8', '8.8.4.4']
+                resolver.timeout = 5
+                resolver.lifetime = 10
+                answer = resolver.resolve(hostname, 'A')
+                if answer:
+                    ip = str(answer[0])
+                    logger.info(f"[v6] Manual DNS resolved {hostname} to {ip}")
+                    return ip
+            except Exception as e:
+                logger.error(f"[v6] Manual DNS also failed for {hostname}: {e}")
+        
+        return None
         
     @staticmethod
     def is_valid_youtube_url(url: str) -> bool: