Update main.py

main.py

@@ -26,6 +26,9 @@ gs_code_app = GSCodeApp()
 from app.ht_noti import HentaiApp
 hentai_app = HentaiApp()
 
+from app.ht_rd import HentaiRandomApp
+hentai_random_app = HentaiRandomApp()
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     await invite_app.start()
@@ -33,12 +36,14 @@ async def lifespan(app: FastAPI):
     await gs_app.start()
     await gs_code_app.start()
     await hentai_app.start()
+    await hentai_random_app.start()
     yield
     await invite_app.stop()
     await lq_app.stop()
     await gs_app.stop()
     await gs_code_app.stop()
     await hentai_app.stop()
+    await hentai_random_app.stop()
 
 app = FastAPI(lifespan=lifespan)
 
@@ -118,7 +123,20 @@ async def hentai_newest(
 ):
     key_to_check = x_api_key or apikey
     return await hentai_app.get_new(key_to_check, API_KEY)
-    
+
+@app.get("/api/v1/hentai/random")
+async def get_hentai_random(
+    limit: int = Query(5, ge=1, le=20),
+    x_api_key: str = Header(None)
+):
+    if API_KEY and x_api_key != API_KEY:
+        raise HTTPException(status_code=403, detail="Cút!")
+        
+    res = await hentai_random_app.get_random(limit)
+    if not res["ok"]:
+        raise HTTPException(status_code=500, detail=res["error"])
+    return res
+
 @app.get("/")
 def home():
     uptime_seconds = int(time.time() - START_TIME)