Upload 2 files

main.py

@@ -70,12 +70,14 @@ from bardapi import Bard
 
 from models import *
 
+from gpytranslate import SyncTranslator
+
 import logging
 import functions as code
 
 logging.basicConfig(level=logging.ERROR)
 
-# I DON'T KNOW LIKE THIS HACKER
+
 load_dotenv()
 
 # Database url
@@ -86,6 +88,7 @@ SOURCE_UNSPLASH_URL = os.environ["SOURCE_UNSPLASH_URL"]
 SOURCE_OCR_URL = os.environ["SOURCE_OCR_URL"]
 SOURCE_ALPHA_URL = os.environ["SOURCE_ALPHA_URL"]
 SOURCE_OPENDALLE_URL = os.environ["SOURCE_OPENDALLE_URL"]
+SOURCE_ANIME_STYLED_URL = os.environ["SOURCE_ANIME_STYLED_URL"]
 SOURCE_WAIFU_URL = os.environ["SOURCE_WAIFU_URL"]
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
@@ -106,6 +109,8 @@ client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
 collection = db["users"]
 
+trans = SyncTranslator()
+
 app = FastAPI(docs_url=None, redoc_url="/")
 
 def get_all_api_keys():
@@ -345,7 +350,7 @@ def sibyl_system(
     api_key: None = Depends(validate_api_key)
 ):
     sibyl_result = get_sibyl_system_banned(item.user_id)
-    if sibyl_result is not None:
+    if sibyl_result:
         sibyl_name, reason, is_banned, date_joined, sibyl_user_id = sibyl_result
         return SuccessResponse(
             status="True",
@@ -365,6 +370,25 @@ def sibyl_system(
             }
         )
 
+@app.get("/ryuzaki/translate", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def get_translate(
+    item: TranslateCustom,
+):
+    try:
+        source = trans.detect(item.text)
+        translation = trans(item.text, sourcelang=source, targetlang=item.setlang)
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "translation": translation.text,
+                "translation_original": item.text
+            }
+        )
+    except:
+        return SuccessResponse(
+            status="True",
+            randydev={"message": "Error not responding"})
+
 @app.get("/ryuzaki/ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def ryuzaki_ai(
     item: RyuzakiAi,
@@ -388,62 +412,6 @@ def ryuzaki_ai(
     except Exception:
         return {"status": "false", "message": "Internal server error"}
 
-@app.post("/ryuzaki/opendalle")
-async def open_dalle(
-    item: OpenDalle,
-    api_key: None = Depends(validate_api_key)
-):
-    API_URL = SOURCE_OPENDALLE_URL
-    try:
-        payload = {"inputs": item.query}
-        headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-    except requests.exceptions.RequestException:
-        raise HTTPException(
-            status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
-        )
-    try:
-        encoded_string = base64.b64encode(response.content).decode("utf-8")
-    except Exception:
-        raise HTTPException(
-            status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
-        )
-    headers = {"Content-Type": "image/jpeg"}
-    return Response(
-        content=encoded_string,
-        media_type="image/jpeg",
-        headers=headers
-    )
-
-@app.get("/ryuzaki/unsplash")
-async def get_image_unsplash(item: GetImageUnsplash):
-    url = SOURCE_UNSPLASH_URL
-    image_url = f"{url}/?{item.query}/{item.size}"
-    try:
-        response = requests.get(image_url)
-        response.raise_for_status()
-    except requests.exceptions.RequestException:
-        raise HTTPException(
-            status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
-        )
-    try:
-        encoded_string = base64.b64encode(response.content).decode("utf-8")
-    except Exception:
-        raise HTTPException(
-            status_code=500,
-            detail=CustomErrorResponseModel(detail=[{}])
-        )
-    headers = {"Content-Type": "image/jpeg"}
-    return Response(
-        content=encoded_string,
-        media_type="image/jpeg",
-        headers=headers
-    )
-
 @app.get("/ryuzaki/reverse", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def google_reverse(
     item: GoogleReverse,
@@ -506,6 +474,119 @@ def ocr_space_url(
     except (json.JSONDecodeError, KeyError):
         return "Error parsing the OCR response."
 
+@app.get("/ryuzaki/freechatgpt-beta", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def free_chatgpt4_beta(item: ChatgptCustom):
+    try:
+        response = RendyDevChat(item.query).get_response_beta(joke=True)
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": response
+            }
+        )
+    except:
+        return {"status": "false", "message": "Error response."}
+
+@app.get("/ryuzaki/freechatgpt-bing", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def free_chatgpt4_bing(item: ChatgptCustom):
+    try:
+        response = RendyDevChat(query).get_response_bing(bing=True)
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": response
+            }
+        )
+    except:
+        return {"status": "false", "message": "Error response."}
+
+@app.post("/ryuzaki/opendalle")
+def open_dalle(
+    item: OpenDalle,
+    api_key: None = Depends(validate_api_key)
+):
+    API_URL = SOURCE_OPENDALLE_URL
+    try:
+        payload = {"inputs": item.query}
+        headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        response.raise_for_status()
+    except requests.exceptions.RequestException:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    headers = {"Content-Type": "image/jpeg"}
+    return Response(
+        content=encoded_string,
+        media_type="image/jpeg",
+        headers=headers
+    )
+
+@app.post("/ryuzaki/anime-styled")
+def Anime_Styled(
+    item: AnimeStyled,
+    api_key: None = Depends(validate_api_key)
+):
+    API_URL = SOURCE_ANIME_STYLED_URL
+    try:
+        payload = {"inputs": item.query}
+        headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
+        response = requests.post(API_URL, headers=headers, json=payload)
+        response.raise_for_status()
+    except requests.exceptions.RequestException:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    headers = {"Content-Type": "image/jpeg"}
+    return Response(
+        content=encoded_string,
+        media_type="image/jpeg",
+        headers=headers
+    )
+
+@app.post("/ryuzaki/unsplash")
+def image_unsplash(item: GetImageUnsplash):
+    url = SOURCE_UNSPLASH_URL
+    image_url = f"{url}/?{item.query}/{item.size}"
+    try:
+        response = requests.get(image_url)
+        response.raise_for_status()
+    except requests.exceptions.RequestException:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception:
+        raise HTTPException(
+            status_code=500,
+            detail=CustomErrorResponseModel(detail=[{}])
+        )
+    headers = {"Content-Type": "image/jpeg"}
+    return Response(
+        content=encoded_string,
+        media_type="image/jpeg",
+        headers=headers
+    )
+
+
 @app.post("/ryuzaki/chatgpt-model", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def chatgpt_model(item: ChatgptModel):
     url = "https://lexica.qewertyy.me/models"
@@ -536,32 +617,6 @@ def chatgpt_model(item: ChatgptModel):
             }
         )
 
-@app.get("/ryuzaki/freechatgpt-beta", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def free_chatgpt4_beta(item: ChatgptCustom):
-    try:
-        response = RendyDevChat(item.query).get_response_beta(joke=True)
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": response
-            }
-        )
-    except:
-        return {"status": "false", "message": "Error response."}
-
-@app.get("/ryuzaki/freechatgpt-bing", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def free_chatgpt4_bing(item: ChatgptCustom):
-    try:
-        response = RendyDevChat(query).get_response_bing(bing=True)
-        return SuccessResponse(
-            status="True",
-            randydev={
-                "message": response
-            }
-        )
-    except:
-        return {"status": "false", "message": "Error response."}
-
 @app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(
     item: ChatgptCustom,

models.py

@@ -124,3 +124,10 @@ class DownloadLink(BaseModel):
 
 class OpenDalle(BaseModel):
     query: str
+
+class TranslateCustom(BaseModel):
+    text: str
+    setlang: str
+
+class AnimeStyled(BaseModel):
+    query: str