Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Create Gameli _file #39

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
75 changes: 75 additions & 0 deletions Gameli _file
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
"""Make some requests to OpenAI's chatbot"""

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()
BROWSER = PLAY.chromium.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):
# Send the 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()