Spaces:
Running
Running
from fastapi import FastAPI, Request, staticfiles | |
from fastapi.responses import HTMLResponse | |
import os | |
import openai | |
app = FastAPI() | |
app.mount("/static", staticfiles.StaticFiles(directory="static"), name="static") | |
async def root(request: Request): | |
with open(os.path.join("static", "main.html")) as f: | |
return HTMLResponse(f.read()) | |
async def chat(request: Request): | |
data = await request.json() | |
message = data["message"] | |
openai.api_key = str(os.getenv("OPENAI_API_KEY")) | |
model_engine = "gpt-3.5-turbo" | |
response = openai.Completion.create( | |
engine=model_engine, | |
prompt=message, | |
max_tokens=3000, | |
n=1, | |
temperature=1, | |
frequency_penalty=0, | |
presence_penalty=0 | |
) | |
return response.choices[0] |