|
import time |
|
import os |
|
import flask |
|
import sys |
|
|
|
from flask import g |
|
|
|
from playwright.sync_api import sync_playwright |
|
|
|
PROFILE_DIR = "/tmp/playwright" if '--profile' not in sys.argv else sys.argv[sys.argv.index('--profile') + 1] |
|
PORT = 5001 if '--port' not in sys.argv else int(sys.argv[sys.argv.index('--port') + 1]) |
|
APP = flask.Flask(__name__) |
|
PLAY = sync_playwright().start() |
|
os.environ["PLAYWRIGHT_BROWSERS_PATH"] = "/usr/bin/firefox" |
|
BROWSER = PLAY.firefox.launch_persistent_context( |
|
user_data_dir=PROFILE_DIR, |
|
headless=False, |
|
) |
|
PAGE = BROWSER.new_page() |
|
|
|
def get_input_box(): |
|
"""Find the input box by searching for the largest visible one.""" |
|
textareas = PAGE.query_selector_all("textarea") |
|
candidate = None |
|
for textarea in textareas: |
|
if textarea.is_visible(): |
|
if candidate is None: |
|
candidate = textarea |
|
elif textarea.bounding_box().width > candidate.bounding_box().width: |
|
candidate = textarea |
|
return candidate |
|
|
|
def is_logged_in(): |
|
try: |
|
return get_input_box() is not None |
|
except AttributeError: |
|
return False |
|
|
|
def send_message(message): |
|
|
|
box = get_input_box() |
|
box.click() |
|
box.fill(message) |
|
box.press("Enter") |
|
while PAGE.query_selector(".result-streaming") is not None: |
|
time.sleep(0.1) |
|
|
|
def get_last_message(): |
|
"""Get the latest message""" |
|
page_elements = PAGE.query_selector_all(".flex.flex-col.items-center > div") |
|
last_element = page_elements[-2] |
|
return last_element.inner_text() |
|
|
|
@APP.route("/chat", methods=["GET"]) |
|
def chat(): |
|
message = flask.request.args.get("q") |
|
print("Sending message: ", message) |
|
send_message(message) |
|
response = get_last_message() |
|
print("Response: ", response) |
|
return response |
|
|
|
def start_browser(): |
|
PAGE.goto("https://chat.openai.com/") |
|
APP.run(port=PORT, threaded=False) |
|
if not is_logged_in(): |
|
print("Please log in to OpenAI Chat") |
|
print("Press enter when you're done") |
|
input() |
|
else: |
|
print("Logged in") |
|
|
|
if __name__ == "__main__": |
|
start_browser() |
|
|
