from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from src.grc import process_grc
from src.ml import model, tokenizer
from src.models import ProcessGrcInput
import os

app = FastAPI(title="GRC API")

origins = [os.environ.get("FRONTEND_ORIGIN", "")]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


@app.get("/")
def index():
    return "Welcome to the GRC API"


@app.post("/run")
def run(input: ProcessGrcInput):
    return process_grc(model=model, tokenizer=tokenizer, input=input)