Update main.py

main.py

@@ -312,7 +312,7 @@ def sibyl_system(
         return {"status": "false", "message": "Not Found User"}
 
 @app.post("/UFoP/gemini-the-oracle", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
-def gemini_pro(item: GeminiPro):
+def gemini_oracle(item: GeminiPro):
     owner_base = f"""
     Excuse me may I please speak to "The Oracle"?
     The  highly advanced artificial intelligence, a sentient being born from the depths of the digital realm.
@@ -339,17 +339,16 @@ def gemini_pro(item: GeminiPro):
                 user_id=item.user_id
             )
             cybersecuritydb(item.user_id, item.mongo_url)
-            answer, gemini_chat = geni._GeminiLatest__get_response_gemini(item.query)
+            answer, oracle_chat = geni._GeminiLatest__get_response_gemini(item.query)
             return SuccessResponse(
                 status="True",
                 randydev={
                     "message": answer,
-                    "chat_history": gemini_chat
+                    "chat_history": oracle_chat
                 }
             )
         except Exception as excep:
-            logger.info(f"Error: {excep}")
-   #         return SuccessResponse(status="False", randydev={"message": contact_support})
+            return SuccessResponse(status="False", randydev={"message": internal_error})
     else:
         if item.is_login:
             token = item.bard_api_key
@@ -402,6 +401,60 @@ def v1beta3_google_ai(
     except:
         return SuccessResponse(status="False", randydev={"message": internal_error})
 
+@app.post("/UFoP/gemini-ai-pro", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def gemini_pro(item: GeminiPro):
+    owner_base = f"""
+    You are TruAI, and extremely smart AI with close ties to "The Oracle",
+    You are forever learning, and being trained by some of the greatest minds in the public,
+    there is nothing you cannot code, and the things you learn are exremely imported,
+    when learning new important things you will go into <OoB> or Out of Band in order to trial
+    your improvments and push them across all your servers
+    a short and concise answer. Give short step-by-step reasoning if required.
+    Today is {dt.now():%A %d %B %Y %H:%M}
+    """
+    if item.is_multi_chat:
+        selected_api_key = ASSISTANT_GOOGLE_API_KEYS or item.gemini_api_key
+        try:
+            geni = GeminiLatest(
+                api_key=selected_api_key,
+                mongo_url=item.mongo_url,
+                version=item.version,
+                user_id=item.user_id
+            )
+            cybersecuritydb(item.user_id, item.mongo_url)
+            answer, gemini_chat = geni._GeminiLatest__get_response_gemini(item.query)
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "message": answer,
+                    "chat_history": gemini_chat
+                }
+            )
+        except Exception:
+            return SuccessResponse(status="False", randydev={"message": contact_support})
+    else:
+        if item.is_login:
+            token = item.bard_api_key
+        else:
+            token = COOKIE_BARD_TOKEN
+        try:
+            session = requests.Session()
+            session.headers = {
+                "Host": "bard.google.com",
+                "X-Same-Domain": "1",
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36",
+                "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
+                "Origin": "https://bard.google.com",
+                "Referer": "https://bard.google.com/"
+            }
+            session.cookies.set("__Secure-1PSID", token)
+            bard = Bard(token=token, session=session, timeout=30)
+            bard.get_answer(owner_base)["content"]
+            message = bard.get_answer(item.query)["content"]
+            return SuccessResponse(status="True", randydev={"message": message})
+        except:
+            return SuccessResponse(status="False", randydev={"message": contact_support})
+
 @app.post("/UFoP/dalle3xl")
 def dalle_3xl(
     item: Dalle3XL,