revert back to non-lazy one

main.py

@@ -1,21 +1,24 @@
-from fastapi import FastAPI
+# app/main.py
+
+from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+import shutil
+from pathlib import Path
 import uvicorn
+
 from vit_captioning.generate import CaptionGenerator
 
 app = FastAPI()
-caption_generator = None  # Lazy-load placeholder
-
-@app.on_event("startup")
-def startup_event():
-    global caption_generator
-    if caption_generator is None:
-        print("Loading CaptionGenerator...")
-        caption_generator = CaptionGenerator(
-            model_type="CLIPEncoder",
-            checkpoint_path="./vit_captioning/artifacts/CLIPEncoder_40epochs_unfreeze12.pth",
-            quantized=False
-        )
+
+# Serve static files
+static_dir = Path(__file__).parent / "vit_captioning" / "static"
+app.mount("/static", StaticFiles(directory=static_dir), name="static")
+
+# ✅ Landing page at `/`
+# @app.get("/", response_class=HTMLResponse)
+# async def landing():
+#     return Path("vit_captioning/static/landing.html").read_text()
 
 @app.get("/", response_class=HTMLResponse)
 def root():
@@ -25,12 +28,32 @@ def root():
 def health_check():
     return {"status": "ok"}
 
-# Example endpoint to trigger model
-@app.get("/caption")
-def caption():
-    if caption_generator is None:
-        return {"error": "Model not loaded"}
-    return {"result": "dummy caption"}  # Replace with real logic
+# ✅ Captioning page at `/captioning`
+@app.get("/captioning", response_class=HTMLResponse)
+async def captioning():
+    return Path("vit_captioning/static/captioning/index.html").read_text()
+
+# ✅ Example: Project 2 placeholder
+@app.get("/project2", response_class=HTMLResponse)
+async def project2():
+    return "<h1>Coming Soon: Project 2</h1>"
+
+# ✅ Caption generation endpoint for captioning app
+# Keep the path consistent with your JS fetch()!
+caption_generator = CaptionGenerator(
+    model_type="CLIPEncoder",
+    checkpoint_path="./vit_captioning/artifacts/CLIPEncoder_40epochs_unfreeze12.pth",
+    quantized=False
+)
+
+@app.post("/generate")
+async def generate(file: UploadFile = File(...)):
+    temp_file = f"temp_{file.filename}"
+    with open(temp_file, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    captions = caption_generator.generate_caption(temp_file)
+    return captions
 
 # if __name__ == "__main__":
 #     uvicorn.run(app, host="0.0.0.0", port=8000)