Upload 3 files

main.py

@@ -69,9 +69,8 @@ from RyuzakiLib.mental import BadWordsList
 from RyuzakiLib.system import OpenReadSystem
 
 from bardapi import Bard
-
+import google.generativeai as genai
 from models import *
-
 from gpytranslate import SyncTranslator
 
 import logging
@@ -117,6 +116,8 @@ collection = db["users"]
 
 trans = SyncTranslator()
 
+genai.configure(api_key=ASSISTANT_GOOGLE_API_KEYS)
+
 app = FastAPI(docs_url=None, redoc_url="/")
 
 def get_all_api_keys():
@@ -177,7 +178,7 @@ def get_sibyl_system_banned(user_id):
         sibyl_user_id = user.get("sibyl_userid")
         return sibyl_name, reason, is_banned, date_joined, sibyl_user_id
     else:
-        return None, None, False, None, None
+        return None
 
 def get_all_banned():
     banned_users = []
@@ -764,38 +765,50 @@ def gemini_pro(
     item: GeminiPro,
     api_key: None = Depends(validate_api_key)
 ):
-    if item.is_login:
-        token = item.bard_api_key
-    else:
-        token = COOKIE_BARD_TOKEN
+    global item.multi_chat_messages
     owner_base = f"""
     Your name is Randy Dev. A kind and friendly AI assistant that answers in
     a short and concise answer. Give short step-by-step reasoning if required.
 
     Today is {dt.now():%A %d %B %Y %H:%M}
     """
-    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
+    if item.is_multi_chat:
+        try:
+            item.multi_chat_messages.append({"role": "user", "parts": [item.query]})
+            model = genai.GenerativeModel("gemini-pro")
+            response_genai = model.generate_content(item.multi_chat_messages)
+            item.multi_chat_messages.append({"role": "model", "parts": [response_genai.text]})
+            return SuccessResponse(
+                status="True",
+                randydev={
+                    "message": response_genai.text,
+                    "chat_history": item.multi_chat_messages
+                }
+            )
+        except:
+            return SuccessResponse(status="False", randydev={"message": "Error not responding"})
+    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/"
             }
-        )
-    except:
-        return {"status": "false", "message": "Error response."}
+            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": "Error not responding"})
 
 @app.post("/ryuzaki/v1beta2-google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta2_google_ai(
@@ -1340,7 +1353,7 @@ def custom_openapi():
         return app.openapi_schema
     openapi_schema = get_openapi(
         title="RyuzakiLib API",
-        version="2.2.7",
+        version="2.3.0",
         summary="Use It Only For Personal Project Else I Need To Delete The Api",
         description=description,
         routes=app.routes,

models.py

@@ -71,8 +71,10 @@ class ChatgptCustom(BaseModel):
 
 class GeminiPro(BaseModel):
     query: str
+    multi_chat_messages: Optional[list] = []
     bard_api_key: Optional[str] = None
-    is_login: bool=False
+    is_login: Optional[bool] = False
+    is_multi_chat: Optional[bool] = False
 
 class WaifuPics(BaseModel):
     types: str="sfw"

requirements.txt

@@ -26,4 +26,5 @@ uvloop
 stripe
 bardapi
 transformers
-g4f
+g4f
+google-generativeai