main.py

from fastapi import FastAPI
import uvicorn
from api.predict_sales import router as predict_router
from api.get_features import router as features_router

# Initialize FastAPI app
app = FastAPI(
    title="Sales Prediction API", 
    version="1.0.0",
    description="API for sales prediction with auto-generated IDs and feature exploration"
)

# Include routers
app.include_router(predict_router, tags=["Prediction"])
app.include_router(features_router, tags=["Features"])

@app.get("/")
async def root():
    return {
        "message": "Sales Prediction API", 
        "version": "1.0.0",
        "endpoints": {
            "predict": "/predict-sales",
            "features": "/get-features", 
            "sample_data": "/sample-data",
            "docs": "/docs"
        }
    }

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)