Upload main.py

main.py

@@ -65,6 +65,8 @@ from RyuzakiLib.hackertools.chatgpt import RendyDevChat
 from RyuzakiLib.hackertools.openai import OpenAiToken
 from RyuzakiLib.mental import BadWordsList
 
+from bardapi import Bard
+
 import logging
 import functions as code
 
@@ -87,6 +89,7 @@ SOURCE_CALLI_GRAPHY_URL = os.environ["SOURCE_CALLI_GRAPHY_URL"]
 SOURCE_WHAT_GAY_URL = os.environ["SOURCE_WHAT_GAY_URL"]
 SOURCE_ASSISTANT_GOOGLE_AI = os.environ["SOURCE_ASSISTANT_GOOGLE_AI"]
 ASSISTANT_GOOGLE_API_KEYS = os.environ["ASSISTANT_GOOGLE_API_KEYS"]
+COOKIE_BARD_TOKEN = os.environ["COOKIE_BARD_TOKEN"]
 
 client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
@@ -551,6 +554,24 @@ def v1beta3_google_ai(query: str, api_key: None = Depends(validate_api_key)):
     except:
         return {"status": "false", "message": "Error response."}
 
+@app.post("/ryuzaki/gemini-ai-pro")
+def gemini_pro(
+    query: str,
+    api_key: None = Depends(validate_api_key)
+):
+    token = COOKIE_BARD_TOKEN
+    try:
+        bard = Bard(token=token)
+        message = bard.get_answer(query)["content"]
+        return {
+            "status": "true",
+            "randydev":{
+                "message": message
+            }
+        }
+    except:
+        return {"status": "false", "message": "Error response."}
+
 @app.post("/ryuzaki/v1beta2-google-ai")
 def v1beta2_google_ai(
     query: str,