Update main.py

main.py

@@ -15,6 +15,7 @@ from fastapi import FastAPI, UploadFile, File, Query
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 from skimage import color
+import hashlib
 
 # OOM PREVENTION
 torch.set_num_threads(1)
@@ -619,13 +620,23 @@ app.add_middleware(
 processor = WatermelonProcessor(MODEL_PATH)
 
 @app.get("/")
-def read_root(): return {"status": "Watermelon API is awake and running!"}
+def read_root(): return {"status": "Phenotyping API is awake and running!"}
 
 @app.post("/process_single")
-async def process_single(file: UploadFile = File(...), include_image: bool = Query(True), apply_smoothing: bool = Query(True)):
+async def process_single(
+    file: UploadFile = File(...), 
+    include_image: bool = Query(True), 
+    apply_smoothing: bool = Query(True),
+    password: str = Form("")
+):
     request_t = time.perf_counter()
-    contents = None
-    img = None
+    
+    # --- SECURE PASSWORD CHECK ---
+    expected_hash = "9139eb3676d5dfafced7613f044d86d9e7c84f40a04c83ddce062878621315d0"
+    if hashlib.sha256(password.encode('utf-8')).hexdigest() != expected_hash:
+        return ProcessResult(success=False, message="Unauthorized: Incorrect password.", filename=file.filename, processing_ms=int(round((time.perf_counter() - request_t) * 1000))).__dict__
+
+    contents, img = None, None
 
     try:
         contents = await file.read()