Spaces:
Runtime error
Runtime error
File size: 2,515 Bytes
5460635 8775662 5460635 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | """
Standalone keepalive script — opens the project's Playwright persistent profile
and visits https://chatgpt.com to let the site refresh session cookies.
Usage (from the project root, with your venv activated):
Windows Task Scheduler or cron can run this daily:
python keepalive.py
"""
import os
import time
import requests
from playwright.sync_api import sync_playwright
PROFILE_DIR = os.path.join(os.path.dirname(__file__), "chrome-profile-real")
SERVER_ENDPOINT = os.getenv("KEEPALIVE_ENDPOINT", "http://localhost:7860/internal/keepalive")
API_KEY = os.getenv("API_KEY", "")
def call_server_keepalive():
if not API_KEY:
print("No API key provided; skipping server keepalive call")
return False
try:
headers = {"Authorization": f"Bearer {API_KEY}"}
r = requests.post(SERVER_ENDPOINT, headers=headers, timeout=10)
if 200 <= r.status_code < 300:
print("Keepalive triggered via server endpoint (status=", r.status_code, ")")
return True
print("Server keepalive returned non-2xx:", r.status_code, r.text)
except Exception as e:
print("Server keepalive call failed:", e)
return False
def main():
# Prefer triggering the running server's internal keepalive (opens a new page in same context).
if call_server_keepalive():
print("Server-side keepalive completed")
return
# Fallback: open the profile directly (only if server is not running).
if not os.path.isdir(PROFILE_DIR):
print(f"Profile directory not found: {PROFILE_DIR}")
return
with sync_playwright() as pw:
browser_ctx = None
try:
browser_ctx = pw.chromium.launch_persistent_context(
user_data_dir=PROFILE_DIR,
headless=True,
)
page = browser_ctx.new_page()
page.goto("https://chatgpt.com/", wait_until="networkidle", timeout=60000)
# Give the site a little time to set/refresh cookies
time.sleep(5)
try:
page.close()
except Exception:
pass
except Exception as e:
print("Keepalive failed:", e)
finally:
try:
if browser_ctx:
browser_ctx.close()
except Exception:
pass
print("Keepalive completed")
if __name__ == "__main__":
main()
|