Update app.py

app.py

@@ -154,20 +154,23 @@ class Scan(db.Model):
 
 # Load Model
 model = None
-try:
-    if os.path.exists(MODEL_PATH):
-        logger.info("Loading model from %s", MODEL_PATH)
-        try:
+model_load_error = None
+def load_model():
+    global model, model_load_error
+    try:
+        if os.path.exists(MODEL_PATH):
+            logger.info("Loading model from %s", MODEL_PATH)
             model = tf.keras.models.load_model(MODEL_PATH, compile=False)
-        except Exception as e:
-            logger.error("Failed to deserialize model: %s", str(e))
-            raise ValueError(f"Model deserialization error: {str(e)}. Please ensure the model is compatible with TensorFlow 2.15.0 or re-save it.")
-    else:
-        logger.error("Model file %s not found", MODEL_PATH)
-        raise FileNotFoundError(f"Model file {MODEL_PATH} not found")
-except Exception as e:
-    logger.error("Failed to load model: %s", str(e))
-    raise
+            logger.info("Model loaded successfully")
+        else:
+            logger.error("Model file %s not found", MODEL_PATH)
+            model_load_error = f"Model file {MODEL_PATH} not found"
+    except Exception as e:
+        logger.error("Failed to load model: %s", str(e))
+        model_load_error = f"Model deserialization error: {str(e)}. Please ensure the model is compatible with TensorFlow 2.15.0 or re-save it."
+
+# Attempt to load model at startup
+load_model()
 
 # Plot training history
 if os.path.exists(HISTORY_PATH):
@@ -351,6 +354,14 @@ def form():
         # Check for required environment variables
         if not app.config['MAIL_USERNAME'] or not app.config['MAIL_PASSWORD']:
             logger.warning("Mail configuration missing, email functionality may fail")
+        # Check if model loaded successfully
+        if model_load_error:
+            return render_template("form.html", history_plot="/training_plot.png", result={
+                "prediction": "Error",
+                "confidence": "N/A",
+                "message": f"Model loading failed: {model_load_error}",
+                "email_status": "N/A"
+            })
         return render_template("form.html", history_plot="/training_plot.png")
     except Exception as e:
         logger.error("Error rendering form: %s", str(e))
@@ -442,8 +453,8 @@ def email_report(scan_id):
 @app.route("/predict", methods=["POST"])
 def predict():
     try:
-        if not model:
-            raise ValueError("Model not loaded, cannot make predictions")
+        if model_load_error or not model:
+            raise ValueError(f"Model not loaded: {model_load_error}")
         if "image" not in request.files:
             raise ValueError("No image uploaded.")
         image = request.files["image"]
@@ -530,7 +541,7 @@ def predict():
 def result():
     try:
         report = session.get("report", {})
-        return render_template("result.html", **report)
+        return render_template("form.html", **report)
     except Exception as e:
         logger.error("Error rendering result: %s", str(e))
         return render_template("form.html", history_plot="/training_plot.png", result={