1 files changed, 77 insertions, 0 deletions

diff --git a/you/__init__.py b/you/__init__.py
new file mode 100644
index 00000000..dc440b79
--- /dev/null
+++ b/you/__init__.py
@@ -0,0 +1,77 @@
+from tls_client   import Session
+from re         import findall
+from json       import loads, dumps
+from uuid       import uuid4
+
+
+class Completion:
+    def create(
+        prompt          : str,
+        page            : int  = 1,
+        count           : int  = 10,
+        safeSearch      : str  = "Moderate",
+        onShoppingpage  : bool = False,
+        mkt             : str  = "",
+        responseFilter  : str  = "WebPages,Translations,TimeZone,Computation,RelatedSearches",
+        domain          : str  = "youchat",
+        queryTraceId    : str  = None,
+        chat            : list = [],
+        includelinks    : bool = False,
+        detailed        : bool = False,
+        debug           : bool = False ) -> dict:
+        
+        client         = Session(client_identifier="chrome_108")
+        client.headers = {
+            "authority"         : "you.com",
+            "accept"            : "text/event-stream",
+            "accept-language"   : "en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3",
+            "cache-control"     : "no-cache",
+            "referer"           : "https://you.com/search?q=who+are+you&tbm=youchat",
+            "sec-ch-ua"         : '"Not_A Brand";v="99", "Google Chrome";v="109", "Chromium";v="109"',
+            "sec-ch-ua-mobile"  : "?0",
+            "sec-ch-ua-platform": '"Windows"',
+            "sec-fetch-dest"    : "empty",
+            "sec-fetch-mode"    : "cors",
+            "sec-fetch-site"    : "same-origin",
+            'cookie'            : f'safesearch_guest=Moderate; uuid_guest={str(uuid4())}',
+            "user-agent"        : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
+        }
+
+        response = client.get(f"https://you.com/api/streamingSearch", params = {
+                "q"              : prompt,
+                "page"           : page,
+                "count"          : count,
+                "safeSearch"     : safeSearch,
+                "onShoppingPage" : onShoppingpage,
+                "mkt"            : mkt,
+                "responseFilter" : responseFilter,
+                "domain"         : domain,
+                "queryTraceId"   : str(uuid4()) if queryTraceId is None else queryTraceId,
+                "chat"           : str(chat),  # {"question":"","answer":" '"}
+            }
+        )
+        
+        
+        if debug:
+            print('\n\n------------------\n\n')
+            print(response.text)
+            print('\n\n------------------\n\n')
+
+        youChatSerpResults      = findall(r'youChatSerpResults\ndata: (.*)\n\nevent', response.text)[0]
+        thirdPartySearchResults = findall(r"thirdPartySearchResults\ndata: (.*)\n\nevent", response.text)[0]
+        slots                   = findall(r"slots\ndata: (.*)\n\nevent", response.text)[0]
+        
+        text = response.text.split('}]}\n\nevent: youChatToken\ndata: {"youChatToken": "')[-1]
+        text = text.replace('"}\n\nevent: youChatToken\ndata: {"youChatToken": "', '')
+        text = text.replace('event: done\ndata: I\'m Mr. Meeseeks. Look at me.\n\n', '')
+        
+        extra = {
+            'youChatSerpResults'      : loads(youChatSerpResults),
+            'slots'                   : loads(slots)
+        }
+
+        return {
+            'response': text,
+            'links'   : loads(thirdPartySearchResults)['search']["third_party_search_results"] if includelinks else None,
+            'extra'   : extra if detailed else None,
+        }
\ No newline at end of file