Upload main.py

main.py

@@ -85,6 +85,8 @@ SOURCE_TIKTOK_WTF_URL = os.environ["SOURCE_TIKTOK_WTF_URL"]
 SOURCE_TIKTOK_TECH_URL = os.environ["SOURCE_TIKTOK_TECH_URL"]
 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"]
 
 client_mongo = MongoClient(MONGO_URL)
 db = client_mongo["tiktokbot"]
@@ -523,6 +525,33 @@ def free_chatgpt4_bing(query: str):
     except:
         return {"status": "false", "message": "Error response."}
 
+@app.post("/ryuzaki/google-ai")
+def google_ai(query: str, api_key: None = Depends(validate_api_key)):
+    url = SOURCE_ASSISTANT_GOOGLE_AI
+    token = ASSISTANT_GOOGLE_API_KEYS
+    try:
+        headers = {"Content-Type": "application/json"}
+        data = {
+            "prompt": {
+                "text": query
+            }
+        }
+        response = requests.post(f"{url}?key={token}", headers=headers, json=data)
+        if response.status_code != 200:
+            return {"message": f"error {response.status_code}"}
+        response_str = response.json()
+        answer = response_str["candidates"]
+        for results in answer:
+            message = results.get("output")
+        return {
+            "status": "true",
+            "randydev":{
+                "message": message
+            }
+        }
+    except:
+        return {"status": "false", "message": "Error response."}
+
 async def get_data(username):
     base_msg = ""
     async with AsyncClient() as gpx: