Upload 2 files

main.py

@@ -768,9 +768,10 @@ def gemini_pro(item: GeminiPro):
     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:
-            gt = GeminiLatest(api_key=ASSISTANT_GOOGLE_API_KEYS)
-            answer, gemini_chat = gt._get_response_gemini(item.query)
+            gt = GeminiLatest(api_key=selected_api_key)
+            answer, gemini_chat = gt._get_response_gemini(item.query, item.gemini_chat)
             return SuccessResponse(
                 status="True",
                 randydev={

models.py

@@ -72,6 +72,8 @@ class ChatgptCustom(BaseModel):
 class GeminiPro(BaseModel):
     query: str
     bard_api_key: Optional[str] = None
+    gemini_api_key: Optional[str] = None
+    gemini_chat: Optional[list] = None
     is_login: Optional[bool] = False
     is_multi_chat: Optional[bool] = False