1 files changed, 63 insertions, 3 deletions

diff --git a/g4f/Provider/You.py b/g4f/Provider/You.py
index 91a195cf..17e6269a 100644
--- a/g4f/Provider/You.py
+++ b/g4f/Provider/You.py
@@ -1,6 +1,8 @@
 from __future__ import annotations
 
 import json
+import base64
+import uuid
 
 from ..requests import StreamSession
 from ..typing import AsyncGenerator, Messages
@@ -11,7 +13,9 @@ class You(AsyncGeneratorProvider):
     url = "https://you.com"
     working = True
     supports_gpt_35_turbo = True
-
+    supports_gpt_4 = True
+    _session_used = 0
+    _session_token = None
 
     @classmethod
     async def create_async_generator(
@@ -27,14 +31,70 @@ class You(AsyncGeneratorProvider):
                 "Accept": "text/event-stream",
                 "Referer": f"{cls.url}/search?fromSearchBar=true&tbm=youchat",
             }
-            data = {"q": format_prompt(messages), "domain": "youchat", "chat": ""}
+            data = {
+                "q": format_prompt(messages),
+                "domain": "youchat",
+                "chat": "", "selectedChatMode": "gpt-4" if model == "gpt-4" else "default"
+            }
             async with session.get(
                 f"{cls.url}/api/streamingSearch",
                 params=data,
-                headers=headers
+                headers=headers,
+                cookies=cls.get_cookies(await cls.get_session_token(proxy, timeout)) if model == "gpt-4" else None
             ) as response:
                 response.raise_for_status()
                 start = b'data: {"youChatToken": '
                 async for line in response.iter_lines():
                     if line.startswith(start):
                         yield json.loads(line[len(start):-1])
+
+    @classmethod
+    async def get_session_token(cls, proxy: str, timeout: int):
+        if not cls._session_token or cls._session_used >= 5:
+            cls._session_token = await cls.create_session_token(proxy, timeout)
+        cls._session_used += 1
+        return cls._session_token
+
+    def get_cookies(access_token: str, session_jwt: str = "0"):
+        return {
+            'stytch_session_jwt': session_jwt,
+            'ydc_stytch_session': access_token,
+            'ydc_stytch_session_jwt': session_jwt
+        }
+
+    @classmethod
+    def get_jwt(cls):
+        return base64.standard_b64encode(json.dumps({
+            "event_id":f"event-id-{str(uuid.uuid4())}",
+            "app_session_id":f"app-session-id-{str(uuid.uuid4())}",
+            "persistent_id":f"persistent-id-{uuid.uuid4()}",
+            "client_sent_at":"","timezone":"",
+            "stytch_user_id":f"user-live-{uuid.uuid4()}",
+            "stytch_session_id":f"session-live-{uuid.uuid4()}",
+            "app":{"identifier":"you.com"},
+            "sdk":{"identifier":"Stytch.js Javascript SDK","version":"3.3.0"
+        }}).encode()).decode()
+
+    @classmethod
+    async def create_session_token(cls, proxy: str, timeout: int):
+        async with StreamSession(proxies={"https": proxy}, impersonate="chrome110", timeout=timeout) as session:
+            user_uuid = str(uuid.uuid4())
+            auth_uuid = "507a52ad-7e69-496b-aee0-1c9863c7c8"
+            auth_token = f"public-token-live-{auth_uuid}bb:public-token-live-{auth_uuid}19"
+            auth = base64.standard_b64encode(auth_token.encode()).decode()
+            async with session.post(
+                "https://web.stytch.com/sdk/v1/passwords",
+                headers={
+                    "Authorization": f"Basic {auth}",
+                    "X-SDK-Client": cls.get_jwt(),
+                    "X-SDK-Parent-Host": "https://you.com"
+                },
+                json={
+                    "email": f"{user_uuid}@gmail.com",
+                    "password": f"{user_uuid}#{user_uuid}",
+                    "session_duration_minutes": 129600
+                }
+            ) as response:
+                if not response.ok:
+                    raise RuntimeError(f"Response: {await response.text()}")
+                return (await response.json())["data"]["session_token"]
\ No newline at end of file