app.py

from fastapi import FastAPI
from dotenv import load_dotenv
from tasks import text, image, audio

# Load environment variables
load_dotenv()

app = FastAPI(
    title="Frugal AI Challenge API",
    description="API for the Frugal AI Challenge evaluation endpoints"
)
from fastapi import FastAPI
from tasks.text import router as text_router

@app.get("/health")
async def health_check():
    return {"status": "ok"}

app.include_router(text_router)
# Include all routers
# app.include_router(text.router)
# app.include_router(image.router)
# app.include_router(audio.router)

@app.get("/")
async def root():
    return {
        "message": "Welcome to the Frugal AI Challenge API",
        "endpoints": {
            "text": "/text - Text classification task"#,
#            "image": "/image - Image classification task (coming soon)",
#            "audio": "/audio - Audio classification task (coming soon)"
        }
    }