Update utils.py

utils.py

@@ -1,4 +1,3 @@
-import torch
 import torch.nn as nn
 
 # ================================
@@ -43,15 +42,10 @@ class GliomaStageModel(nn.Module):
     def forward(self, x):
         return self.model(x)
 
-# ================================
-# 💡 PRECAUTIONS
-# ================================
-
-def get_precautions_from_gemini(tumor_type):
-    precaution_db = {
+def get_precautions_from_gemini(tumor_type: str) -> str:
+    db = {
         "meningioma": "Avoid radiation exposure and get regular check-ups.",
-        "pituitary": "Monitor hormonal levels and follow medication strictly.",
-        "notumor": "Stay healthy and get annual MRI scans if symptoms appear.",
-        "glioma": "Maintain a healthy lifestyle and follow up with neuro-oncologist."
+        "pituitary":   "Monitor hormonal levels and follow medication strictly.",
+        "notumor":     "Stay healthy and get annual MRI scans if symptoms appear."
     }
-    return precaution_db.get(tumor_type.lower(), "No specific precautions found.")
+    return db.get(tumor_type.lower(), "No specific precautions found.")