root endpoint as GUI

main.py

@@ -1,4 +1,7 @@
 from fastapi import FastAPI
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 import uvicorn
 from typing import List, Literal, Optional
 from pydantic import BaseModel
@@ -9,7 +12,8 @@ import json
 import logging
 
 # logger
-logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)
+logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.DEBUG)
+
 
 # Util Functions & Classes
 def loading(fp):
@@ -26,7 +30,7 @@ def predict(df, endpoint="simple"):
     print(
         f"[Info] 'predict' function has been called through the endpoint '{endpoint}'.\n"
     )
-    
+
     logging.info(f" \n{df.to_markdown()}")
 
     # scaling
@@ -96,8 +100,10 @@ class Lands(BaseModel):
 
 
 # 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")
+app = FastAPI(
+    title="Agri-Tech API",
+    description="This is a ML API for classification of crop to plant on a land regarding some features",
+)
 
 # ML Config
 ml_objects = loading(fp=os.path.join("assets", "ml", "crop_recommandation2.pkl"))
@@ -110,8 +116,23 @@ labels = ml_objects["labels"]
 # Endpoints
 @app.get("/")
 def root():
-    return {"Description": " This is a ML API for classification of crop to plant on a land regarding some features.",
-            "Documentation": "Go to the docs: https://eaedk-agri-tech-fastapi.hf.space/docs"}
+    return {
+        "Description": " This is a ML API for classification of crop to plant on a land regarding some features.",
+        "Documentation": "Go to the docs: https://eaedk-agri-tech-fastapi.hf.space/docs",
+    }
+
+
+# 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
+
+
+# Root endpoint to serve index.html template
+@app.get("/", response_class=HTMLResponse)
+def root(request):
+    return templates.TemplateResponse("index.html", {"request": request})
 
 
 @app.get("/checkup")