FastAPi script for Edgegpt

[raymerjacque]

_from fastapi import FastAPI, WebSocket
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from EdgeGPT import Chatbot, ConversationStyle
import json

class InputData(BaseModel):
    userId: int
    prompt: str
    style: str

app = FastAPI()
user_sessions = {}


@app.on_event("startup")
async def startup_event():
    # Connect to EdgeGPT
    cookies = json.loads(open("cookies.json", encoding="utf-8").read())
    user_sessions[0] = {
        "style": "creative",
        "bot": await Chatbot.create(cookies=cookies)
    }

@app.post("/api/ask")
async def ask(data: InputData):
    userId = data.userId
    prompt = data.prompt
    style = data.style
    url = "https://api.edgegpt.com/v1/ask";

    print(f"Received request from user {userId}:", prompt)

    # Get user's session or use the default session
    user_data = user_sessions.get(userId, user_sessions[0])
    bot = user_data["bot"]

    if style == "balanced":
        style = ConversationStyle.balanced
    elif style == "creative":
        style = ConversationStyle.creative
    elif style == "precise":
        style = ConversationStyle.precise

    # Send prompt to EdgeGPT and get response
    try:
      response = await bot.ask(prompt=prompt, conversation_style=style, url=url)
    except Exception as e:
      print(e)
      return {"response": ""}

    return {"response": response, "chat_id": userId}

@app.post("/api/newchat")
async def newchat(jsonData: dict):
    id = len(user_sessions)
    style = jsonData.get("style", "creative")
    cookies = jsonData.get("cookie", {})

    bot = await Chatbot.create(cookies=cookies)

    user_sessions[id] = {
        "id": id,
        "style": style,
        "bot": bot,
        "cookie": cookies
    }

    return user_sessions[id]

@app.get("/api/get")
async def get(jsonData: dict):
    id = jsonData["id"]
    return user_sessions[id]

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=5000)_

####################

This is my attempt at making a Fastapi script with Edgegpt, However it has failed, i suspect im not using the url correctly, I do not know how to code the wss link into fastapi... i do have a cookies.json file i exported and is in the same directory as this script. I would appreciate any assistance on how to fix this to make it work. thank you.