Update newapi.py

newapi.py

@@ -11,14 +11,12 @@ from huggingface_hub import hf_hub_download
 from models.TumorModel import TumorClassification, GliomaStageModel
 from utils import get_precautions_from_gemini
 
-# ✅ Create a writable cache directory inside the current working directory
-cache_dir = os.path.join(os.getcwd(), "cache")
+# ✅ Use /data as Hugging Face allows writing here only
+cache_dir = os.path.join("/data", "cache")
 os.makedirs(cache_dir, exist_ok=True)
 
-# ✅ Initialize FastAPI app
 app = FastAPI(title="Brain Tumor Detection API")
 
-# ✅ Enable CORS for frontend requests
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -47,7 +45,6 @@ glioma_model = GliomaStageModel()
 glioma_model.load_state_dict(torch.load(glioma_model_path, map_location="cpu"))
 glioma_model.eval()
 
-# ✅ Image preprocessing steps
 transform = transforms.Compose([
     transforms.Grayscale(),
     transforms.Resize((224, 224)),
@@ -55,15 +52,12 @@ transform = transforms.Compose([
     transforms.Normalize(mean=[0.5], std=[0.5]),
 ])
 
-# ✅ Health check route
 @app.get("/")
 async def root():
     return {"message": "Brain Tumor Detection API is running."}
 
-# ✅ Tumor labels
 labels = ['glioma', 'meningioma', 'notumor', 'pituitary']
 
-# ✅ Predict tumor type
 @app.post("/predict-image")
 async def predict_image(file: UploadFile = File(...)):
     img_bytes = await file.read()
@@ -81,7 +75,6 @@ async def predict_image(file: UploadFile = File(...)):
         precautions = get_precautions_from_gemini(tumor_type)
         return {"tumor_type": tumor_type, "precaution": precautions}
 
-# ✅ Input format for glioma prediction
 class MutationInput(BaseModel):
     gender: str
     age: float
@@ -93,7 +86,6 @@ class MutationInput(BaseModel):
     cic: int
     pik3ca: int
 
-# ✅ Predict glioma stage
 @app.post("/predict-glioma-stage")
 async def predict_glioma_stage(data: MutationInput):
     gender_val = 0 if data.gender.lower() == 'm' else 1
@@ -109,7 +101,7 @@ async def predict_glioma_stage(data: MutationInput):
         stages = ['Stage 1', 'Stage 2', 'Stage 3', 'Stage 4']
         return {"glioma_stage": stages[idx]}
 
-# ✅ Optional: Only used when running locally (ignored on Spaces)
+# Only used when running locally
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run("newapi:app", host="0.0.0.0", port=10000)