Update app.py

app.py

@@ -5,9 +5,12 @@ import whisper
 import tempfile
 import os
 import io
+import urllib.request
+import urllib.parse
+import json
 from diffusers import StableDiffusionPipeline
 
-# --- 1. METADATA ---
+# --- 1. METADATA & NEON UI ---
 api_description = """
 <div style="text-align: center; margin-top: 20px; border-bottom: 1px solid #333; padding-bottom: 20px;">
     <img src="https://i.ibb.co/C5nyyyXH/cccfe44a8d63663a60eed6f0300a8b44.jpg" width="150" style="border-radius: 15px; box-shadow: 0 0 20px rgba(0, 255, 204, 0.6); border: 2px solid #00ffcc;">
@@ -32,11 +35,10 @@ app = FastAPI(
     description=api_description,
     version="2.0.0",
     openapi_tags=tags_metadata,
-    docs_url=None, # We disable the default so we can use our custom Neon UI
+    docs_url=None, 
     redoc_url=None
 )
 
-# --- 2. CUSTOM NEON UI ---
 @app.get("/docs", include_in_schema=False)
 async def neon_swagger_ui():
     html = """
@@ -48,39 +50,25 @@ async def neon_swagger_ui():
         <link rel="stylesheet" type="text/css" href="https://cdn.jsdelivr.net/npm/swagger-ui-dist@5/swagger-ui.css" />
         <link rel="icon" type="image/png" href="https://i.ibb.co/C5nyyyXH/cccfe44a8d63663a60eed6f0300a8b44.jpg" />
         <style>
-            /* 🚀 CUSTOM NEON CYBERPUNK CSS 🚀 */
             body { background-color: #050505 !important; color: #00ffcc !important; font-family: 'Segoe UI', Tahoma, sans-serif; }
             .swagger-ui .info .title { color: #00ffcc !important; text-shadow: 0 0 10px rgba(0, 255, 204, 0.7); }
             .swagger-ui .info p { color: #a0aec0 !important; }
             .swagger-ui .info h1, .swagger-ui .info h2, .swagger-ui .info h3, .swagger-ui .info h4, .swagger-ui .info h5 { color: #00ffcc !important; }
             .swagger-ui .info a { color: #bd00ff !important; text-shadow: 0 0 8px rgba(189, 0, 255, 0.7); }
             .swagger-ui .scheme-container { background-color: #0a0a0a !important; box-shadow: 0 0 15px rgba(0, 255, 204, 0.1); border-bottom: 1px solid #00ffcc; }
-            
-            /* GET Requests (Neon Cyan) */
             .swagger-ui .opblock.opblock-get { background: rgba(0, 255, 204, 0.05) !important; border: 1px solid #00ffcc !important; box-shadow: 0 0 10px rgba(0, 255, 204, 0.2); border-radius: 8px; }
             .swagger-ui .opblock.opblock-get .opblock-summary-method { background: #00ffcc !important; color: #000 !important; font-weight: bold; }
-            
-            /* POST Requests (Neon Purple) */
             .swagger-ui .opblock.opblock-post { background: rgba(189, 0, 255, 0.05) !important; border: 1px solid #bd00ff !important; box-shadow: 0 0 10px rgba(189, 0, 255, 0.2); border-radius: 8px; }
             .swagger-ui .opblock.opblock-post .opblock-summary-method { background: #bd00ff !important; color: #fff !important; font-weight: bold; }
-            
-            /* Glowing Execute Button */
             .swagger-ui .btn.execute { background-color: #00ffcc !important; color: #000 !important; border: none !important; box-shadow: 0 0 15px rgba(0, 255, 204, 0.6) !important; font-weight: bold; transition: 0.3s; }
             .swagger-ui .btn.execute:hover { box-shadow: 0 0 25px rgba(0, 255, 204, 1) !important; }
             .swagger-ui .btn { color: #00ffcc !important; border-color: #00ffcc !important; }
-            
-            /* Text & Inputs */
             .swagger-ui .opblock-body pre.microlight { background-color: #000 !important; border: 1px solid #333 !important; border-radius: 8px; color: #fff !important; }
-            .swagger-ui .parameters-col_name { color: #bd00ff !important; }
             .swagger-ui input[type=text], .swagger-ui input[type=file] { background: #000 !important; color: #00ffcc !important; border: 1px solid #bd00ff !important; border-radius: 4px; padding: 5px; }
             .swagger-ui .responses-inner h4, .swagger-ui .responses-inner h5 { color: #00ffcc !important; }
             .swagger-ui svg { fill: #00ffcc !important; }
-            
-            /* Models Section */
             .swagger-ui section.models { border: 1px solid #333 !important; background: #0a0a0a !important; border-radius: 8px;}
             .swagger-ui section.models h4 { color: #00ffcc !important; border-bottom: 1px solid #333; }
-            .swagger-ui .model-title { color: #bd00ff !important; }
-            .swagger-ui .model { color: #a0aec0 !important; }
         </style>
     </head>
     <body>
@@ -103,8 +91,7 @@ async def neon_swagger_ui():
     """
     return HTMLResponse(html)
 
-
-# --- 3. LOAD AI MODELS ---
+# --- 2. LOAD AI MODELS ---
 print("Loading Whisper Voice AI...")
 voice_model = whisper.load_model("base")
 
@@ -112,8 +99,7 @@ print("Loading Uncensored Image AI...")
 image_model = StableDiffusionPipeline.from_pretrained("prompthero/openjourney", safety_checker=None)
 image_model.to("cpu")
 
-
-# --- 4. API ENDPOINTS ---
+# --- 3. API ENDPOINTS ---
 @app.get("/", tags=["System"])
 def read_root():
     return {"status": "Silent Utils API is ONLINE", "version": "2.0.0"}
@@ -123,6 +109,24 @@ def download_media(url: str):
     """Bypasses protections and gets the direct raw MP4/MP3 link."""
     clean_url = url.strip() 
     
+    # 💥 FIREWALL BYPASS: If it's YouTube, route it through an external proxy API!
+    if "youtube.com" in clean_url or "youtu.be" in clean_url:
+        try:
+            bypass_url = f"https://api.bk9.site/yt/mp4?url={urllib.parse.quote(clean_url)}"
+            req = urllib.request.Request(bypass_url, headers={'User-Agent': 'Mozilla/5.0'})
+            with urllib.request.urlopen(req) as response:
+                data = json.loads(response.read().decode())
+                if data.get("status") and "BK9" in data:
+                    return {
+                        "success": True,
+                        "title": data["BK9"].get("title", "YouTube Video"),
+                        "download_url": data["BK9"].get("url")
+                    }
+        except Exception as bypass_e:
+            print("Bypass failed:", str(bypass_e))
+            # If bypass fails, fall back to yt-dlp just in case
+    
+    # Standard yt-dlp for TikTok, Instagram, Twitter, etc.
     ydl_opts = {
         'format': 'best', 
         'quiet': True,