Update app.py

app.py

@@ -84,7 +84,7 @@ def predict():
             'age': float(input_data.get('age', 0)),
             'gender': input_data.get('gender', 'Unknown'),
             'ever_married': input_data.get('ever_married', 'No'),
-            'residence_type': input_data.get('residence_type', 'Unknown'),  # Matches form
+            'residence_type': input_data.get('residence_type', 'Unknown'),
             'work_type': input_data.get('work_type', 'Unknown'),
             'hypertension': int(input_data.get('hypertension', 0)),
             'heart_disease': int(input_data.get('heart_disease', 0)),
@@ -118,13 +118,13 @@ def predict():
 
         # Make prediction
         probability = model.predict_proba(df)[0][1] * 100
-        risk_level = 'High' if probability > 50 else 'Moderate' if probability > 25 else 'Low'
+        risk_prediction = "Stroke Risk" if probability > 50 else "No Stroke Risk"
 
         # Determine contributing factors
         contributing_factors = {
             'glucose': features['avg_glucose_level'] > 140,
             'hypertension': features['hypertension'] == 1,
-            'heartDisease': features['heart_disease'] == 1,  # Matches index.html
+            'heartDisease': features['heart_disease'] == 1,
             'smoking': features['smoking_status'] in ['smokes', 'formerly smoked']
         }
 
@@ -150,16 +150,16 @@ def predict():
             features['bmi'],
             features['smoking_status'],
             round(probability),
-            risk_level
+            risk_prediction
         ))
         conn.commit()
         conn.close()
 
-        print(f"Prediction success: probability={probability}%, risk_level={risk_level}")
+        print(f"Prediction success: probability={probability}%, prediction={risk_prediction}")
         return jsonify({
             'success': True,
+            'prediction': risk_prediction,  # Simplified output
             'probability': round(probability),
-            'riskLevel': risk_level,
             'contributingFactors': contributing_factors
         })
     except Exception as e: