Upload main.py

main.py

@@ -36,6 +36,9 @@ from datetime import datetime as dt
 from dotenv import load_dotenv
 from bs4 import BeautifulSoup
 
+from PIL import Image, ImageEnhance
+from RyuzakiLib import AsyicXSearcher
+
 from typing import *
 from typing_extensions import Annotated
 from typing import Annotated, Union
@@ -87,7 +90,6 @@ import functions as code
 from fluxai import router as fluxai_router
 from whisper import router as whisper_router
 from instagram import router as instagram_router
-from tools import router as tools_router
 from driver import YoutubeDriver
 from yt_dlp import YoutubeDL
 
@@ -115,6 +117,7 @@ SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
 SOURCE_MONITOR_URL = os.environ["SOURCE_MONITOR_URL"]
 SOURCE_OPENAI_ACCESS_URL = os.environ["SOURCE_OPENAI_ACCESS_URL"]
 SOURCE_PICSART_URL = os.environ["SOURCE_PICSART_URL"]
+TOOLS_NEW_URL = os.environ["TOOLS_NEW_URL"]
 
 # api keys
 REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
@@ -136,7 +139,6 @@ app = FastAPI(docs_url=None, redoc_url="/")
 app.include_router(fluxai_router, prefix="/api/v1")
 app.include_router(whisper_router, prefix="/api/v1")
 app.include_router(instagram_router, prefix="/api/v1")
-app.include_router(tools_router, prefix="/api/v1")
 
 timeout = 100
 
@@ -245,6 +247,160 @@ async def youtube_api(payload: YouTubeBase):
         return SuccessResponse(
             status="False",
             randydev={"error": f"An error occurred: {str(e)}"}
+
+class XnxxSearch(BaseModel):
+    q: str
+
+class XnxxLinks(BaseModel):
+    url: str
+
+async def tools_search(
+    name=None,
+    parameter=None,
+    ai_model=None,
+    upload_check=False
+):
+    if upload_check:
+        TOOLS_API_URL = f"{TOOLS_NEW_URL}/{ai_model}/{parameter}"
+        return TOOLS_API_URL
+    else:
+        TOOLS_API_URL = f"{TOOLS_NEW_URL}/tools/{name}?{parameter}"
+        return TOOLS_API_URL
+
+async def toanime(input):
+    url = await tools_search(
+        ai_model="ai",
+        parameter="toanime",
+        upload_check=True
+    )
+    try:
+        image = Image.open(input)
+        buffer = io.BytesIO()
+        image.save(buffer, format='JPEG')
+        buffer.seek(0)
+        files = {
+            'image': ('toanime.jpg', buffer, 'image/jpeg')
+        }
+        response = requests.post(
+            url,
+            files=files,
+            headers={
+            'accept': 'application/json'
+            }
+        )
+        if response.status_code == 200:
+            data = response.json()
+            res = {
+                "image_data": data['result'],
+                "image_size": data['size']
+            }
+            return res
+        else:
+            return 'Identifikasi Gagal'
+    except Exception:
+        return 'Identifikasi Gagal'
+
+@app.post("/akeno/toanime", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def toanime_endpoint(
+    file: UploadFile = File(...),
+    api_key: str = Depends(validate_api_key)
+):
+    file_path = f"./uploads/{file.filename}"
+    try:
+        with open(file_path, "wb") as f:
+            f.write(await file.read())
+    except Exception:
+        raise HTTPException(status_code=500, detail="Failed to save file")
+    try:
+        response = await toanime(file_path)
+        url_image = response["image_data"]
+        return SuccessResponse(
+            status="True",
+            randydev={"url": url_image}
+        )
+    except Exception:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "Error during processing"}
+        )
+    finally:
+        if os.path.exists(file_path):
+            os.remove(file_path)
+
+@app.post("/akeno/xnxxsearch", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def xnxx_search(
+    payload: XnxxSearch,
+    api_key: None = Depends(validate_api_key)
+):
+    url = await tools_search(name="xnxxsearch", parameter=f"q={payload.query}")
+    try:
+        response = await AsyicXSearcher.search(url, re_json=True)
+        result = response["result"]
+        return SuccessResponse(
+            status="True",
+            randydev={"results": result}
+        )
+    except:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "Error fucking"}
+        )
+
+@app.post("/akeno/xnxx-dl", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def xnxx_download(
+    payload: XnxxLinks,
+    api_key: None = Depends(validate_api_key)
+):
+    url = await tools_search(name="xnxxdl", parameter=f"url={payload.link}")
+    try:
+        response = await AsyicXSearcher.search(url, re_json=True)
+        result = response["result"]
+        return SuccessResponse(
+            status="True",
+            randydev={"results": result}
+        )
+    except:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "Error fucking"}
+        )
+
+@app.post("/akeno/xnxx-videodl", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def xnxx_videodl(
+    payload: XnxxLinks,
+    api_key: None = Depends(validate_api_key)
+):
+    url = await tools_search(name="xvideosdl", parameter=f"url={payload.link}")
+    try:
+        response = await AsyicXSearcher.search(url, re_json=True)
+        result = response["result"]
+        return SuccessResponse(
+            status="True",
+            randydev={"results": result}
+        )
+    except:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "Error fucking"}
+        )
+
+@app.post("/akeno/instagramdl", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
+async def instagramdl(
+    payload: XnxxLinks,
+    api_key: None = Depends(validate_api_key)
+):
+    url = await tools_search(name="instagramdl", parameter=f"url={payload.link}")
+    try:
+        response = await AsyicXSearcher.search(url, re_json=True)
+        result = response["result"]
+        return SuccessResponse(
+            status="True",
+            randydev={"results": result}
+        )
+    except:
+        return SuccessResponse(
+            status="False",
+            randydev={"error": "Error fucking"}
         )
 
 @app.post("/uploadfile/")