Update main.py

main.py

@@ -17,7 +17,6 @@ app = FastAPI(
     version="1.0.0"
 )
 
-# Allow your bots to connect from anywhere
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -25,7 +24,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Background task to clean up storage after sending the file
 def cleanup_files(files: list):
     for f in files:
         if os.path.exists(f):
@@ -35,12 +33,12 @@ def cleanup_files(files: list):
                 pass
 
 # ==========================================
-# 📡 ENDPOINTS
+# 📡 ENDPOINTS (WHITE-LABELED)
 # ==========================================
 
 @app.get("/")
 def read_root():
-    return {"status": "Online", "message": "SILENT TECH Engine is running. Visit /docs for the API Documentation!"}
+    return {"status": "Online", "creator": "SILENT TECH", "message": "Visit /docs for the API Documentation!"}
 
 @app.get("/api/search")
 async def search_yt(query: str):
@@ -51,6 +49,10 @@ async def search_yt(query: str):
             data = await resp.json()
             if not data.get("status"):
                 raise HTTPException(status_code=404, detail="Not found")
+            
+            # 🥷 THE NINJA MOVE: Overwrite the creator name!
+            data["creator"] = "SILENT TECH"
+            
             return data
 
 @app.get("/api/ytmp4")
@@ -62,6 +64,10 @@ async def get_ytmp4(url: str):
             data = await resp.json()
             if not data.get("success"):
                 raise HTTPException(status_code=400, detail="Failed to fetch video link")
+            
+            # 🥷 THE NINJA MOVE: Overwrite the creator name!
+            data["creator"] = "SILENT TECH"
+            
             return data
 
 @app.get("/api/ytmp3")
@@ -71,11 +77,9 @@ async def get_ytmp3(url: str, background_tasks: BackgroundTasks):
     temp_vid = f"temp_vid_{timestamp}.mp4"
     temp_aud = f"temp_aud_{timestamp}.mp3"
     
-    # Tell the API to delete these files AFTER the user finishes downloading them
     background_tasks.add_task(cleanup_files, [temp_vid, temp_aud])
 
     try:
-        # 1. Fetch the raw MP4 stream link
         async with aiohttp.ClientSession() as session:
             api_url = f"https://apis.davidcyril.name.ng/download/ytmp4?url={urllib.parse.quote(url)}"
             async with session.get(api_url) as resp:
@@ -86,15 +90,13 @@ async def get_ytmp3(url: str, background_tasks: BackgroundTasks):
                 download_url = data["result"]["download_url"]
                 title = data["result"].get("title", "Silent_Tech_Audio").replace("/", "_")
 
-            # 2. Download the MP4 file to Hugging Face Disk
             async with session.get(download_url) as video_resp:
                 with open(temp_vid, 'wb') as f:
                     while True:
-                        chunk = await video_resp.content.read(2 * 1024 * 1024) # 2MB chunks
+                        chunk = await video_resp.content.read(2 * 1024 * 1024)
                         if not chunk: break
                         f.write(chunk)
         
-        # 3. SuperSonic FFmpeg Conversion
         command = [
             "ffmpeg", "-y", "-i", temp_vid, 
             "-vn", "-acodec", "libmp3lame", "-q:a", "2", 
@@ -104,7 +106,6 @@ async def get_ytmp3(url: str, background_tasks: BackgroundTasks):
         if process.returncode != 0:
             raise Exception("FFmpeg audio extraction failed.")
 
-        # 4. Stream the MP3 directly back to the user/bot
         return FileResponse(
             path=temp_aud, 
             media_type="audio/mpeg", 
@@ -112,5 +113,5 @@ async def get_ytmp3(url: str, background_tasks: BackgroundTasks):
         )
         
     except Exception as e:
-        background_tasks.add_task(cleanup_files, [temp_vid, temp_aud]) # Cleanup on error
+        background_tasks.add_task(cleanup_files, [temp_vid, temp_aud])
         raise HTTPException(status_code=500, detail=str(e))