Upload main.py

main.py

@@ -269,6 +269,20 @@ class GeminiPro(BaseModel):
     bard_api_key: Optional[str] = None
     is_login: bool=False
 
+class WaifuPics(BaseModel):
+    types: str="sfw"
+    category: str="neko"
+    media_type: Optional[str] = None
+    is_bytes: bool=False
+
+class MakeRayso(BaseModel):
+    code: str
+    title: str="Ryuzaki Dev"
+    theme: str
+    setlang: str="en"
+    auto_translate: Optional[bool] = None
+    ryuzaki_dark: bool=True
+
 @app.post("/ryuzaki/profile-clone", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def profile_clone(
     item: ProfileClone,
@@ -845,9 +859,9 @@ def chatbot(
         return {"status": "false", "message": "Error response."}
 
 @app.get("/ryuzaki/llama")
-def get_llama(query: str):
+def get_llama(item: ChatgptCustom):
     api_url = SOURCE_WHAT_GAY_URL
-    params = {"query": query}
+    params = {"query": item.query}
     x = requests.get(f"{api_url}/llama", params=params)
     if x.status_code != 200:
         return "Error api request"
@@ -864,14 +878,9 @@ def get_llama(query: str):
         return {"status": "false", "message": "Error response."}
 
 @app.get("/ryuzaki/waifu")
-def waifu_pics(
-    types: str="sfw",
-    category: str="neko",
-    media_type: str=None,
-    is_bytes: bool=False
-):
-    waifu_api = f"{SOURCE_WAIFU_URL}/{types}"
-    waifu_param = f"{waifu_api}/{category}"
+def waifu_pics(item: WaifuPics):
+    waifu_api = f"{SOURCE_WAIFU_URL}/{item.types}"
+    waifu_param = f"{waifu_api}/{item.category}"
 
     response = requests.get(waifu_param)
 
@@ -883,13 +892,13 @@ def waifu_pics(
     except Exception as e:
         return f"Error request {e}"
     if waifu_image_url:
-        if is_bytes:
+        if item.is_bytes:
             try:
                 response_two = requests.get(waifu_image_url)
                 response_two.raise_for_status()
             except requests.exceptions.RequestException:
                 raise HTTPException(status_code=500, detail="Internal server error")
-            return StreamingResponse(BytesIO(response_two.content), media_type=media_type)
+            return StreamingResponse(BytesIO(response_two.content), media_type=item.media_type)
         else:
             return {
                 "status": "true",
@@ -901,29 +910,22 @@ def waifu_pics(
         return {"status": "false", "message": "Error response."}
 
 @app.get("/ryuzaki/rayso")
-def make_rayso(
-    code=None,
-    title: str="Ryuzaki Dev",
-    theme: str=None,
-    setlang: str="en",
-    auto_translate: bool=None,
-    ryuzaki_dark: bool=None
-):
+def make_rayso(item: MakeRayso):
     trans = SyncTranslator()
     api_url = b64decode("aHR0cHM6Ly9hcGkuc2Fmb25lLm1lL3JheXNv").decode("utf-8")
-    if auto_translate:
-        source = trans.detect(code)
-        translation = trans(code, sourcelang=source, targetlang=setlang)
+    if item.auto_translate:
+        source = trans.detect(item.code)
+        translation = trans(item.code, sourcelang=source, targetlang=item.setlang)
         code = translation.text
     else:
-        code = code
-    if ryuzaki_dark:
+        code = item.code
+    if item.ryuzaki_dark:
         x = requests.post(
             f"{api_url}",
             json={
                 "code": code,
-                "title": title,
-                "theme": theme,
+                "title": item.title,
+                "theme": item.theme,
                 "darkMode": True
             }
         )
@@ -945,8 +947,8 @@ def make_rayso(
             f"{api_url}",
             json={
                 "code": code,
-                "title": title,
-                "theme": theme,
+                "title": item.title,
+                "theme": item.theme,
                 "darkMode": False
             }
         )