Update app.py

app.py

@@ -131,7 +131,7 @@ recommendations = {
 }
 
 # Logger
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 # Database Models
@@ -344,13 +344,16 @@ def home():
 @app.route("/form")
 def form():
     try:
+        # 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")
         return render_template("form.html", history_plot="/training_plot.png")
     except Exception as e:
         logger.error("Error rendering form: %s", str(e))
         return render_template("form.html", history_plot=None, result={
             "prediction": "Error",
             "confidence": "N/A",
-            "message": "Failed to load form. Please try again.",
+            "message": f"Failed to load form: {str(e)}",
             "email_status": "N/A"
         })
 
@@ -488,6 +491,8 @@ def predict():
 
         # Send email automatically
         try:
+            if not app.config['MAIL_USERNAME'] or not app.config['MAIL_PASSWORD']:
+                raise ValueError("Mail configuration missing")
             pdf_path = f"/tmp/report_{scan.id}.pdf"
             generate_pdf(report, pdf_path)
             msg = Message(
@@ -511,7 +516,7 @@ def predict():
         return render_template("form.html", history_plot="/training_plot.png", result={
             "prediction": "Error",
             "confidence": "N/A",
-            "message": str(e),
+            "message": f"Prediction failed: {str(e)}",
             "email_status": "Error occurred, no email sent."
         })
 
@@ -525,7 +530,7 @@ def result():
         return render_template("form.html", history_plot="/training_plot.png", result={
             "prediction": "Error",
             "confidence": "N/A",
-            "message": "Failed to load result. Please try again.",
+            "message": f"Failed to load result: {str(e)}",
             "email_status": "N/A"
         })