from dotenv import load_dotenv
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from langserve import add_routes

from graph import create_graph
from utils.chat_types import ChatInputType

# Load environment variables from .env file
load_dotenv()

app = FastAPI(
    title="CRAG Backend",
    version="1.0",
    description="Backend to run agent performing corrective RAG over annual reports",
)

# Configure CORS
origins = [
    "http://localhost",
    "http://localhost:3000",
]

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

graph = create_graph()


runnable = graph.with_types(input_type=ChatInputType, output_type=dict)

add_routes(app, runnable, path="/chat", playground_type="default")