Update app.py

app.py

@@ -1,7 +1,7 @@
 from contextlib import asynccontextmanager
 from datetime import datetime
 from json import dumps, loads
-from logging import Formatter, INFO, DEBUG, StreamHandler, WARNING, getLogger
+from logging import Formatter, INFO, StreamHandler, WARNING, getLogger
 from os import environ
 from pathlib import Path
 from random import randint
@@ -11,7 +11,7 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from fastapi import Depends, FastAPI, HTTPException, Header, Request
 from fastapi.responses import FileResponse, HTMLResponse, JSONResponse, StreamingResponse
 from httpx import AsyncClient
-from playwright.async_api import Page, async_playwright
+from patchright.async_api import Page, async_playwright
 from starlette.responses import Response
 
 screenshot_path = Path(__file__).parent / 'screenshot.jpeg'
@@ -32,7 +32,7 @@ API_TOKEN = str(environ.get('API_TOKEN')).strip()
 REKA_API_URL = 'https://chat.reka.ai/api/chat'
 
 logger = getLogger('REKA_API')
-logger.setLevel(DEBUG)
+logger.setLevel(INFO)
 handler = StreamHandler()
 handler.setLevel(INFO)
 formatter = Formatter('%(asctime)s | %(levelname)s : %(message)s', datefmt='%d.%m.%Y %H:%M:%S')
@@ -51,10 +51,19 @@ async def make_screenshot(page: Page):
 async def refresh_token():
     max_timeout = BROWSER_TIMEOUT_SECONDS * 1000
     json_data = {'accessToken': None}
+    user_data_dir = Path(__file__).parent / 'user_data'
+    user_data_dir.mkdir(exist_ok=True, parents=True)
     async with async_playwright() as playwright:
         logger.info('запуск браузера')
         browser = await playwright.chromium.launch(headless=True, args=['--disable-blink-features=AutomationControlled'])
-        context = await browser.new_context(color_scheme='dark', ignore_https_errors=True, locale='en-US', user_agent=UA)
+        context = await browser.new_context(
+            color_scheme='dark',
+            ignore_https_errors=True,
+            locale='en-US',
+            user_agent=UA,
+            no_viewport=True,
+            user_data_dir=user_data_dir.resolve().as_posix()
+        )
         context.set_default_timeout(max_timeout)
         context.set_default_navigation_timeout(max_timeout)
         page = await context.new_page()
@@ -264,9 +273,8 @@ async def root():
     return HTMLResponse('ну пролапс, ну и что', status_code=200)
 
 
-@app.post('/v1/chat/completions')
 @app.post('/chat/completions')
-async def chat_completions(request: Request):
+async def chat_completions(request: Request, token: str = Depends(verify_token)):
     logger.debug('запрос `completions`')
     data = await request.json()
     messages = data.get('messages', [])
@@ -284,7 +292,11 @@ async def chat_completions(request: Request):
     return StreamingResponse(fetch_reka_stream(reka_data), media_type='text/event-stream')
 
 
-@app.get('/v1/models')
+@app.post('/v1/chat/completions')
+async def chat_completions_v1(request: Request, token: str = Depends(verify_token)):
+    return await chat_completions(request)
+
+
 @app.get('/models')
 async def models():
     logger.debug('запрос `models`')
@@ -294,15 +306,20 @@ async def models():
     }, status_code=200, media_type='application/json')
 
 
+@app.get('/v1/models')
+async def models_v1():
+    return await models()
+
+
 @app.get('/update_token')
-async def update_token():
+async def update_token(token: str = Depends(verify_token)):
     logger.info('запрос `update_token`')
     task = get_token()
     return JSONResponse({'status': 'обновление токена запущено'}, status_code=200, media_type='application/json')
 
 
 @app.get('/show_last_screen')
-async def show_last_screen():
+async def show_last_screen(token: str = Depends(verify_token)):
     logger.info('запрос `show_last_screen`')
     return FileResponse(screenshot_path.resolve().as_posix(), media_type='image/jpeg', status_code=200)