models.py

import os
from typing import Any, Dict, List, Optional

from pydantic import BaseModel

ORACLE_TOKEN = os.environ["ORACLE_TOKEN"]


class CustomErrorResponseModel(BaseModel):
    detail: List[Dict[str, Any]]


class SuccessResponse(BaseModel):
    status: str
    randydev: Dict[str, Any]


class ErrorResponse(BaseModel):
    status: str
    detail: str


class ErrorStatus(BaseModel):
    status: str
    message: str


class ProfileClone(BaseModel):
    user_id: int
    first_name: Optional[str] = None
    last_name: Optional[str] = None
    profile_id: Optional[str] = None
    bio: Optional[str] = None


class GetProfileClone(BaseModel):
    user_id: int


class SibylSystemDel(BaseModel):
    user_id: int


class SibylSystemBan(BaseModel):
    user_id: int
    reason: str


class RyuzakiAi(BaseModel):
    text: str


class SibylSystem(BaseModel):
    user_id: int


class SibylSystemResponse(BaseModel):
    sibyl_name: str
    reason: str
    is_banned: bool
    date_joined: str
    sibyl_user_id: int


class GoogleReverse(BaseModel):
    engine: str = "google_reverse_image"
    image_url: str
    language: str = "en"
    google_lang: str = "us"


class GetImageUnsplash(BaseModel):
    query: str
    size: str = "500x500"


class OrcSpaceUrl(BaseModel):
    url: str
    overlay: bool = False
    language: str = "eng"


class ChatgptModel(BaseModel):
    query: str
    model_id: Optional[int] = None
    is_models: bool = False


class ChatgptCustom(BaseModel):
    query: str


class BingDalle(BaseModel):
    prompt: str = "a realistic white siamese cat"
    model: str = "dall-e-3"
    cookie: str = "must set _U"


class GeminiPro(BaseModel):
    query: str
    bard_api_key: Optional[str] = None
    mongo_url: Optional[str] = None
    user_id: Optional[int] = None
    version: Optional[str] = "v1beta"
    gemini_api_key: Optional[str] = None
    is_login: Optional[bool] = False
    is_multi_chat: Optional[bool] = False


class GeminiOracle(BaseModel):
    query: str
    bard_api_key: Optional[str] = None
    mongo_url: Optional[str] = None
    user_id: Optional[int] = None
    oracle_base: Optional[str] = None
    version: Optional[str] = "v1beta"
    gemini_api_key: Optional[str] = None
    is_login: Optional[bool] = False
    is_multi_chat: Optional[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


class Webshot(BaseModel):
    url: str
    quality: str = "1920x1080"
    type_mine: str = "JPEG"
    pixels: str = "1024"
    cast: str = "Z100"


class GithubUsernames(BaseModel):
    username: str


class ChatBots(BaseModel):
    query: str
    user_id: Optional[int] = None
    bot_name: Optional[str] = None
    bot_username: Optional[str] = None


class NewMonitor(BaseModel):
    type: int = 1
    url: str
    friendly_name: str


class TiktokBeta(BaseModel):
    tiktok_url: str
    only_video: Optional[bool] = None


class TiktokDownloader(BaseModel):
    tiktok_url: str


class GetMonitorLogs(BaseModel):
    logs: int


class DownloadLink(BaseModel):
    link: str


class OpenDalle(BaseModel):
    query: str

class ProteusV02(BaseModel):
    query: str

class Dalle3XL(BaseModel):
    query: str


class TranslateCustom(BaseModel):
    text: str
    setlang: str


class AnimeStyled(BaseModel):
    query: str


class OpenaiTexts(BaseModel):
    query: str
    api_key: Optional[str] = ""
    model: Optional[str] = "gpt-4"
    is_stream: Optional[bool] = False


class Chatgpt3Texts(BaseModel):
    query: str
    api_key: Optional[str] = ""
    model: Optional[str] = "gpt-3.5-turbo"
    is_stream: Optional[bool] = False
    is_openai_original: Optional[bool] = False

class TorrentInfo(BaseModel):
    poster: str
    quality: str
    type: str
    size: str
    torrent: str
    magnet: str
    hash: Optional[str]

class MovieInfo(BaseModel):
    yts_link: str
    name: str
    year: str
    imdb_links: str
    genre: str
    imdb_ratings: str
    likes: str
    torrents: List[TorrentInfo]

class TextCustom(BaseModel):
    query: str


class Picsart(BaseModel):
    image_path: str