From 90e7797a4cdff8b6d9d4a1f3677ab8e765bc660c Mon Sep 17 00:00:00 2001
From: Raju Komati <komatiraju032@gmail.com>
Date: Thu, 27 Apr 2023 04:10:02 +0530
Subject: [PATCH 01/13] refactored the code and updated Poe class to accept
 keyword arguments  to specify driver type

---
 quora/README.md   |   2 +-
 quora/__init__.py | 455 ++++++++++++++++++++++++----------------------
 quora/api.py      |  83 +++------
 3 files changed, 259 insertions(+), 281 deletions(-)

diff --git a/quora/README.md b/quora/README.md
index 88f9ceb..09f7536 100644
--- a/quora/README.md
+++ b/quora/README.md
@@ -59,7 +59,7 @@ from quora import Poe
 
 # available models:  ['Sage', 'GPT-4', 'Claude+', 'Claude-instant', 'ChatGPT', 'Dragonfly', 'NeevaAI']
 
-poe = Poe(model='ChatGPT')
+poe = Poe(model='ChatGPT', driver='firefox', cookie_path='cookie.json')
 poe.chat('who won the football world cup most?')
 
 # new bot creation
diff --git a/quora/__init__.py b/quora/__init__.py
index 4c78313..b9ca6e5 100644
--- a/quora/__init__.py
+++ b/quora/__init__.py
@@ -6,13 +6,14 @@ from pathlib import Path
 from random import choice, choices, randint
 from re import search, findall
 from string import ascii_letters, digits
-from typing import Optional
+from typing import Optional, Union
 from urllib.parse import unquote
 
 import selenium.webdriver.support.expected_conditions as EC
+from fake_useragent import UserAgent
 from pypasser import reCaptchaV3
 from requests import Session
-from selenium import webdriver
+from selenium.webdriver import Firefox, Chrome, FirefoxOptions, ChromeOptions
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.wait import WebDriverWait
 from tls_client import Session as TLS
@@ -20,33 +21,51 @@ from tls_client import Session as TLS
 from quora.api import Client as PoeClient
 from quora.mail import Emailnator
 
+CHROME_DRIVER_URL = 'https://chromedriver.storage.googleapis.com'
+FIREFOX_DRIVER_URL = 'https://api.github.com/repos/mozilla/geckodriver/releases'
+
+SELENIUM_WEB_DRIVER_ERROR_MSG = b'''The error message you are receiving is due to the `geckodriver` executable not 
+being found in your system\'s PATH. To resolve this issue, you need to download the geckodriver and add its location 
+to your system\'s PATH.\n\nHere are the steps to resolve the issue:\n\n1. Download the geckodriver for your platform 
+(Windows, macOS, or Linux) from the following link: https://github.com/mozilla/geckodriver/releases\n\n2. Extract the 
+downloaded archive and locate the geckodriver executable.\n\n3. Add the geckodriver executable to your system\'s 
+PATH.\n\nFor macOS and Linux:\n\n- Open a terminal window.\n- Move the geckodriver executable to a directory that is 
+already in your PATH, or create a new directory and add it to your PATH:\n\n```bash\n# Example: Move geckodriver to 
+/usr/local/bin\nmv /path/to/your/geckodriver /usr/local/bin\n```\n\n- If you created a new directory, add it to your 
+PATH:\n\n```bash\n# Example: Add a new directory to PATH\nexport PATH=$PATH:/path/to/your/directory\n```\n\nFor 
+Windows:\n\n- Right-click on "My Computer" or "This PC" and select "Properties".\n- Click on "Advanced system 
+settings".\n- Click on the "Environment Variables" button.\n- In the "System variables" section, find the "Path" 
+variable, select it, and click "Edit".\n- Click "New" and add the path to the directory containing the geckodriver 
+executable.\n\nAfter adding the geckodriver to your PATH, restart your terminal or command prompt and try running 
+your script again. The error should be resolved.'''
+
 # from twocaptcha import TwoCaptcha
 # solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
 
 MODELS = {
-    "Sage": "capybara",
-    "GPT-4": "beaver",
-    "Claude+": "a2_2",
-    "Claude-instant": "a2",
-    "ChatGPT": "chinchilla",
-    "Dragonfly": "nutria",
-    "NeevaAI": "hutia",
+    'Sage': 'capybara',
+    'GPT-4': 'beaver',
+    'Claude+': 'a2_2',
+    'Claude-instant': 'a2',
+    'ChatGPT': 'chinchilla',
+    'Dragonfly': 'nutria',
+    'NeevaAI': 'hutia',
 }
 
 
 def extract_formkey(html):
-    script_regex = r"<script>if\(.+\)throw new Error;(.+)</script>"
+    script_regex = r'<script>if\(.+\)throw new Error;(.+)</script>'
     script_text = search(script_regex, html).group(1)
     key_regex = r'var .="([0-9a-f]+)",'
     key_text = search(key_regex, script_text).group(1)
-    cipher_regex = r".\[(\d+)\]=.\[(\d+)\]"
+    cipher_regex = r'.\[(\d+)\]=.\[(\d+)\]'
     cipher_pairs = findall(cipher_regex, script_text)
 
-    formkey_list = [""] * len(cipher_pairs)
+    formkey_list = [''] * len(cipher_pairs)
     for pair in cipher_pairs:
         formkey_index, key_index = map(int, pair)
         formkey_list[formkey_index] = key_text[key_index]
-    formkey = "".join(formkey_list)
+    formkey = ''.join(formkey_list)
 
     return formkey
 
@@ -55,35 +74,35 @@ class PoeResponse:
     class Completion:
         class Choices:
             def __init__(self, choice: dict) -> None:
-                self.text = choice["text"]
+                self.text = choice['text']
                 self.content = self.text.encode()
-                self.index = choice["index"]
-                self.logprobs = choice["logprobs"]
-                self.finish_reason = choice["finish_reason"]
+                self.index = choice['index']
+                self.logprobs = choice['logprobs']
+                self.finish_reason = choice['finish_reason']
 
             def __repr__(self) -> str:
-                return f"""<__main__.APIResponse.Completion.Choices(\n    text           = {self.text.encode()},\n    index          = {self.index},\n    logprobs       = {self.logprobs},\n    finish_reason  = {self.finish_reason})object at 0x1337>"""
+                return f'''<__main__.APIResponse.Completion.Choices(\n    text           = {self.text.encode()},\n    index          = {self.index},\n    logprobs       = {self.logprobs},\n    finish_reason  = {self.finish_reason})object at 0x1337>'''
 
         def __init__(self, choices: dict) -> None:
             self.choices = [self.Choices(choice) for choice in choices]
 
     class Usage:
         def __init__(self, usage_dict: dict) -> None:
-            self.prompt_tokens = usage_dict["prompt_tokens"]
-            self.completion_tokens = usage_dict["completion_tokens"]
-            self.total_tokens = usage_dict["total_tokens"]
+            self.prompt_tokens = usage_dict['prompt_tokens']
+            self.completion_tokens = usage_dict['completion_tokens']
+            self.total_tokens = usage_dict['total_tokens']
 
         def __repr__(self):
-            return f"""<__main__.APIResponse.Usage(\n    prompt_tokens      = {self.prompt_tokens},\n    completion_tokens  = {self.completion_tokens},\n    total_tokens       = {self.total_tokens})object at 0x1337>"""
+            return f'''<__main__.APIResponse.Usage(\n    prompt_tokens      = {self.prompt_tokens},\n    completion_tokens  = {self.completion_tokens},\n    total_tokens       = {self.total_tokens})object at 0x1337>'''
 
     def __init__(self, response_dict: dict) -> None:
         self.response_dict = response_dict
-        self.id = response_dict["id"]
-        self.object = response_dict["object"]
-        self.created = response_dict["created"]
-        self.model = response_dict["model"]
-        self.completion = self.Completion(response_dict["choices"])
-        self.usage = self.Usage(response_dict["usage"])
+        self.id = response_dict['id']
+        self.object = response_dict['object']
+        self.created = response_dict['created']
+        self.model = response_dict['model']
+        self.completion = self.Completion(response_dict['choices'])
+        self.usage = self.Usage(response_dict['usage'])
 
     def json(self) -> dict:
         return self.response_dict
@@ -91,139 +110,135 @@ class PoeResponse:
 
 class ModelResponse:
     def __init__(self, json_response: dict) -> None:
-        self.id = json_response["data"]["poeBotCreate"]["bot"]["id"]
-        self.name = json_response["data"]["poeBotCreate"]["bot"]["displayName"]
-        self.limit = json_response["data"]["poeBotCreate"]["bot"]["messageLimit"][
-            "dailyLimit"
-        ]
-        self.deleted = json_response["data"]["poeBotCreate"]["bot"]["deletionState"]
+        self.id = json_response['data']['poeBotCreate']['bot']['id']
+        self.name = json_response['data']['poeBotCreate']['bot']['displayName']
+        self.limit = json_response['data']['poeBotCreate']['bot']['messageLimit']['dailyLimit']
+        self.deleted = json_response['data']['poeBotCreate']['bot']['deletionState']
 
 
 class Model:
+    @staticmethod
     def create(
         token: str,
-        model: str = "gpt-3.5-turbo",  # claude-instant
-        system_prompt: str = "You are ChatGPT a large language model developed by Openai. Answer as consisely as possible",
-        description: str = "gpt-3.5 language model from openai, skidded by poe.com",
+        model: str = 'gpt-3.5-turbo',  # claude-instant
+        system_prompt: str = 'You are ChatGPT a large language model developed by Openai. Answer as consisely as possible',
+        description: str = 'gpt-3.5 language model from openai, skidded by poe.com',
         handle: str = None,
     ) -> ModelResponse:
         models = {
-            "gpt-3.5-turbo": "chinchilla",
-            "claude-instant-v1.0": "a2",
-            "gpt-4": "beaver",
+            'gpt-3.5-turbo': 'chinchilla',
+            'claude-instant-v1.0': 'a2',
+            'gpt-4': 'beaver',
         }
 
         if not handle:
-            handle = f"gptx{randint(1111111, 9999999)}"
+            handle = f'gptx{randint(1111111, 9999999)}'
 
         client = Session()
-        client.cookies["p-b"] = token
+        client.cookies['p-b'] = token
 
-        formkey = extract_formkey(client.get("https://poe.com").text)
-        settings = client.get("https://poe.com/api/settings").json()
+        formkey = extract_formkey(client.get('https://poe.com').text)
+        settings = client.get('https://poe.com/api/settings').json()
 
         client.headers = {
-            "host": "poe.com",
-            "origin": "https://poe.com",
-            "referer": "https://poe.com/",
-            "poe-formkey": formkey,
-            "poe-tchannel": settings["tchannelData"]["channel"],
-            "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36",
-            "connection": "keep-alive",
-            "sec-ch-ua": '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": '"macOS"',
-            "content-type": "application/json",
-            "sec-fetch-site": "same-origin",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-dest": "empty",
-            "accept": "*/*",
-            "accept-encoding": "gzip, deflate, br",
-            "accept-language": "en-GB,en-US;q=0.9,en;q=0.8",
+            'host': 'poe.com',
+            'origin': 'https://poe.com',
+            'referer': 'https://poe.com/',
+            'poe-formkey': formkey,
+            'poe-tchannel': settings['tchannelData']['channel'],
+            'user-agent': UserAgent().random,
+            'connection': 'keep-alive',
+            'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+            'sec-ch-ua-mobile': '?0',
+            'sec-ch-ua-platform': '"macOS"',
+            'content-type': 'application/json',
+            'sec-fetch-site': 'same-origin',
+            'sec-fetch-mode': 'cors',
+            'sec-fetch-dest': 'empty',
+            'accept': '*/*',
+            'accept-encoding': 'gzip, deflate, br',
+            'accept-language': 'en-GB,en-US;q=0.9,en;q=0.8',
         }
 
         payload = dumps(
-            separators=(",", ":"),
+            separators=(',', ':'),
             obj={
-                "queryName": "CreateBotMain_poeBotCreate_Mutation",
-                "variables": {
-                    "model": models[model],
-                    "handle": handle,
-                    "prompt": system_prompt,
-                    "isPromptPublic": True,
-                    "introduction": "",
-                    "description": description,
-                    "profilePictureUrl": "https://qph.fs.quoracdn.net/main-qimg-24e0b480dcd946e1cc6728802c5128b6",
-                    "apiUrl": None,
-                    "apiKey": "".join(choices(ascii_letters + digits, k=32)),
-                    "isApiBot": False,
-                    "hasLinkification": False,
-                    "hasMarkdownRendering": False,
-                    "hasSuggestedReplies": False,
-                    "isPrivateBot": False,
+                'queryName': 'CreateBotMain_poeBotCreate_Mutation',
+                'variables': {
+                    'model': models[model],
+                    'handle': handle,
+                    'prompt': system_prompt,
+                    'isPromptPublic': True,
+                    'introduction': '',
+                    'description': description,
+                    'profilePictureUrl': 'https://qph.fs.quoracdn.net/main-qimg-24e0b480dcd946e1cc6728802c5128b6',
+                    'apiUrl': None,
+                    'apiKey': ''.join(choices(ascii_letters + digits, k=32)),
+                    'isApiBot': False,
+                    'hasLinkification': False,
+                    'hasMarkdownRendering': False,
+                    'hasSuggestedReplies': False,
+                    'isPrivateBot': False,
                 },
-                "query": "mutation CreateBotMain_poeBotCreate_Mutation(\n  $model: String!\n  $handle: String!\n  $prompt: String!\n  $isPromptPublic: Boolean!\n  $introduction: String!\n  $description: String!\n  $profilePictureUrl: String\n  $apiUrl: String\n  $apiKey: String\n  $isApiBot: Boolean\n  $hasLinkification: Boolean\n  $hasMarkdownRendering: Boolean\n  $hasSuggestedReplies: Boolean\n  $isPrivateBot: Boolean\n) {\n  poeBotCreate(model: $model, handle: $handle, promptPlaintext: $prompt, isPromptPublic: $isPromptPublic, introduction: $introduction, description: $description, profilePicture: $profilePictureUrl, apiUrl: $apiUrl, apiKey: $apiKey, isApiBot: $isApiBot, hasLinkification: $hasLinkification, hasMarkdownRendering: $hasMarkdownRendering, hasSuggestedReplies: $hasSuggestedReplies, isPrivateBot: $isPrivateBot) {\n    status\n    bot {\n      id\n      ...BotHeader_bot\n    }\n  }\n}\n\nfragment BotHeader_bot on Bot {\n  displayName\n  messageLimit {\n    dailyLimit\n  }\n  ...BotImage_bot\n  ...BotLink_bot\n  ...IdAnnotation_node\n  ...botHelpers_useViewerCanAccessPrivateBot\n  ...botHelpers_useDeletion_bot\n}\n\nfragment BotImage_bot on Bot {\n  displayName\n  ...botHelpers_useDeletion_bot\n  ...BotImage_useProfileImage_bot\n}\n\nfragment BotImage_useProfileImage_bot on Bot {\n  image {\n    __typename\n    ... on LocalBotImage {\n      localName\n    }\n    ... on UrlBotImage {\n      url\n    }\n  }\n  ...botHelpers_useDeletion_bot\n}\n\nfragment BotLink_bot on Bot {\n  displayName\n}\n\nfragment IdAnnotation_node on Node {\n  __isNode: __typename\n  id\n}\n\nfragment botHelpers_useDeletion_bot on Bot {\n  deletionState\n}\n\nfragment botHelpers_useViewerCanAccessPrivateBot on Bot {\n  isPrivateBot\n  viewerIsCreator\n}\n",
+                'query': 'mutation CreateBotMain_poeBotCreate_Mutation(\n  $model: String!\n  $handle: String!\n  $prompt: String!\n  $isPromptPublic: Boolean!\n  $introduction: String!\n  $description: String!\n  $profilePictureUrl: String\n  $apiUrl: String\n  $apiKey: String\n  $isApiBot: Boolean\n  $hasLinkification: Boolean\n  $hasMarkdownRendering: Boolean\n  $hasSuggestedReplies: Boolean\n  $isPrivateBot: Boolean\n) {\n  poeBotCreate(model: $model, handle: $handle, promptPlaintext: $prompt, isPromptPublic: $isPromptPublic, introduction: $introduction, description: $description, profilePicture: $profilePictureUrl, apiUrl: $apiUrl, apiKey: $apiKey, isApiBot: $isApiBot, hasLinkification: $hasLinkification, hasMarkdownRendering: $hasMarkdownRendering, hasSuggestedReplies: $hasSuggestedReplies, isPrivateBot: $isPrivateBot) {\n    status\n    bot {\n      id\n      ...BotHeader_bot\n    }\n  }\n}\n\nfragment BotHeader_bot on Bot {\n  displayName\n  messageLimit {\n    dailyLimit\n  }\n  ...BotImage_bot\n  ...BotLink_bot\n  ...IdAnnotation_node\n  ...botHelpers_useViewerCanAccessPrivateBot\n  ...botHelpers_useDeletion_bot\n}\n\nfragment BotImage_bot on Bot {\n  displayName\n  ...botHelpers_useDeletion_bot\n  ...BotImage_useProfileImage_bot\n}\n\nfragment BotImage_useProfileImage_bot on Bot {\n  image {\n    __typename\n    ... on LocalBotImage {\n      localName\n    }\n    ... on UrlBotImage {\n      url\n    }\n  }\n  ...botHelpers_useDeletion_bot\n}\n\nfragment BotLink_bot on Bot {\n  displayName\n}\n\nfragment IdAnnotation_node on Node {\n  __isNode: __typename\n  id\n}\n\nfragment botHelpers_useDeletion_bot on Bot {\n  deletionState\n}\n\nfragment botHelpers_useViewerCanAccessPrivateBot on Bot {\n  isPrivateBot\n  viewerIsCreator\n}\n',
             },
         )
 
-        base_string = payload + client.headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
-        client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
+        base_string = payload + client.headers['poe-formkey'] + 'WpuLMiXEKKE98j56k'
+        client.headers['poe-tag-id'] = md5(base_string.encode()).hexdigest()
 
-        response = client.post("https://poe.com/api/gql_POST", data=payload)
+        response = client.post('https://poe.com/api/gql_POST', data=payload)
 
-        if "success" not in response.text:
+        if 'success' not in response.text:
             raise Exception(
-                """
+                '''
                 Bot creation Failed
                 !! Important !!
                 Bot creation was not enabled on this account
                 please use: quora.Account.create with enable_bot_creation set to True
-            """
+            '''
             )
 
         return ModelResponse(response.json())
 
 
 class Account:
+    @staticmethod
     def create(
         proxy: Optional[str] = None,
         logging: bool = False,
         enable_bot_creation: bool = False,
     ):
-        client = TLS(client_identifier="chrome110")
-        client.proxies = (
-            {"http": f"http://{proxy}", "https": f"http://{proxy}"} if proxy else None
-        )
+        client = TLS(client_identifier='chrome110')
+        client.proxies = {'http': f'http://{proxy}', 'https': f'http://{proxy}'} if proxy else None
 
         mail_client = Emailnator()
         mail_address = mail_client.get_mail()
 
         if logging:
-            print("email", mail_address)
+            print('email', mail_address)
 
         client.headers = {
-            "authority": "poe.com",
-            "accept": "*/*",
-            "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",
-            "content-type": "application/json",
-            "origin": "https://poe.com",
-            "poe-tag-id": "null",
-            "referer": "https://poe.com/login",
-            "sec-ch-ua": '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
-            "sec-ch-ua-mobile": "?0",
-            "sec-ch-ua-platform": '"macOS"',
-            "sec-fetch-dest": "empty",
-            "sec-fetch-mode": "cors",
-            "sec-fetch-site": "same-origin",
-            "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36",
-            "poe-formkey": extract_formkey(client.get("https://poe.com/login").text),
-            "poe-tchannel": client.get("https://poe.com/api/settings").json()[
-                "tchannelData"
-            ]["channel"],
+            'authority': 'poe.com',
+            'accept': '*/*',
+            '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',
+            'content-type': 'application/json',
+            'origin': 'https://poe.com',
+            'poe-tag-id': 'null',
+            'referer': 'https://poe.com/login',
+            'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+            'sec-ch-ua-mobile': '?0',
+            'sec-ch-ua-platform': '"macOS"',
+            'sec-fetch-dest': 'empty',
+            'sec-fetch-mode': 'cors',
+            'sec-fetch-site': 'same-origin',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+            'poe-formkey': extract_formkey(client.get('https://poe.com/login').text),
+            'poe-tchannel': client.get('https://poe.com/api/settings').json()['tchannelData']['channel'],
         }
 
         token = reCaptchaV3(
-            "https://www.recaptcha.net/recaptcha/enterprise/anchor?ar=1&k=6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG&co=aHR0cHM6Ly9wb2UuY29tOjQ0Mw..&hl=en&v=4PnKmGB9wRHh1i04o7YUICeI&size=invisible&cb=bi6ivxoskyal"
+            'https://www.recaptcha.net/recaptcha/enterprise/anchor?ar=1&k=6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG&co=aHR0cHM6Ly9wb2UuY29tOjQ0Mw..&hl=en&v=4PnKmGB9wRHh1i04o7YUICeI&size=invisible&cb=bi6ivxoskyal'
         )
         # token = solver.recaptcha(sitekey='6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG',
         #     url        = 'https://poe.com/login?redirect_url=%2F',
@@ -233,77 +248,74 @@ class Account:
         #     action     = 'login',)['code']
 
         payload = dumps(
-            separators=(",", ":"),
+            separators=(',', ':'),
             obj={
-                "queryName": "MainSignupLoginSection_sendVerificationCodeMutation_Mutation",
-                "variables": {
-                    "emailAddress": mail_address,
-                    "phoneNumber": None,
-                    "recaptchaToken": token,
+                'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
+                'variables': {
+                    'emailAddress': mail_address,
+                    'phoneNumber': None,
+                    'recaptchaToken': token,
                 },
-                "query": "mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n",
+                'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n',
             },
         )
 
-        base_string = payload + client.headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
-        client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
+        base_string = payload + client.headers['poe-formkey'] + 'WpuLMiXEKKE98j56k'
+        client.headers['poe-tag-id'] = md5(base_string.encode()).hexdigest()
 
         print(dumps(client.headers, indent=4))
 
-        response = client.post("https://poe.com/api/gql_POST", data=payload)
+        response = client.post('https://poe.com/api/gql_POST', data=payload)
 
-        if "automated_request_detected" in response.text:
-            print("please try using a proxy / wait for fix")
+        if 'automated_request_detected' in response.text:
+            print('please try using a proxy / wait for fix')
 
-        if "Bad Request" in response.text:
+        if 'Bad Request' in response.text:
             if logging:
-                print("bad request, retrying...", response.json())
+                print('bad request, retrying...', response.json())
             quit()
 
         if logging:
-            print("send_code", response.json())
+            print('send_code', response.json())
 
         mail_content = mail_client.get_message()
         mail_token = findall(r';">(\d{6,7})</div>', mail_content)[0]
 
         if logging:
-            print("code", mail_token)
+            print('code', mail_token)
 
         payload = dumps(
-            separators=(",", ":"),
+            separators=(',', ':'),
             obj={
-                "queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
-                "variables": {
-                    "verificationCode": str(mail_token),
-                    "emailAddress": mail_address,
-                    "phoneNumber": None,
+                'queryName': 'SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation',
+                'variables': {
+                    'verificationCode': str(mail_token),
+                    'emailAddress': mail_address,
+                    'phoneNumber': None,
                 },
-                "query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n",
+                'query': 'mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n',
             },
         )
 
-        base_string = payload + client.headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
-        client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
+        base_string = payload + client.headers['poe-formkey'] + 'WpuLMiXEKKE98j56k'
+        client.headers['poe-tag-id'] = md5(base_string.encode()).hexdigest()
 
-        response = client.post("https://poe.com/api/gql_POST", data=payload)
+        response = client.post('https://poe.com/api/gql_POST', data=payload)
         if logging:
-            print("verify_code", response.json())
+            print('verify_code', response.json())
 
     def get(self):
-        cookies = (
-            open(Path(__file__).resolve().parent / "cookies.txt", "r")
-            .read()
-            .splitlines()
-        )
+        cookies = open(Path(__file__).resolve().parent / 'cookies.txt', 'r').read().splitlines()
         return choice(cookies)
 
 
 class StreamingCompletion:
+    @staticmethod
     def create(
-        model: str = "gpt-4",
+        model: str = 'gpt-4',
         custom_model: bool = None,
-        prompt: str = "hello world",
-        token: str = "",
+        prompt: str = 'hello world',
+        token: str = '',
     ):
         _model = MODELS[model] if not custom_model else custom_model
 
@@ -312,22 +324,22 @@ class StreamingCompletion:
         for chunk in client.send_message(_model, prompt):
             yield PoeResponse(
                 {
-                    "id": chunk["messageId"],
-                    "object": "text_completion",
-                    "created": chunk["creationTime"],
-                    "model": _model,
-                    "choices": [
+                    'id': chunk['messageId'],
+                    'object': 'text_completion',
+                    'created': chunk['creationTime'],
+                    'model': _model,
+                    'choices': [
                         {
-                            "text": chunk["text_new"],
-                            "index": 0,
-                            "logprobs": None,
-                            "finish_reason": "stop",
+                            'text': chunk['text_new'],
+                            'index': 0,
+                            'logprobs': None,
+                            'finish_reason': 'stop',
                         }
                     ],
-                    "usage": {
-                        "prompt_tokens": len(prompt),
-                        "completion_tokens": len(chunk["text_new"]),
-                        "total_tokens": len(prompt) + len(chunk["text_new"]),
+                    'usage': {
+                        'prompt_tokens': len(prompt),
+                        'completion_tokens': len(chunk['text_new']),
+                        'total_tokens': len(prompt) + len(chunk['text_new']),
                     },
                 }
             )
@@ -335,17 +347,17 @@ class StreamingCompletion:
 
 class Completion:
     def create(
-        model: str = "gpt-4",
+        model: str = 'gpt-4',
         custom_model: str = None,
-        prompt: str = "hello world",
-        token: str = "",
+        prompt: str = 'hello world',
+        token: str = '',
     ):
         models = {
-            "sage": "capybara",
-            "gpt-4": "beaver",
-            "claude-v1.2": "a2_2",
-            "claude-instant-v1.0": "a2",
-            "gpt-3.5-turbo": "chinchilla",
+            'sage': 'capybara',
+            'gpt-4': 'beaver',
+            'claude-v1.2': 'a2_2',
+            'claude-instant-v1.0': 'a2',
+            'gpt-3.5-turbo': 'chinchilla',
         }
 
         _model = models[model] if not custom_model else custom_model
@@ -357,73 +369,68 @@ class Completion:
 
         return PoeResponse(
             {
-                "id": chunk["messageId"],
-                "object": "text_completion",
-                "created": chunk["creationTime"],
-                "model": _model,
-                "choices": [
+                'id': chunk['messageId'],
+                'object': 'text_completion',
+                'created': chunk['creationTime'],
+                'model': _model,
+                'choices': [
                     {
-                        "text": chunk["text"],
-                        "index": 0,
-                        "logprobs": None,
-                        "finish_reason": "stop",
+                        'text': chunk['text'],
+                        'index': 0,
+                        'logprobs': None,
+                        'finish_reason': 'stop',
                     }
                 ],
-                "usage": {
-                    "prompt_tokens": len(prompt),
-                    "completion_tokens": len(chunk["text"]),
-                    "total_tokens": len(prompt) + len(chunk["text"]),
+                'usage': {
+                    'prompt_tokens': len(prompt),
+                    'completion_tokens': len(chunk['text']),
+                    'total_tokens': len(prompt) + len(chunk['text']),
                 },
             }
         )
 
 
 class Poe:
-    def __init__(self, model: str = "ChatGPT"):
+    def __init__(
+        self,
+        model: str = 'ChatGPT',
+        driver: str = 'firefox',
+        download_driver: bool = False,
+        driver_path: Optional[str] = None,
+        cookie_path: str = './quora/cookie.json',
+    ):
         # validating the model
         if model and model not in MODELS:
-            raise RuntimeError(
-                "Sorry, the model you provided does not exist. Please check and try again."
-            )
+            raise RuntimeError('Sorry, the model you provided does not exist. Please check and try again.')
         self.model = MODELS[model]
-        self.cookie = self.__load_cookie()
+        self.cookie_path = cookie_path
+        self.cookie = self.__load_cookie(driver, download_driver, driver_path=driver_path)
         self.client = PoeClient(self.cookie)
 
-    def __load_cookie(self) -> str:
-        if (cookie_file := Path("./quora/cookie.json")).exists():
+    def __load_cookie(self, driver: str, download_driver: bool, driver_path: Optional[str] = None) -> str:
+        if (cookie_file := Path(self.cookie_path)).exists():
             with cookie_file.open() as fp:
                 cookie = json.load(fp)
-                if datetime.fromtimestamp(cookie["expiry"]) < datetime.now():
-                    cookie = self.__register_and_get_cookie()
+                if datetime.fromtimestamp(cookie['expiry']) < datetime.now():
+                    cookie = self.__register_and_get_cookie(driver, driver_path=driver_path)
                 else:
-                    print("Loading the cookie from file")
+                    print('Loading the cookie from file')
         else:
-            cookie = self.__register_and_get_cookie()
+            cookie = self.__register_and_get_cookie(driver, driver_path=driver_path)
 
-        return unquote(cookie["value"])
+        return unquote(cookie['value'])
 
-    @classmethod
-    def __register_and_get_cookie(cls) -> dict:
+    def __register_and_get_cookie(self, driver: str, driver_path: Optional[str] = None) -> dict:
         mail_client = Emailnator()
         mail_address = mail_client.get_mail()
 
-        print(mail_address)
-        options = webdriver.FirefoxOptions()
-        # options.add_argument("-headless")
-        try:
-            driver = webdriver.Firefox(options=options)
-        
-        except Exception:
-            raise Exception(b'The error message you are receiving is due to the `geckodriver` executable not being found in your system\'s PATH. To resolve this issue, you need to download the geckodriver and add its location to your system\'s PATH.\n\nHere are the steps to resolve the issue:\n\n1. Download the geckodriver for your platform (Windows, macOS, or Linux) from the following link: https://github.com/mozilla/geckodriver/releases\n\n2. Extract the downloaded archive and locate the geckodriver executable.\n\n3. Add the geckodriver executable to your system\'s PATH.\n\nFor macOS and Linux:\n\n- Open a terminal window.\n- Move the geckodriver executable to a directory that is already in your PATH, or create a new directory and add it to your PATH:\n\n```bash\n# Example: Move geckodriver to /usr/local/bin\nmv /path/to/your/geckodriver /usr/local/bin\n```\n\n- If you created a new directory, add it to your PATH:\n\n```bash\n# Example: Add a new directory to PATH\nexport PATH=$PATH:/path/to/your/directory\n```\n\nFor Windows:\n\n- Right-click on "My Computer" or "This PC" and select "Properties".\n- Click on "Advanced system settings".\n- Click on the "Environment Variables" button.\n- In the "System variables" section, find the "Path" variable, select it, and click "Edit".\n- Click "New" and add the path to the directory containing the geckodriver executable.\n\nAfter adding the geckodriver to your PATH, restart your terminal or command prompt and try running your script again. The error should be resolved.')
-        
+        driver = self.__resolve_driver(driver, driver_path=driver_path)
         driver.get("https://www.poe.com")
 
         # clicking use email button
         driver.find_element(By.XPATH, '//button[contains(text(), "Use email")]').click()
 
-        email = WebDriverWait(driver, 30).until(
-            EC.presence_of_element_located((By.XPATH, '//input[@type="email"]'))
-        )
+        email = WebDriverWait(driver, 30).until(EC.presence_of_element_located((By.XPATH, '//input[@type="email"]')))
         email.send_keys(mail_address)
         driver.find_element(By.XPATH, '//button[text()="Go"]').click()
 
@@ -434,46 +441,50 @@ class Poe:
             EC.presence_of_element_located((By.XPATH, '//input[@placeholder="Code"]'))
         )
         verification_code.send_keys(code)
-        verify_button = EC.presence_of_element_located(
-            (By.XPATH, '//button[text()="Verify"]')
-        )
-        login_button = EC.presence_of_element_located(
-            (By.XPATH, '//button[text()="Log In"]')
-        )
+        verify_button = EC.presence_of_element_located((By.XPATH, '//button[text()="Verify"]'))
+        login_button = EC.presence_of_element_located((By.XPATH, '//button[text()="Log In"]'))
 
         WebDriverWait(driver, 30).until(EC.any_of(verify_button, login_button)).click()
 
-        cookie = driver.get_cookie("p-b")
+        cookie = driver.get_cookie('p-b')
 
-        with open("./quora/cookie.json", "w") as fw:
+        with open(self.cookie_path, 'w') as fw:
             json.dump(cookie, fw)
 
         driver.close()
         return cookie
 
+    @classmethod
+    def __resolve_driver(cls, driver: str, driver_path: Optional[str] = None) -> Union[Firefox, Chrome]:
+        options = FirefoxOptions() if driver == 'firefox' else ChromeOptions()
+        options.add_argument('-headless')
+
+        if driver_path:
+            options.binary_location = driver_path
+        try:
+            return Firefox(options=options) if driver == 'firefox' else Chrome(options=options)
+        except Exception:
+            raise Exception(SELENIUM_WEB_DRIVER_ERROR_MSG)
+
     def chat(self, message: str, model: Optional[str] = None) -> str:
         if model and model not in MODELS:
-            raise RuntimeError(
-                "Sorry, the model you provided does not exist. Please check and try again."
-            )
+            raise RuntimeError('Sorry, the model you provided does not exist. Please check and try again.')
         model = MODELS[model] if model else self.model
         response = None
         for chunk in self.client.send_message(model, message):
-            response = chunk["text"]
+            response = chunk['text']
         return response
 
     def create_bot(
         self,
         name: str,
         /,
-        prompt: str = "",
-        base_model: str = "ChatGPT",
-        description: str = "",
+        prompt: str = '',
+        base_model: str = 'ChatGPT',
+        description: str = '',
     ) -> None:
         if base_model not in MODELS:
-            raise RuntimeError(
-                "Sorry, the base_model you provided does not exist. Please check and try again."
-            )
+            raise RuntimeError('Sorry, the base_model you provided does not exist. Please check and try again.')
 
         response = self.client.create_bot(
             handle=name,
diff --git a/quora/api.py b/quora/api.py
index b28c124..42814f2 100644
--- a/quora/api.py
+++ b/quora/api.py
@@ -18,23 +18,21 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-import requests
-import re
-import json
-import random
-import logging
-import time
-import queue
-import threading
-import traceback
 import hashlib
-import string
+import json
+import logging
+import queue
 import random
-import requests.adapters
-import websocket
+import re
+import threading
+import time
+import traceback
 from pathlib import Path
 from urllib.parse import urlparse
 
+import requests
+import requests.adapters
+import websocket
 
 parent_path = Path(__file__).resolve().parent
 queries_path = parent_path / "graphql"
@@ -66,7 +64,7 @@ def request_with_retries(method, *args, **kwargs):
         if r.status_code == 200:
             return r
         logger.warn(
-            f"Server returned a status code of {r.status_code} while downloading {url}. Retrying ({i+1}/{attempts})..."
+            f"Server returned a status code of {r.status_code} while downloading {url}. Retrying ({i + 1}/{attempts})..."
         )
 
     raise RuntimeError(f"Failed to download {url} too many times.")
@@ -81,9 +79,7 @@ class Client:
     def __init__(self, token, proxy=None):
         self.proxy = proxy
         self.session = requests.Session()
-        self.adapter = requests.adapters.HTTPAdapter(
-            pool_connections=100, pool_maxsize=100
-        )
+        self.adapter = requests.adapters.HTTPAdapter(pool_connections=100, pool_maxsize=100)
         self.session.mount("http://", self.adapter)
         self.session.mount("https://", self.adapter)
 
@@ -139,9 +135,7 @@ class Client:
         logger.info("Downloading next_data...")
 
         r = request_with_retries(self.session.get, self.home_url)
-        json_regex = (
-            r'<script id="__NEXT_DATA__" type="application\/json">(.+?)</script>'
-        )
+        json_regex = r'<script id="__NEXT_DATA__" type="application\/json">(.+?)</script>'
         json_text = re.search(json_regex, r.text).group(1)
         next_data = json.loads(json_text)
 
@@ -213,19 +207,14 @@ class Client:
         if channel is None:
             channel = self.channel
         query = f'?min_seq={channel["minSeq"]}&channel={channel["channel"]}&hash={channel["channelHash"]}'
-        return (
-            f'wss://{self.ws_domain}.tch.{channel["baseHost"]}/up/{channel["boxName"]}/updates'
-            + query
-        )
+        return f'wss://{self.ws_domain}.tch.{channel["baseHost"]}/up/{channel["boxName"]}/updates' + query
 
     def send_query(self, query_name, variables):
         for i in range(20):
             json_data = generate_payload(query_name, variables)
             payload = json.dumps(json_data, separators=(",", ":"))
 
-            base_string = (
-                payload + self.gql_headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
-            )
+            base_string = payload + self.gql_headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
 
             headers = {
                 "content-type": "application/json",
@@ -233,15 +222,11 @@ class Client:
             }
             headers = {**self.gql_headers, **headers}
 
-            r = request_with_retries(
-                self.session.post, self.gql_url, data=payload, headers=headers
-            )
+            r = request_with_retries(self.session.post, self.gql_url, data=payload, headers=headers)
 
             data = r.json()
             if data["data"] == None:
-                logger.warn(
-                    f'{query_name} returned an error: {data["errors"][0]["message"]} | Retrying ({i+1}/20)'
-                )
+                logger.warn(f'{query_name} returned an error: {data["errors"][0]["message"]} | Retrying ({i + 1}/20)')
                 time.sleep(2)
                 continue
 
@@ -304,9 +289,7 @@ class Client:
 
     def on_ws_close(self, ws, close_status_code, close_message):
         self.ws_connected = False
-        logger.warn(
-            f"Websocket closed with status {close_status_code}: {close_message}"
-        )
+        logger.warn(f"Websocket closed with status {close_status_code}: {close_message}")
 
     def on_ws_error(self, ws, error):
         self.disconnect_ws()
@@ -333,11 +316,7 @@ class Client:
                         return
 
                     # indicate that the response id is tied to the human message id
-                    elif (
-                        key != "pending"
-                        and value == None
-                        and message["state"] != "complete"
-                    ):
+                    elif key != "pending" and value == None and message["state"] != "complete":
                         self.active_messages[key] = message["messageId"]
                         self.message_queues[key].put(message)
                         return
@@ -381,9 +360,7 @@ class Client:
             human_message = message_data["data"]["messageEdgeCreate"]["message"]
             human_message_id = human_message["node"]["messageId"]
         except TypeError:
-            raise RuntimeError(
-                f"An unknown error occurred. Raw response data: {message_data}"
-            )
+            raise RuntimeError(f"An unknown error occurred. Raw response data: {message_data}")
 
         # indicate that the current message is waiting for a response
         self.active_messages[human_message_id] = None
@@ -418,9 +395,7 @@ class Client:
 
     def send_chat_break(self, chatbot):
         logger.info(f"Sending chat break to {chatbot}")
-        result = self.send_query(
-            "AddMessageBreakMutation", {"chatId": self.bots[chatbot]["chatId"]}
-        )
+        result = self.send_query("AddMessageBreakMutation", {"chatId": self.bots[chatbot]["chatId"]})
         return result["data"]["messageBreakCreate"]["message"]
 
     def get_message_history(self, chatbot, count=25, cursor=None):
@@ -437,15 +412,11 @@ class Client:
 
         cursor = str(cursor)
         if count > 50:
-            messages = (
-                self.get_message_history(chatbot, count=50, cursor=cursor) + messages
-            )
+            messages = self.get_message_history(chatbot, count=50, cursor=cursor) + messages
             while count > 0:
                 count -= 50
                 new_cursor = messages[0]["cursor"]
-                new_messages = self.get_message_history(
-                    chatbot, min(50, count), cursor=new_cursor
-                )
+                new_messages = self.get_message_history(chatbot, min(50, count), cursor=new_cursor)
                 messages = new_messages + messages
             return messages
         elif count <= 0:
@@ -523,9 +494,7 @@ class Client:
 
         data = result["data"]["poeBotCreate"]
         if data["status"] != "success":
-            raise RuntimeError(
-                f"Poe returned an error while trying to create a bot: {data['status']}"
-            )
+            raise RuntimeError(f"Poe returned an error while trying to create a bot: {data['status']}")
         self.get_bots()
         return data
 
@@ -568,9 +537,7 @@ class Client:
 
         data = result["data"]["poeBotEdit"]
         if data["status"] != "success":
-            raise RuntimeError(
-                f"Poe returned an error while trying to edit a bot: {data['status']}"
-            )
+            raise RuntimeError(f"Poe returned an error while trying to edit a bot: {data['status']}")
         self.get_bots()
         return data
 

From 7ad5b95842640c6ef9d9c0cb32d0aacd0d590d55 Mon Sep 17 00:00:00 2001
From: Raju Komati <komatiraju032@gmail.com>
Date: Thu, 27 Apr 2023 04:13:09 +0530
Subject: [PATCH 02/13] removed unused variables

---
 quora/__init__.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/quora/__init__.py b/quora/__init__.py
index b9ca6e5..d0ed302 100644
--- a/quora/__init__.py
+++ b/quora/__init__.py
@@ -21,9 +21,6 @@ from tls_client import Session as TLS
 from quora.api import Client as PoeClient
 from quora.mail import Emailnator
 
-CHROME_DRIVER_URL = 'https://chromedriver.storage.googleapis.com'
-FIREFOX_DRIVER_URL = 'https://api.github.com/repos/mozilla/geckodriver/releases'
-
 SELENIUM_WEB_DRIVER_ERROR_MSG = b'''The error message you are receiving is due to the `geckodriver` executable not 
 being found in your system\'s PATH. To resolve this issue, you need to download the geckodriver and add its location 
 to your system\'s PATH.\n\nHere are the steps to resolve the issue:\n\n1. Download the geckodriver for your platform 

From 68359d9a9a9c6f990ff7afd7303ec1bb393d59b7 Mon Sep 17 00:00:00 2001
From: Raju Komati <komatiraju032@gmail.com>
Date: Thu, 27 Apr 2023 04:14:27 +0530
Subject: [PATCH 03/13] updated quora readme

---
 quora/README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/quora/README.md b/quora/README.md
index 09f7536..1acd4a6 100644
--- a/quora/README.md
+++ b/quora/README.md
@@ -59,7 +59,7 @@ from quora import Poe
 
 # available models:  ['Sage', 'GPT-4', 'Claude+', 'Claude-instant', 'ChatGPT', 'Dragonfly', 'NeevaAI']
 
-poe = Poe(model='ChatGPT', driver='firefox', cookie_path='cookie.json')
+poe = Poe(model='ChatGPT', driver='firefox', cookie_path='cookie.json', driver_path='path_of_driver')
 poe.chat('who won the football world cup most?')
 
 # new bot creation

From bf5d773e525814a075f5195d2d17a33eb78be36a Mon Sep 17 00:00:00 2001
From: naa <44613678+naa7@users.noreply.github.com>
Date: Wed, 26 Apr 2023 23:48:40 -0400
Subject: [PATCH 04/13] Updated README.md

tweaked it a little bit for minor fixes
---
 README.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index b03a9e3..c68ed6f 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # GPT4free - use ChatGPT, for free!!
 
 ##### You may join our discord server for updates and support ; ) 
-- https://discord.gg/gpt4free 
+- [Discord Link](https://discord.gg/gpt4free)
 
 <img width="1383" alt="image" src="https://user-images.githubusercontent.com/98614666/233799515-1a7cb6a3-b17f-42c4-956d-8d2a0664466f.png">
 
@@ -14,7 +14,7 @@ By the way, thank you so much for [![Stars](https://img.shields.io/github/stars/
 ## Announcement
 Dear  Gpt4free Community,
 
-I want to thank you for your interest in and support of this project, which I only intended to be for entertainment and educational purposes; I had no idea it would end up being so popular.
+I would like to thank you for your interest in and support of this project, which I only intended to be for entertainment and educational purposes; I had no idea it would end up being so popular.
 
 I'm aware of the concerns about the project's legality and its impact on smaller sites hosting APIs. I take these concerns seriously and plan to address them.
 
@@ -27,7 +27,7 @@ Here's what I'm doing to fix these issues:
 Thank you for your support and understanding. I appreciate your continued interest in gpt4free and am committed to addressing your concerns.
 
 Sincerely,
-xtekky
+**xtekky**
 
 ## Legal Notice <a name="legal-notice"></a>
 

From d5d5d8b1d02bbee8132d9e35a75ccc4188844498 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 27 Apr 2023 09:59:15 +0100
Subject: [PATCH 05/13] Update main.py

---
 unfinished/easyai/main.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/unfinished/easyai/main.py b/unfinished/easyai/main.py
index 3b24697..95abb12 100644
--- a/unfinished/easyai/main.py
+++ b/unfinished/easyai/main.py
@@ -28,5 +28,4 @@ while True:
         
         if b'content' in chunk:
             data = loads(chunk.decode('utf-8').split('data:')[1])
-            print(data['content'], end='')
-            
\ No newline at end of file
+            print(data['content'], end='')
\ No newline at end of file

From 98d2b4109e8cd06ffd46ce9fe747bf889a4e7fff Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 27 Apr 2023 10:34:05 +0100
Subject: [PATCH 06/13] forefront (gpt-4)

---
 README.md                 |   7 +-
 forefront/README.md       |  15 ++++
 forefront/__init__.py     | 145 ++++++++++++++++++++++++++++++++++++++
 forefront/mail.py         |  55 +++++++++++++++
 forefront/typing.py       |  37 ++++++++++
 testing/forefront_test.py |  11 +++
 6 files changed, 269 insertions(+), 1 deletion(-)
 create mode 100644 forefront/README.md
 create mode 100644 forefront/__init__.py
 create mode 100644 forefront/mail.py
 create mode 100644 forefront/typing.py
 create mode 100644 testing/forefront_test.py

diff --git a/README.md b/README.md
index c68ed6f..0703770 100644
--- a/README.md
+++ b/README.md
@@ -55,8 +55,12 @@ Please note the following:
 | **Legal Notice** | Legal notice or disclaimer | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#legal-notice) | - |
 | **Copyright** | Copyright information | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#copyright) | - |
 | **Usage Examples** | | | |
+| `forefront` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
+
 | `quora (poe)` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./quora/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
+
 | `phind` | Example usage for phind | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./phind/README.md) | ![Inactive](https://img.shields.io/badge/Active-brightgreen) |
+
 | `you` | Example usage for you | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./you/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen)
 | **Try it Out** | | | |
 | Google Colab Jupyter Notebook | Example usage for gpt4free | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DanielShemesh/gpt4free-colab/blob/main/gpt4free.ipynb) | - |
@@ -74,8 +78,9 @@ Please note the following:
 
 ## Current Sites <a name="current-sites"></a>
 
-| Website                                              | Model(s)                        |
+| Website                                    s          | Model(s)                        |
 | ---------------------------------------------------- | ------------------------------- |
+| [forefront.ai](https://chat.forefront.ai)            | GPT-4/3.5                       |
 | [poe.com](https://poe.com)                           | GPT-4/3.5                       |
 | [writesonic.com](https://writesonic.com)             | GPT-3.5 / Internet              |
 | [t3nsor.com](https://t3nsor.com)                     | GPT-3.5                         |
diff --git a/forefront/README.md b/forefront/README.md
new file mode 100644
index 0000000..5b084af
--- /dev/null
+++ b/forefront/README.md
@@ -0,0 +1,15 @@
+### Example: `forefront` (use like openai pypi package) <a name="example-forefront"></a>
+
+```python
+import forefront
+
+# create an account
+token = forefront.Account.create(logging=True)
+print(token)
+
+# get a response
+for response in forefront.StreamingCompletion.create(token = token,
+    prompt = 'hello world', model='gpt-4'):
+    
+    print(response.completion.choices[0].text, end = '')
+```
\ No newline at end of file
diff --git a/forefront/__init__.py b/forefront/__init__.py
new file mode 100644
index 0000000..b247503
--- /dev/null
+++ b/forefront/__init__.py
@@ -0,0 +1,145 @@
+from tls_client import Session
+from forefront.mail import Mail
+from time import time, sleep
+from re import match
+from forefront.typing import ForeFrontResponse
+from uuid import uuid4
+from requests import post
+from json import loads
+
+
+class Account:
+    def create(proxy = None, logging = False):
+        
+        proxies = {
+            'http': 'http://' + proxy,
+            'https': 'http://' + proxy } if proxy else False
+        
+        start = time()
+
+        mail           = Mail(proxies)
+        mail_token     = None
+        mail_adress    = mail.get_mail()
+        
+        #print(mail_adress)
+        
+        client         = Session(client_identifier='chrome110')
+        client.proxies = proxies
+        client.headers = {
+            "origin": "https://accounts.forefront.ai",
+            "user-agent" : "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36",
+        }
+        
+        response = client.post('https://clerk.forefront.ai/v1/client/sign_ups?_clerk_js_version=4.32.6', 
+            data    = { 
+                "email_address": mail_adress 
+            }
+        )
+        
+        trace_token = response.json()['response']['id']
+        if logging: print(trace_token)
+
+        response = client.post(f"https://clerk.forefront.ai/v1/client/sign_ups/{trace_token}/prepare_verification?_clerk_js_version=4.32.6", 
+            data    = { 
+                "strategy"     : "email_code",
+            }
+        )
+        
+        if logging: print(response.text)
+
+        if not 'sign_up_attempt' in response.text:
+            return 'Failed to create account!'
+
+        while True:
+            sleep(1)
+            for _ in mail.fetch_inbox():
+                print(mail.get_message_content(_["id"]))
+                mail_token = match(r"(\d){5,6}", mail.get_message_content(_["id"])).group(0)
+
+            if mail_token:
+                break
+        
+        if logging: print(mail_token)
+        
+        response = client.post(f'https://clerk.forefront.ai/v1/client/sign_ups/{trace_token}/attempt_verification?_clerk_js_version=4.38.4', data = {
+            'code': mail_token,
+            'strategy': 'email_code'
+        })
+        
+        if logging: print(response.json())
+        
+        token     = response.json()['client']['sessions'][0]['last_active_token']['jwt']
+        
+        with open('accounts.txt', 'a') as f:
+            f.write(f'{mail_adress}:{token}\n')
+            
+        if logging: print(time() - start)
+        
+        return token
+
+
+class StreamingCompletion:
+    def create(
+        token  = None,
+        chatId = None,
+        prompt = '',
+        actionType = 'new',
+        defaultPersona = '607e41fe-95be-497e-8e97-010a59b2e2c0', # default
+        model = 'gpt-4') -> ForeFrontResponse:
+        
+        if not token: raise Exception('Token is required!')
+        if not chatId: chatId = str(uuid4())
+        
+        headers = {
+            'authority'         : 'chat-server.tenant-forefront-default.knative.chi.coreweave.com',
+            'accept'            : '*/*',
+            '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',
+            'authorization'     : 'Bearer ' + token,
+            'cache-control'     : 'no-cache',
+            'content-type'      : 'application/json',
+            'origin'            : 'https://chat.forefront.ai',
+            'pragma'            : 'no-cache',
+            'referer'           : 'https://chat.forefront.ai/',
+            'sec-ch-ua'         : '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+            'sec-ch-ua-mobile'  : '?0',
+            'sec-ch-ua-platform': '"macOS"',
+            'sec-fetch-dest'    : 'empty',
+            'sec-fetch-mode'    : 'cors',
+            'sec-fetch-site'    : 'cross-site',
+            'user-agent'        : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+        }
+
+        json_data = {
+            'text'           : prompt,
+            'action'         : actionType,
+            'parentId'       : chatId,
+            'workspaceId'    : chatId,
+            'messagePersona' : defaultPersona,
+            'model'          : model
+        }
+
+        for chunk in post('https://chat-server.tenant-forefront-default.knative.chi.coreweave.com/chat',
+            headers=headers, json=json_data, stream=True).iter_lines():
+            
+            if b'finish_reason":null' in chunk:
+                data  = loads(chunk.decode('utf-8').split('data: ')[1])
+                token = data['choices'][0]['delta'].get('content')
+                
+                if token != None:
+                    yield ForeFrontResponse({
+                                'id'     : chatId, 
+                                'object' : 'text_completion', 
+                                'created': int(time()), 
+                                'model'  : model, 
+                                'choices': [{
+                                        'text'          : token, 
+                                        'index'         : 0, 
+                                        'logprobs'      : None, 
+                                        'finish_reason' : 'stop'
+                                }],
+                                'usage': {
+                                    'prompt_tokens'     : len(prompt), 
+                                    'completion_tokens' : len(token), 
+                                    'total_tokens'      : len(prompt) + len(token)
+                                }
+                            })
\ No newline at end of file
diff --git a/forefront/mail.py b/forefront/mail.py
new file mode 100644
index 0000000..64694e7
--- /dev/null
+++ b/forefront/mail.py
@@ -0,0 +1,55 @@
+from requests import Session
+from string import ascii_letters
+from random import choices
+
+class Mail:
+    def __init__(self, proxies: dict = None) -> None:
+        self.client = Session()
+        self.client.proxies = proxies
+        self.client.headers = {
+            "host": "api.mail.tm",
+            "connection": "keep-alive",
+            "sec-ch-ua": "\"Google Chrome\";v=\"111\", \"Not(A:Brand\";v=\"8\", \"Chromium\";v=\"111\"",
+            "accept": "application/json, text/plain, */*",
+            "content-type": "application/json",
+            "sec-ch-ua-mobile": "?0",
+            "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36",
+            "sec-ch-ua-platform": "\"macOS\"",
+            "origin": "https://mail.tm",
+            "sec-fetch-site": "same-site",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-dest": "empty",
+            "referer": "https://mail.tm/",
+            "accept-encoding": "gzip, deflate, br",
+            "accept-language": "en-GB,en-US;q=0.9,en;q=0.8"
+        }
+        
+    def get_mail(self) -> str:
+        token = ''.join(choices(ascii_letters, k=14)).lower()
+        init  = self.client.post("https://api.mail.tm/accounts", json={
+            "address" : f"{token}@bugfoo.com",
+            "password": token
+        })
+        
+        if init.status_code == 201:
+            resp = self.client.post("https://api.mail.tm/token", json = {
+                **init.json(),
+                "password": token
+            })
+            
+            self.client.headers['authorization'] = 'Bearer ' + resp.json()['token']
+            
+            return f"{token}@bugfoo.com"
+        
+        else:
+            raise Exception("Failed to create email")
+    
+    def fetch_inbox(self):
+        return self.client.get(f"https://api.mail.tm/messages").json()["hydra:member"]
+
+    def get_message(self, message_id: str):
+        return self.client.get(f"https://api.mail.tm/messages/{message_id}").json()
+
+    def get_message_content(self, message_id: str):
+        return self.get_message(message_id)["text"]
+
diff --git a/forefront/typing.py b/forefront/typing.py
new file mode 100644
index 0000000..0fff6b1
--- /dev/null
+++ b/forefront/typing.py
@@ -0,0 +1,37 @@
+class ForeFrontResponse:
+    class Completion:
+        class Choices:
+            def __init__(self, choice: dict) -> None:
+                self.text           = choice['text']
+                self.content        = self.text.encode()
+                self.index          = choice['index']
+                self.logprobs       = choice['logprobs']
+                self.finish_reason  = choice['finish_reason']
+                
+            def __repr__(self) -> str:
+                return f'''<__main__.APIResponse.Completion.Choices(\n    text           = {self.text.encode()},\n    index          = {self.index},\n    logprobs       = {self.logprobs},\n    finish_reason  = {self.finish_reason})object at 0x1337>'''
+
+        def __init__(self, choices: dict) -> None:
+            self.choices = [self.Choices(choice) for choice in choices]
+
+    class Usage:
+        def __init__(self, usage_dict: dict) -> None:
+            self.prompt_tokens      = usage_dict['prompt_tokens']
+            self.completion_tokens  = usage_dict['completion_tokens']
+            self.total_tokens       = usage_dict['total_tokens']
+
+        def __repr__(self):
+            return f'''<__main__.APIResponse.Usage(\n    prompt_tokens      = {self.prompt_tokens},\n    completion_tokens  = {self.completion_tokens},\n    total_tokens       = {self.total_tokens})object at 0x1337>'''
+    
+    def __init__(self, response_dict: dict) -> None:
+        
+        self.response_dict  = response_dict
+        self.id             = response_dict['id']
+        self.object         = response_dict['object']
+        self.created        = response_dict['created']
+        self.model          = response_dict['model']
+        self.completion     = self.Completion(response_dict['choices'])
+        self.usage          = self.Usage(response_dict['usage'])
+
+    def json(self) -> dict:
+        return self.response_dict
\ No newline at end of file
diff --git a/testing/forefront_test.py b/testing/forefront_test.py
new file mode 100644
index 0000000..b5c682b
--- /dev/null
+++ b/testing/forefront_test.py
@@ -0,0 +1,11 @@
+import forefront
+
+# create an account
+token = forefront.Account.create(logging=True)
+print(token)
+
+# get a response
+for response in forefront.StreamingCompletion.create(token = token,
+    prompt = 'hello world', model='gpt-4'):
+    
+    print(response.completion.choices[0].text, end = '')
\ No newline at end of file

From 3e48284e40275592a76d9feb2cb44c8d12f5465d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=93=20sanz?= <sanzbernward@gmail.com>
Date: Thu, 27 Apr 2023 17:49:42 +0800
Subject: [PATCH 07/13] Create gpt4free.def

---
 Singularity/gpt4free.def | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 Singularity/gpt4free.def

diff --git a/Singularity/gpt4free.def b/Singularity/gpt4free.def
new file mode 100644
index 0000000..67bc124
--- /dev/null
+++ b/Singularity/gpt4free.def
@@ -0,0 +1,15 @@
+Bootstrap: docker
+From: python:3.10-slim
+
+%post
+    apt-get update && apt-get install -y git
+    git clone https://github.com/xtekky/gpt4free.git
+    cd gpt4free
+    pip install --no-cache-dir -r requirements.txt
+    cp gui/streamlit_app.py .
+
+%expose
+    8501
+
+%startscript
+    exec streamlit run streamlit_app.py

From 9f1a159c57af2643c3b7f478d2e36867c338f3ff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=93=20sanz?= <sanzbernward@gmail.com>
Date: Thu, 27 Apr 2023 17:51:42 +0800
Subject: [PATCH 08/13] Update and rename gpt4free.def to gpt4free.sif

---
 Singularity/{gpt4free.def => gpt4free.sif} | 0
 1 file changed, 0 insertions(+), 0 deletions(-)
 rename Singularity/{gpt4free.def => gpt4free.sif} (100%)

diff --git a/Singularity/gpt4free.def b/Singularity/gpt4free.sif
similarity index 100%
rename from Singularity/gpt4free.def
rename to Singularity/gpt4free.sif

From 992caddc30404e5b752f1c9d231e76a6527b4d6e Mon Sep 17 00:00:00 2001
From: Daniel Shemesh <danielshsycamore@gmail.com>
Date: Thu, 27 Apr 2023 13:01:02 +0300
Subject: [PATCH 09/13] Update README.md

fixed table
---
 README.md | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 0703770..db5ea34 100644
--- a/README.md
+++ b/README.md
@@ -55,13 +55,10 @@ Please note the following:
 | **Legal Notice** | Legal notice or disclaimer | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#legal-notice) | - |
 | **Copyright** | Copyright information | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#copyright) | - |
 | **Usage Examples** | | | |
-| `forefront` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
-
-| `quora (poe)` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./quora/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
-
+| `forefront` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | | |
+| `quora (poe)` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./quora/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | |
 | `phind` | Example usage for phind | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./phind/README.md) | ![Inactive](https://img.shields.io/badge/Active-brightgreen) |
-
-| `you` | Example usage for you | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./you/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen)
+| `you` | Example usage for you | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./you/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
 | **Try it Out** | | | |
 | Google Colab Jupyter Notebook | Example usage for gpt4free | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DanielShemesh/gpt4free-colab/blob/main/gpt4free.ipynb) | - |
 | replit Example (feel free to fork this repl) | Example usage for gpt4free | [![](https://img.shields.io/badge/Open%20in-Replit-1A1E27?logo=replit)](https://replit.com/@gpt4free/gpt4free-webui) | - |

From d9e6cbc1df2dc9f8712eeccce8f130efd9585dd7 Mon Sep 17 00:00:00 2001
From: Aymane Hrouch <jo.aymane@gmail.com>
Date: Thu, 27 Apr 2023 11:05:42 +0100
Subject: [PATCH 10/13] Add twocaptcha to requirements

---
 requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 3f5e9e6..a567311 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,5 @@ colorama
 curl_cffi
 streamlit==1.21.0
 selenium
-fake-useragent
\ No newline at end of file
+fake-useragent
+twocaptcha

From 355b2979d86a68f9435df604d6981468ec1d2553 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 27 Apr 2023 12:15:17 +0100
Subject: [PATCH 11/13] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index db5ea34..ce84ddc 100644
--- a/README.md
+++ b/README.md
@@ -55,7 +55,7 @@ Please note the following:
 | **Legal Notice** | Legal notice or disclaimer | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#legal-notice) | - |
 | **Copyright** | Copyright information | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#copyright) | - |
 | **Usage Examples** | | | |
-| `forefront` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | | |
+| `forefront` | Example usage for forefront (gpt-4) | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | | |
 | `quora (poe)` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./quora/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | |
 | `phind` | Example usage for phind | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./phind/README.md) | ![Inactive](https://img.shields.io/badge/Active-brightgreen) |
 | `you` | Example usage for you | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./you/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |

From f42ed825a0686be99d4abb4c62b57e4fc9f80f05 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 27 Apr 2023 12:26:37 +0100
Subject: [PATCH 12/13] Update README.md

---
 README.md | 20 +-------------------
 1 file changed, 1 insertion(+), 19 deletions(-)

diff --git a/README.md b/README.md
index ce84ddc..6728511 100644
--- a/README.md
+++ b/README.md
@@ -9,25 +9,7 @@ Have you ever come across some amazing projects that you couldn't use **just bec
 
 **We've got you covered!** This repository offers **reverse-engineered** third-party APIs for `GPT-4/3.5`, sourced from various websites. You can simply **download** this repository, and use the available modules, which are designed to be used **just like OpenAI's official package**. **Unleash ChatGPT's potential for your projects, now!** You are welcome ; ).
 
-By the way, thank you so much for [![Stars](https://img.shields.io/github/stars/xtekky/gpt4free?style=social)](https://github.com/xtekky/gpt4free/stargazers) and all the support!!
-
-## Announcement
-Dear  Gpt4free Community,
-
-I would like to thank you for your interest in and support of this project, which I only intended to be for entertainment and educational purposes; I had no idea it would end up being so popular.
-
-I'm aware of the concerns about the project's legality and its impact on smaller sites hosting APIs. I take these concerns seriously and plan to address them.
-
-Here's what I'm doing to fix these issues:
-
-1. Removing APIs from smaller sites: To reduce the impact on smaller sites, I have removed their APIs from the repository. Please shoot me a dm if you are an owner of a site and want it removed.
-
-2. Commitment to ethical use: I want to emphasize my commitment to promoting ethical use of language models. I don't support any illegal or unethical behavior, and I expect users to follow the same principles.
-
-Thank you for your support and understanding. I appreciate your continued interest in gpt4free and am committed to addressing your concerns.
-
-Sincerely,
-**xtekky**
+By the way, thank you so much for `11k` stars and all the support!!
 
 ## Legal Notice <a name="legal-notice"></a>
 

From 28a820a8bc00c34e14cae2941250ce5ac150c987 Mon Sep 17 00:00:00 2001
From: Aymane Hrouch <jo.aymane@gmail.com>
Date: Thu, 27 Apr 2023 13:12:04 +0100
Subject: [PATCH 13/13] Fix module not found: add root directory to sys path

---
 gui/streamlit_app.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/gui/streamlit_app.py b/gui/streamlit_app.py
index ff1cb6d..44507e1 100644
--- a/gui/streamlit_app.py
+++ b/gui/streamlit_app.py
@@ -1,3 +1,8 @@
+import os
+import sys
+
+sys.path.append(os.path.join(os.path.dirname(__file__), os.path.pardir))
+
 import streamlit as st
 import phind
 
@@ -45,4 +50,4 @@ hide_streamlit_style = """
             footer {visibility: hidden;}
             </style>
             """
-st.markdown(hide_streamlit_style, unsafe_allow_html=True) 
\ No newline at end of file
+st.markdown(hide_streamlit_style, unsafe_allow_html=True)