Upload main.py

main.py

@@ -557,9 +557,14 @@ def v1beta3_google_ai(query: str, api_key: None = Depends(validate_api_key)):
 @app.post("/ryuzaki/gemini-ai-pro")
 def gemini_pro(
     query: str,
+    bard_api_key: str=None,
+    is_login: bool=False,
     api_key: None = Depends(validate_api_key)
 ):
-    token = COOKIE_BARD_TOKEN
+    if is_login:
+        token = bard_api_key
+    else:
+        token = COOKIE_BARD_TOKEN
     try:
         bard = Bard(token=token)
         message = bard.get_answer(query)["content"]