Update app.py

app.py

@@ -12,37 +12,35 @@ from flask_cors import CORS
 # Suppress sklearn warnings
 warnings.filterwarnings('ignore', category=UserWarning, module='sklearn')
 
-# Configure logging
+# Logging setup
 logging.basicConfig(level=logging.INFO)
 
-# Get model URLs from environment variables
+# Model URLs from env
 DIABETES_MODEL_URL = os.getenv("DIABETES_MODEL_URL")
 SCALER_URL = os.getenv("SCALER_URL")
 MULTI_MODEL_URL = os.getenv("MULTI_MODEL_URL")
 
-# Local paths for downloaded models
+# Model ZIP names
 MODEL_PATHS = {
     "DIABETES_MODEL": "finaliseddiabetes_model.zip",
     "SCALER": "finalisedscaler.zip",
     "MULTI_MODEL": "nodiabetes.zip",
 }
 
-# Extracted model names
+# Extracted joblib names
 EXTRACTED_MODELS = {
     "DIABETES_MODEL": "finaliseddiabetes_model.joblib",
     "SCALER": "finalisedscaler.joblib",
     "MULTI_MODEL": "nodiabetes.joblib",
 }
 
-BASE_DIR = os.getcwd()
+# Use writeable temp dir
+TMP_DIR = "/tmp"
 
-# Flask app initialization
+# Flask app init
 app = Flask(__name__)
-
-# Enable CORS for all origins
 CORS(app, supports_credentials=True)
 
-# Root route
 @app.route('/')
 def index():
     return """
@@ -52,7 +50,7 @@ def index():
     """
 
 def download_model(url, zip_filename):
-    zip_path = os.path.join(BASE_DIR, zip_filename)
+    zip_path = os.path.join(TMP_DIR, zip_filename)
     if not url:
         logging.error(f"URL for {zip_filename} is missing!")
         return False
@@ -71,8 +69,8 @@ def download_model(url, zip_filename):
         return False
 
 def extract_if_needed(zip_filename, extracted_filename):
-    zip_path = os.path.join(BASE_DIR, zip_filename)
-    extracted_path = os.path.join(BASE_DIR, extracted_filename)
+    zip_path = os.path.join(TMP_DIR, zip_filename)
+    extracted_path = os.path.join(TMP_DIR, extracted_filename)
     if os.path.exists(extracted_path):
         logging.info(f"{extracted_filename} already exists. Skipping extraction.")
         return True
@@ -81,15 +79,15 @@ def extract_if_needed(zip_filename, extracted_filename):
         return False
     try:
         with zipfile.ZipFile(zip_path, 'r') as zip_ref:
-            zip_ref.extractall(BASE_DIR)
-        logging.info(f"Extracted {zip_filename}")
+            zip_ref.extractall(TMP_DIR)
+        logging.info(f"Extracted {zip_filename} to {TMP_DIR}")
         return True
     except Exception as e:
         logging.error(f"Error extracting {zip_filename}: {e}")
         return False
 
 def load_model(model_filename):
-    model_path = os.path.join(BASE_DIR, model_filename)
+    model_path = os.path.join(TMP_DIR, model_filename)
     if not os.path.exists(model_path):
         logging.error(f"Model file not found: {model_path}")
         return None
@@ -105,7 +103,8 @@ def initialize_models():
     models = {}
     for model_key, zip_filename in MODEL_PATHS.items():
         extracted_filename = EXTRACTED_MODELS[model_key]
-        if not os.path.exists(os.path.join(BASE_DIR, zip_filename)):
+        zip_path = os.path.join(TMP_DIR, zip_filename)
+        if not os.path.exists(zip_path):
             download_model(globals()[f"{model_key}_URL"], zip_filename)
         extract_if_needed(zip_filename, extracted_filename)
         models[model_key] = load_model(extracted_filename)