g-ai

main.py

@@ -73,18 +73,25 @@ logging.basicConfig(level=logging.ERROR)
 
 # I DON'T KNOW LIKE THIS HACKER
 load_dotenv()
-REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
-OCR_API_KEY = os.environ["OCR_API_KEY"]
-ONLY_DEVELOPER_API_KEYS = os.environ["ONLY_DEVELOPER_API_KEYS"]
-HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
 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_WAIFU_URL = "https://api.waifu.pics"
 SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
+SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
+SOURCE_OPENDALLE_URL = os.environ["SOURCE_OPENDALLE_URL"]
+SOURCE_OPENAI_ACCESS_URL = os.environ["SOURCE_OPENAI_ACCESS_URL"]
 DEVELOPER_ID = os.environ["DEVELOPER_ID"]
 
+# api keys
+REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
+OCR_API_KEY = os.environ["OCR_API_KEY"]
+ONLY_DEVELOPER_API_KEYS = os.environ["ONLY_DEVELOPER_API_KEYS"]
+HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
+ASSISTANT_GOOGLE_API_KEYS = os.environ["ASSISTANT_GOOGLE_API_KEYS"]
+#COOKIE_BARD_TOKEN = os.environ["COOKIE_BARD_TOKEN"]
+
 description = """
 ~ Developed written and powered by
 - Ryuzaki Library: [Library Here](https://github.com/TeamKillerX/RyuzakiLib)
@@ -112,6 +119,15 @@ def validate_api_key_only_devs(api_key: str = Header(...)):
         raise HTTPException(status_code=401, detail="Invalid API key")
 
 
+@app.get("/UFoP/blacklist-words")
+def blacklist_words():
+    try:
+        BLACKLIST_WORDS = BadWordsList()
+        results_all = BLACKLIST_WORDS.banned_by_google(file_txt="banned_by_google.txt", storage=True)
+        return {"status": "true", "results": results_all}
+    except Exception as e:
+        return {"status": "false", "message": f"Internal server error: {str(e)}"}
+
 @app.get("/UFoP/getbanlist")
 def sibyl_get_all_banlist():
     banned_users = db.get_all_banned()
@@ -122,15 +138,6 @@ def sibyl_get_all_banlist():
         }
     }
 
-@app.get("/UFoP/blacklist-words")
-def blacklist_words():
-    try:
-        BLACKLIST_WORDS = BadWordsList()
-        results_all = BLACKLIST_WORDS.banned_by_google(file_txt="banned_by_google.txt", storage=True)
-        return {"status": "true", "results": results_all}
-    except Exception as e:
-        return {"status": "false", "message": f"Internal server error: {str(e)}"}
-
 @app.delete("/UFoP/bandel", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def sibyl_system_delete(
     item: SibylSystemDel,
@@ -207,29 +214,34 @@ def sibyl_system(
     else:
         return {"status": "false", "message": "Not Found User"}
 
-@app.get("/ryuzaki/ai")
-def ryuzaki_ai(
-    text: str = Query(..., description="text in query parameter"),
-    api_key: None = Depends(validate_api_key_only_devs)
+@app.post("/UFoP/G-AI", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def v1beta3_google_ai(
+    item: ChatgptCustom,
+    api_key: None = Depends(validate_api_key)
 ):
+    url = SOURCE_ASSISTANT_GOOGLE_AI
+    token = ASSISTANT_GOOGLE_API_KEYS
+    api_url = f"{SOURCE_ASSISTANT_GOOGLE_AI}/models/text-bison-001:generateText?key={ASSISTANT_GOOGLE_API_KEYS}"
     try:
-        response_data = code.ryuzaki_ai_text(text)
-
-        if isinstance(response_data, list) and len(response_data) > 0:
-            first_result = response_data[0]
-            if "generated_text" in first_result:
-                message = first_result["generated_text"]
-                return {
-                    "status": "true",
-                    "sukuna": {
-                        "ryuzaki_text": message
-                    }
-                }
-
-        return {"status": "false", "message": "Invalid response format"}
-
-    except Exception as e:
-        return {"status": "false", "message": f"error: {e}"}
+        headers = {"Content-Type": "application/json"}
+        data = {
+            "prompt": {
+                "text": item.query
+            }
+        }
+        response = requests.post(api_url, headers=headers, json=data)
+        response_str = response.json()
+        answer = response_str["candidates"]
+        for results in answer:
+            message = results.get("output")
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": message
+            }
+        )
+    except:
+        return {"status": "false", "message": "Error response."}
 
 @app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(query: str, size: str="500x500"):