Update app.py

app.py

@@ -3,7 +3,7 @@ from pydantic import BaseModel
 import requests
 import json
 import re
-import uuid
+import time
 
 app = FastAPI()
 
@@ -19,17 +19,15 @@ class GeminiAPI:
         self.session.cookies.update(self.cookies)
         
         self.headers = {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-            'Content-Type': 'application/x-www-form-urlencoded;charset=UTF-8',
-            'Origin': 'https://gemini.google.com',
-            'Referer': 'https://gemini.google.com/',
+            'User-Agent': 'Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Mobile Safari/537.36',
+            'Content-Type': 'application/json+protobuf',
+            'Accept': '*/*',
+            'Origin': 'https://aistudio.google.com',
+            'Referer': 'https://aistudio.google.com/',
+            'X-Goog-Api-Key': 'AIzaSyD0fP816MBRBSkj20Q04XbjsIcI0jGI0WOs',
+            'X-Client-Data': 'CJGVywE=',
         }
         
-        self.snlm0e = None
-        self.conversation_id = None
-        self.response_id = None
-        self.choice_id = None
-        
     def _parse_cookies(self, cookies_input):
         cookies = {}
         
@@ -52,87 +50,64 @@ class GeminiAPI:
         
         return cookies
     
-    def get_snlm0e_token(self):
-        try:
-            response = self.session.get(
-                'https://gemini.google.com/',
-                headers=self.headers,
-                timeout=10
-            )
-            
-            if 'Sign in' in response.text or 'Login' in response.text:
-                return False, "Cookie tidak valid atau expired"
-            
-            match = re.search(r'"SNlM0e":"(.*?)"', response.text)
-            if match:
-                self.snlm0e = match.group(1)
-                return True, "Token berhasil didapat"
-            
-            return False, "Token tidak ditemukan"
-                
-        except Exception as e:
-            return False, str(e)
-    
     def send_message(self, prompt, chat_id=None):
-        if not self.snlm0e:
-            success, msg = self.get_snlm0e_token()
-            if not success:
-                return None, msg
+        url = 'https://alkalimakersuite-pa.clients6.google.com/$rpc/google.internal.alkali.applications.makersuite.v1.MakerSuiteService/GenerateContent'
         
-        if chat_id:
-            self.conversation_id = chat_id
-        elif not self.conversation_id:
-            self.conversation_id = str(uuid.uuid4())
-        
-        url = 'https://gemini.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate'
-        
-        message_struct = [
-            [prompt],
+        payload = [
+            "models/gemini-3-flash-preview",
+            [[[[None, prompt]], "user"]],
+            None,
+            0,
+            "models/gemini-3-flash-preview",
+            [None, None, None, 65536, 1, 0.95, 64, None, None, None, None, None, None, None],
+            "!GBu1G0PNAAZ19fabc_VClWQf4_wu6go7ADQBEArZ1NrXSqgcKEzmXhn1r70At0ogWMU1G6bOhL3",
             None,
-            [self.conversation_id, self.response_id, self.choice_id] if self.conversation_id else None
+            [[None, None, [[]]],
+            None,
+            None,
+            None,
+            None,
+            None,
+            None,
+            None,
+            None,
+            [[None, None, "Asia/Makassar"]]
         ]
         
-        data = {
-            'f.req': json.dumps([None, json.dumps(message_struct)]),
-            'at': self.snlm0e,
-        }
-        
         try:
             response = self.session.post(
                 url,
                 headers=self.headers,
-                data=data,
+                json=payload,
                 timeout=30
             )
             
             if response.status_code != 200:
                 return None, f"HTTP Error: {response.status_code}"
             
-            response_text = response.text
-            lines = response_text.split('\n')
+            data = response.json()
+            
+            text_response = ""
+            
+            if isinstance(data, list) and len(data) > 0:
+                for item in data:
+                    if isinstance(item, list) and len(item) > 0:
+                        for subitem in item:
+                            if isinstance(subitem, list) and len(subitem) > 0:
+                                for content in subitem:
+                                    if isinstance(content, list) and len(content) > 0:
+                                        for text_item in content:
+                                            if isinstance(text_item, list) and len(text_item) > 0:
+                                                if text_item[0] is None and len(text_item) > 1:
+                                                    if isinstance(text_item[1], str):
+                                                        text_response += text_item[1]
             
-            for line in lines:
-                if line.startswith('['):
-                    try:
-                        data = json.loads(line)
-                        if data and len(data) > 0:
-                            response_data = json.loads(data[0][2])
-                            if response_data and len(response_data) > 4:
-                                text_response = response_data[4][0][1][0]
-                                
-                                if len(response_data) > 1:
-                                    self.conversation_id = response_data[1][0]
-                                    self.response_id = response_data[1][1]
-                                    self.choice_id = response_data[4][0][0]
-                                
-                                return {
-                                    'response': text_response,
-                                    'chat_id': self.conversation_id,
-                                    'response_id': self.response_id,
-                                    'choice_id': self.choice_id
-                                }, None
-                    except Exception as e:
-                        continue
+            if text_response:
+                return {
+                    'response': text_response.strip(),
+                    'chat_id': chat_id,
+                    'timestamp': int(time.time())
+                }, None
             
             return None, "Tidak bisa parse response"
             
@@ -157,11 +132,12 @@ async def chat(request: ChatRequest):
 async def root():
     return {
         "message": "Gemini API",
+        "version": "2.0",
         "endpoints": {
             "POST /chat": {
                 "cookies": "string (JSON array atau cookie string)",
                 "prompt": "string",
-                "chat_id": "string (optional, untuk continue chat)"
+                "chat_id": "string (optional)"
             }
         }
     }