import sys, os
from fastapi import FastAPI, Form, Body
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse, JSONResponse
from typing import List, Dict
from pydantic import BaseModel
from utils import product_categorizer


app = FastAPI()

# Define allowed origins for CORS
origins = [
    "http://localhost:5300",
    "http://localhost:3000",
    # Add other allowed origins here
]

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



class ProductRequest(BaseModel):
    products: List[str]

@app.post("/categorize-products/")
async def categorize_products(request: ProductRequest):
    categorized_output = product_categorizer(request.products)
    return JSONResponse(content={"categorized_products": categorized_output})