from fastapi import FastAPI, Request
from pydantic import BaseModel
import openai
import os

openai.api_key = os.getenv("OPENAI_API_KEY")

app = FastAPI()

class ChatRequest(BaseModel):
    message: str
    history: list = []

@app.post("/chat")
async def chat(data: ChatRequest):
    messages = [{"role": "system", "content": "You are a helpful assistant."}]
    messages += data.history
    messages.append({"role": "user", "content": data.message})

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )

    reply = response['choices'][0]['message']['content']
    return {
        "reply": reply,
        "history": messages + [{"role": "assistant", "content": reply}]
    }