pgsoft/ai_patcher_interface

1 2from fastapi import APIRouter,WebSocket,WebSocketDisconnect3from fastapi.responses import FileResponse4import requests5import json6 7 8router = APIRouter(9    prefix="/watermelon",10    tags=["watermelon"],11)12game = "watermelon"13header = {'accept': 'application/json', 'Content-Type': 'application/json'}14url_base = "https://pgsoft-ai-patcher.hf.space/watermelon"15 16 17@router.get("/")18def index():19    return FileResponse(path="static/watermelon/index.html", media_type="text/html")20 21 22@router.websocket("/ws")23async def websocket_endpoint(websocket: WebSocket):24    await websocket.accept()25    while True:26        try:27            data = await websocket.receive_json()28        except WebSocketDisconnect:29            continue30        print(f"received data: {data}")31        url = f"{url_base}/{data['url']}"32        myobj = data["data"]33        try:34            res = requests.post(url, headers=header, json=myobj)35            print(res.status_code)36            await websocket.send_text(json.dumps(json.loads(res.text),indent=4))37        except Exception as e:38            res = {39                "result": "Failure",40                "detail": f"failed to request, {e}",41            }42            await websocket.send_text(json.dumps(res, indent=4))43 44