update to include the root endpoint

src/main.py

@@ -1,9 +1,24 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from pydantic import BaseModel, validator
 import pandas as pd
 import pickle, uvicorn, os, logging
 
-app = FastAPI()
+
+# API Config
+app = FastAPI(
+    # title="Agri-Tech API",
+    # description="This is a ML API for classification of crop to plant on a land regarding some features",
+)
+
+## Configure static and template files
+app.mount(
+    "/static", StaticFiles(directory="assets/static"), name="static"
+)  # Mount static files
+templates = Jinja2Templates(directory="assets/templates")  # Mount templates for HTML
+
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -109,5 +124,11 @@ async def predict_price(device_id: int, specs: DeviceSpecs):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+# Root endpoint to serve index.html template
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
+
+
 if __name__ == "__main__":
     uvicorn.run(app, host="127.0.0.1", port=8000, reload=True)