Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import Response
+from fastapi.responses import Response, HTMLResponse
+from fastapi.openapi.docs import get_swagger_ui_html
 import yt_dlp
 import whisper
 import tempfile
@@ -7,38 +8,101 @@ import os
 import io
 from diffusers import StableDiffusionPipeline
 
-app = FastAPI(title="Silent Tech Utils API")
+# --- 1. BEAUTIFUL API METADATA ---
+api_description = """
+<div style="text-align: center; margin-top: 20px;">
+    <img src="https://i.ibb.co/C5nyyyXH/cccfe44a8d63663a60eed6f0300a8b44.jpg" width="150" style="border-radius: 15px; box-shadow: 0 4px 8px rgba(0,0,0,0.5);">
+    <h2>Welcome to the Silent Tech Utility Engine</h2>
+</div>
 
+This is the **Private Backend API** for the Silent Bot network.  
+It is completely independent, uncensored, and runs on a dedicated 16GB RAM Linux container.
+
+### 🚀 Available Microservices
+* **Media Downloader:** Bypass website protections to get direct MP4/MP3 links.
+* **Voice AI:** Local voice-to-text transcription using OpenAI's Whisper.
+* **Image Studio:** Uncensored Text-to-Image generation using Stable Diffusion.
+"""
+
+tags_metadata = [
+    {"name": "System", "description": "Server health status."},
+    {"name": "Media Downloader", "description": "Extract raw media URLs from social platforms."},
+    {"name": "Voice AI", "description": "Transcribe audio files perfectly."},
+    {"name": "Image AI", "description": "Generate uncensored images."}
+]
+
+app = FastAPI(
+    title="🌟 Silent Tech Private API",
+    description=api_description,
+    version="2.0.0",
+    openapi_tags=tags_metadata,
+    docs_url=None, # Disable the ugly default docs
+    redoc_url=None
+)
+
+# --- 2. DARK MODE CUSTOM DOCS ---
+@app.get("/docs", include_in_schema=False)
+async def custom_swagger_ui_html():
+    return get_swagger_ui_html(
+        openapi_url=app.openapi_url,
+        title="Silent Tech API - Docs",
+        swagger_favicon_url="https://i.ibb.co/C5nyyyXH/cccfe44a8d63663a60eed6f0300a8b44.jpg",
+        # Injecting a sleek hacker dark mode theme!
+        swagger_css_url="https://cdn.jsdelivr.net/gh/Itz-fork/Swagger-Dark-Theme@main/SwaggerDark.css"
+    )
+
+# --- 3. LOAD AI MODELS ---
 print("Loading Whisper Voice AI...")
 voice_model = whisper.load_model("base")
 
 print("Loading Uncensored Image AI...")
-# safety_checker=None completely disables the NSFW and violence filters!
 image_model = StableDiffusionPipeline.from_pretrained("prompthero/openjourney", safety_checker=None)
 image_model.to("cpu")
 
-@app.get("/")
+# --- 4. API ENDPOINTS ---
+@app.get("/", tags=["System"])
 def read_root():
-    return {"status": "Silent Utils API is ONLINE"}
+    return {"status": "Silent Utils API is ONLINE", "version": "2.0.0"}
 
-@app.get("/api/download")
+@app.get("/api/download", tags=["Media Downloader"])
 def download_media(url: str):
     """Bypasses protections and gets the direct raw MP4/MP3 link."""
-    ydl_opts = {'format': 'best', 'quiet': True, 'no_warnings': True, 'skip_download': True}
+    clean_url = url.strip() 
+    
+    ydl_opts = {
+        'format': 'best', 
+        'quiet': True, 
+        'no_warnings': True, 
+        'skip_download': True,
+        'nocheckcertificate': True # Fixes strict SSL blocks
+    }
+    
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=False)
+            info = ydl.extract_info(clean_url, download=False)
+            
+            # If the user accidentally links a playlist, grab the first video
+            if 'entries' in info:
+                info = info['entries'][0]
+                
+            final_url = info.get('url')
+            if not final_url and 'requested_downloads' in info:
+                final_url = info['requested_downloads'][0].get('url')
+                
+            if not final_url:
+                raise Exception("Could not extract the raw video URL.")
+                
             return {
                 "success": True,
-                "title": info.get('title'),
-                "download_url": info.get('url') # The direct raw link!
+                "title": info.get('title', 'Silent Media'),
+                "download_url": final_url
             }
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
-@app.post("/api/transcribe")
+@app.post("/api/transcribe", tags=["Voice AI"])
 async def transcribe_audio(file: UploadFile = File(...)):
-    """Converts WhatsApp voice notes to text."""
+    """Converts WhatsApp voice notes (or any audio) to text."""
     try:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".ogg") as temp_audio:
             temp_audio.write(await file.read())
@@ -50,18 +114,16 @@ async def transcribe_audio(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.get("/api/image")
+@app.get("/api/image", tags=["Image AI"])
 def generate_image(prompt: str):
     """Generates an uncensored image and returns it directly as a PNG."""
     try:
-        # Generate the image (15 steps is faster for CPU)
-        image = image_model(prompt, num_inference_steps=15).images[0]
+        # Optimized for fast CPU generation
+        image = image_model(prompt, num_inference_steps=8, height=384, width=384).images[0]
         
-        # Convert it to a raw PNG image file
         img_bytes = io.BytesIO()
         image.save(img_bytes, format="PNG")
         
-        # Send the actual image picture directly to the bot, not just text!
         return Response(content=img_bytes.getvalue(), media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))