Upload 11 files

app.py

@@ -0,0 +1,193 @@
+from flask import Flask, render_template, request, flash, redirect, url_for
+import numpy as np
+import pickle
+
+app = Flask(__name__)
+app.secret_key = 'your_secret_key_here'  # Replace with a strong secret key
+
+def predict_disease(patient_data):
+    """
+    Predicts disease risks based on patient data.
+    Loads pre-trained models and scalers, prepares features, and returns a risk dictionary.
+    """
+    try:
+        # Load models
+        heart_model = pickle.load(open('heart_rf_model.pkl', 'rb'))
+        diabetes_model = pickle.load(open('diabetes_model.pkl', 'rb'))
+        cirrhosis_model = pickle.load(open('cirrhosis_model.pkl', 'rb'))
+        hep_c_model = pickle.load(open('hep_c_model.pkl', 'rb'))
+
+        # Load scalers
+        heart_scaler = pickle.load(open('heart_scaler.pkl', 'rb'))
+        diabetes_scaler = pickle.load(open('diabetes_scaler.pkl', 'rb'))
+        cirrhosis_scaler = pickle.load(open('cirrhosis_scaler.pkl', 'rb'))
+        hep_c_scaler = pickle.load(open('hep_c_scaler.pkl', 'rb'))
+
+        # Heart Disease Features
+        heart_features = np.array([[
+            patient_data.get('Age', 55),
+            patient_data.get('Sex', 1),
+            patient_data.get('cp', 0),
+            patient_data.get('BP', 130),
+            patient_data.get('Cholesterol', 200),
+            patient_data.get('FBS', 0),
+            patient_data.get('EKG', 0),
+            patient_data.get('MaxHR', 150),
+            patient_data.get('ExerciseAngina', 0),
+            patient_data.get('STdepression', 0.0),
+            patient_data.get('STslope', 0),
+            patient_data.get('Vessels', 0),
+            patient_data.get('Thallium', 2)
+        ]])
+
+        # Diabetes Features – only scaling Age
+        diabetes_features = np.array([[
+            1 if patient_data.get('Polyuria', 0) == 1 else 0,
+            1 if patient_data.get('Polydipsia', 0) == 1 else 0,
+            patient_data.get('Age', 55),  # Age will be scaled
+            1 if patient_data.get('Gender', 'Male') == 'Male' else 0,
+            1 if patient_data.get('partial_paresis', 0) == 1 else 0,
+            1 if patient_data.get('sudden_weight_loss', 0) == 1 else 0,
+            1 if patient_data.get('Irritability', 0) == 1 else 0,
+            1 if patient_data.get('delayed_healing', 0) == 1 else 0,
+            1 if patient_data.get('Alopecia', 0) == 1 else 0,
+            1 if patient_data.get('Itching', 0) == 1 else 0
+        ]])
+        # Scale Age for diabetes
+        age_scaled = diabetes_scaler.transform([[patient_data.get('Age', 55)]])
+        diabetes_features[0, 2] = age_scaled[0, 0]
+
+        # Cirrhosis Features
+        cirrhosis_features = np.array([[
+            patient_data.get('Bilirubin', 1.2),
+            patient_data.get('Albumin', 3.8),
+            patient_data.get('Copper', 80),
+            patient_data.get('Alk_Phos', 70),
+            patient_data.get('SGOT', 40),
+            patient_data.get('Tryglicerides', 150),
+            patient_data.get('Platelets', 250),
+            patient_data.get('Prothrombin', 11),
+            patient_data.get('Stage', 1),
+            patient_data.get('Age', 55),
+            patient_data.get('Sex', 1),
+            patient_data.get('Ascites', 0),
+            patient_data.get('Hepatomegaly', 0),
+            patient_data.get('Spiders', 0),
+            patient_data.get('Edema', 0)
+        ]])
+
+        # Hepatitis C Features
+        hep_c_features = np.array([[
+            patient_data.get('Age', 55),
+            patient_data.get('Sex', 1),
+            patient_data.get('ALB', 4.0),
+            patient_data.get('ALP', 70),
+            patient_data.get('ALT', 45),
+            patient_data.get('AST', 38),
+            patient_data.get('BIL', 0.8),
+            patient_data.get('CHE', 8000),
+            patient_data.get('CHOL', 180),
+            patient_data.get('CREA', 0.9),
+            patient_data.get('GGT', 30),
+            patient_data.get('PROT', 7.0)
+        ]])
+
+        # Scale features
+        heart_scaled = heart_scaler.transform(heart_features)
+        cirrhosis_scaled = cirrhosis_scaler.transform(cirrhosis_features)
+        hep_c_scaled = hep_c_scaler.transform(hep_c_features)
+
+        # Get prediction probabilities
+        heart_prob = heart_model.predict_proba(heart_scaled)[:, 1][0]
+        diabetes_prob = diabetes_model.predict_proba(diabetes_features)[:, 1][0]
+        cirrhosis_prob = cirrhosis_model.predict_proba(cirrhosis_scaled)[:, 1][0]
+        hep_c_prob = hep_c_model.predict_proba(hep_c_scaled)[:, 1][0]
+
+        # Compute overall risk score
+        final_score = (
+            (0.30 * heart_prob) +
+            (0.25 * diabetes_prob) +
+            (0.25 * cirrhosis_prob) +
+            (0.20 * hep_c_prob)
+        )
+
+        return {
+            'Heart Disease': {'Risk': 'High' if heart_prob > 0.5 else 'Low', 'Probability': round(heart_prob, 3)},
+            'Diabetes': {'Risk': 'High' if diabetes_prob > 0.5 else 'Low', 'Probability': round(diabetes_prob, 3)},
+            'Cirrhosis': {'Risk': 'High' if cirrhosis_prob > 0.5 else 'Low', 'Probability': round(cirrhosis_prob, 3)},
+            'Hepatitis C': {'Risk': 'High' if hep_c_prob > 0.5 else 'Low', 'Probability': round(hep_c_prob, 3)},
+            'Overall Risk Score': round(final_score, 3)
+        }
+
+    except Exception as e:
+        raise Exception(f"Error in prediction: {str(e)}")
+
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    result = None
+    if request.method == 'POST':
+        try:
+            # Collect input parameters from the form
+            patient_data = {
+                # General / Heart Disease
+                'Age': int(request.form.get('Age', 55)),
+                'Sex': int(request.form.get('Sex', 1)),
+                'cp': int(request.form.get('cp', 0)),
+                'BP': float(request.form.get('BP', 130)),
+                'Cholesterol': float(request.form.get('Cholesterol', 200)),
+                'FBS': int(request.form.get('FBS', 0)),
+                'EKG': int(request.form.get('EKG', 0)),
+                'MaxHR': int(request.form.get('MaxHR', 150)),
+                'ExerciseAngina': int(request.form.get('ExerciseAngina', 0)),
+                'STdepression': float(request.form.get('STdepression', 0.0)),
+                'STslope': int(request.form.get('STslope', 0)),
+                'Vessels': int(request.form.get('Vessels', 0)),
+                'Thallium': int(request.form.get('Thallium', 2)),
+
+                # Diabetes
+                'Polyuria': int(request.form.get('Polyuria', 0)),
+                'Polydipsia': int(request.form.get('Polydipsia', 0)),
+                'Gender': request.form.get('Gender', 'Male'),
+                'partial_paresis': int(request.form.get('partial_paresis', 0)),
+                'sudden_weight_loss': int(request.form.get('sudden_weight_loss', 0)),
+                'Irritability': int(request.form.get('Irritability', 0)),
+                'delayed_healing': int(request.form.get('delayed_healing', 0)),
+                'Alopecia': int(request.form.get('Alopecia', 0)),
+                'Itching': int(request.form.get('Itching', 0)),
+
+                # Cirrhosis
+                'Bilirubin': float(request.form.get('Bilirubin', 1.2)),
+                'Albumin': float(request.form.get('Albumin', 3.8)),
+                'Copper': float(request.form.get('Copper', 80)),
+                'Alk_Phos': float(request.form.get('Alk_Phos', 70)),
+                'SGOT': float(request.form.get('SGOT', 40)),
+                'Tryglicerides': float(request.form.get('Tryglicerides', 150)),
+                'Platelets': float(request.form.get('Platelets', 250)),
+                'Prothrombin': float(request.form.get('Prothrombin', 11)),
+                'Stage': int(request.form.get('Stage', 1)),
+                'Ascites': int(request.form.get('Ascites', 0)),
+                'Hepatomegaly': int(request.form.get('Hepatomegaly', 0)),
+                'Spiders': int(request.form.get('Spiders', 0)),
+                'Edema': int(request.form.get('Edema', 0)),
+
+                # Hepatitis C
+                'ALB': float(request.form.get('ALB', 4.0)),
+                'ALP': float(request.form.get('ALP', 70)),
+                'ALT': float(request.form.get('ALT', 45)),
+                'AST': float(request.form.get('AST', 38)),
+                'BIL': float(request.form.get('BIL_hep', 0.8)),  # To distinguish from cirrhosis bilirubin
+                'CHE': float(request.form.get('CHE', 8000)),
+                'CHOL': float(request.form.get('CHOL_hep', 180)),  # To distinguish from heart cholesterol
+                'CREA': float(request.form.get('CREA', 0.9)),
+                'GGT': float(request.form.get('GGT', 30)),
+                'PROT': float(request.form.get('PROT_hep', 7.0))
+            }
+
+            # Get the prediction result
+            result = predict_disease(patient_data)
+        except Exception as e:
+            flash(str(e))
+    return render_template('index.html', result=result)
+
+if __name__ == '__main__':
+    app.run(debug=True, port=1234)

chd_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bda2cf8d2bbb1a2de14c7f4df6227b04be6085fe959a9a46c098021079d8d0e2
+size 620304

cirrhosis_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f41586619726233b6a5828ee585e34defb55f8bbc29462b79fbb5583d310edad
+size 50654

cirrhosis_scaler.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:454799941712d0d757bbcdd9df04c2ce083195c864a8df48f74f3f9b076d2893
+size 1046

diabetes_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:13abf699fafcd7194b1e2695b23fba47fbbdbf311f998adeff06f6f3d1a0ee8f
+size 669962

diabetes_scaler.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2aa8f1134492222d92a98426e73c1be151969dfd6c3291f0ab8443bdd76d239
+size 613

heart_rf_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bda2cf8d2bbb1a2de14c7f4df6227b04be6085fe959a9a46c098021079d8d0e2
+size 620304

heart_scaler.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87121f601efd875d54937ed7abe5af3666750c14a1dcf5307c705bd2106924ab
+size 1020

hep_c_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b93332e20437a96e7f00969c4763ee24ce6e8c0de1aef91aebbc0e04480f5f27
+size 356270

hep_c_scaler.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1d6bfc3f1db929ce45cb91c81e5360c450a995d27ab6e3ef1f599b5b6620fb3
+size 899

templates/index.html

@@ -0,0 +1,248 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <meta charset="UTF-8">
+  <title>Disease Risk Prediction</title>
+  <style>
+    /* styles.css */
+
+body {
+  font-family: Arial, sans-serif;
+  margin: 20px;
+  padding: 0;
+  background-color: #f2f2f2;
+  color: #333;
+}
+
+.container {
+  max-width: 900px;
+  margin: 0 auto;
+  background: #fff;
+  padding: 20px;
+  border-radius: 8px;
+  box-shadow: 0 0 10px rgba(0,0,0,0.1);
+}
+
+h1 {
+  text-align: center;
+  margin-bottom: 20px;
+}
+
+form {
+  margin-bottom: 20px;
+}
+
+fieldset {
+  border: 1px solid #ccc;
+  padding: 15px;
+  margin-bottom: 20px;
+  border-radius: 5px;
+  background: #fafafa;
+}
+
+legend {
+  font-weight: bold;
+  padding: 0 10px;
+}
+
+label {
+  display: inline-block;
+  width: 250px;
+  margin-bottom: 10px;
+  vertical-align: top;
+}
+
+input[type="number"],
+input[type="text"] {
+  width: calc(100% - 260px);
+  padding: 8px;
+  margin-bottom: 10px;
+  border: 1px solid #ccc;
+  border-radius: 4px;
+}
+
+input[type="submit"] {
+  background-color: #4285f4;
+  color: #fff;
+  border: none;
+  padding: 10px 15px;
+  font-size: 16px;
+  border-radius: 5px;
+  cursor: pointer;
+  width: 100%;
+}
+
+input[type="submit"]:hover {
+  background-color: #357ae8;
+}
+
+.results {
+  background: #e9f5e9;
+  border: 1px solid #c3e6c3;
+  padding: 15px;
+  border-radius: 5px;
+}
+
+.results h2 {
+  margin-top: 0;
+}
+
+ul {
+  list-style: none;
+  padding: 0;
+}
+
+ul li {
+  margin-bottom: 10px;
+}
+
+  </style>
+</head>
+<body>
+  <h1>Disease Risk Prediction</h1>
+
+  {% with messages = get_flashed_messages() %}
+    {% if messages %}
+      <ul style="color: red;">
+        {% for message in messages %}
+          <li>{{ message }}</li>
+        {% endfor %}
+      </ul>
+    {% endif %}
+  {% endwith %}
+
+  <form method="POST">
+    <!-- General / Heart Disease Information -->
+    <fieldset>
+      <legend>General &amp; Heart Disease Parameters</legend>
+      <label for="Age">Age:</label>
+      <input type="number" name="Age" id="Age" value="55" required><br>
+      <label for="Sex">Sex (1=Male, 0=Female):</label>
+      <input type="number" name="Sex" id="Sex" value="1" min="0" max="1" required><br>
+      <label for="cp">Chest Pain Type (cp):</label>
+      <input type="number" name="cp" id="cp" value="0"><br>
+      <label for="BP">Blood Pressure (BP):</label>
+      <input type="number" name="BP" id="BP" value="130"><br>
+      <label for="Cholesterol">Cholesterol:</label>
+      <input type="number" name="Cholesterol" id="Cholesterol" value="200"><br>
+      <label for="FBS">Fasting Blood Sugar (FBS):</label>
+      <input type="number" name="FBS" id="FBS" value="0"><br>
+      <label for="EKG">EKG:</label>
+      <input type="number" name="EKG" id="EKG" value="0"><br>
+      <label for="MaxHR">Max Heart Rate (MaxHR):</label>
+      <input type="number" name="MaxHR" id="MaxHR" value="150"><br>
+      <label for="ExerciseAngina">Exercise Angina (1=Yes,0=No):</label>
+      <input type="number" name="ExerciseAngina" id="ExerciseAngina" value="0" min="0" max="1"><br>
+      <label for="STdepression">ST Depression:</label>
+      <input type="number" step="0.1" name="STdepression" id="STdepression" value="0.0"><br>
+      <label for="STslope">ST Slope:</label>
+      <input type="number" name="STslope" id="STslope" value="0"><br>
+      <label for="Vessels">Number of Vessels:</label>
+      <input type="number" name="Vessels" id="Vessels" value="0"><br>
+      <label for="Thallium">Thallium:</label>
+      <input type="number" name="Thallium" id="Thallium" value="2">
+    </fieldset>
+
+    <!-- Diabetes Parameters -->
+    <fieldset>
+      <legend>Diabetes Parameters</legend>
+      <label for="Polyuria">Polyuria (1=Yes,0=No):</label>
+      <input type="number" name="Polyuria" id="Polyuria" value="0" min="0" max="1"><br>
+      <label for="Polydipsia">Polydipsia (1=Yes,0=No):</label>
+      <input type="number" name="Polydipsia" id="Polydipsia" value="0" min="0" max="1"><br>
+      <label for="Gender">Gender (Male/Female):</label>
+      <input type="text" name="Gender" id="Gender" value="Male"><br>
+      <label for="partial_paresis">Partial Paresis (1=Yes,0=No):</label>
+      <input type="number" name="partial_paresis" id="partial_paresis" value="0" min="0" max="1"><br>
+      <label for="sudden_weight_loss">Sudden Weight Loss (1=Yes,0=No):</label>
+      <input type="number" name="sudden_weight_loss" id="sudden_weight_loss" value="0" min="0" max="1"><br>
+      <label for="Irritability">Irritability (1=Yes,0=No):</label>
+      <input type="number" name="Irritability" id="Irritability" value="0" min="0" max="1"><br>
+      <label for="delayed_healing">Delayed Healing (1=Yes,0=No):</label>
+      <input type="number" name="delayed_healing" id="delayed_healing" value="0" min="0" max="1"><br>
+      <label for="Alopecia">Alopecia (1=Yes,0=No):</label>
+      <input type="number" name="Alopecia" id="Alopecia" value="0" min="0" max="1"><br>
+      <label for="Itching">Itching (1=Yes,0=No):</label>
+      <input type="number" name="Itching" id="Itching" value="0" min="0" max="1">
+    </fieldset>
+
+    <!-- Cirrhosis Parameters -->
+    <fieldset>
+      <legend>Cirrhosis Parameters</legend>
+      <label for="Bilirubin">Bilirubin:</label>
+      <input type="number" step="0.1" name="Bilirubin" id="Bilirubin" value="1.2"><br>
+      <label for="Albumin">Albumin:</label>
+      <input type="number" step="0.1" name="Albumin" id="Albumin" value="3.8"><br>
+      <label for="Copper">Copper:</label>
+      <input type="number" name="Copper" id="Copper" value="80"><br>
+      <label for="Alk_Phos">Alkaline Phosphatase (Alk_Phos):</label>
+      <input type="number" name="Alk_Phos" id="Alk_Phos" value="70"><br>
+      <label for="SGOT">SGOT:</label>
+      <input type="number" name="SGOT" id="SGOT" value="40"><br>
+      <label for="Tryglicerides">Tryglicerides:</label>
+      <input type="number" name="Tryglicerides" id="Tryglicerides" value="150"><br>
+      <label for="Platelets">Platelets:</label>
+      <input type="number" name="Platelets" id="Platelets" value="250"><br>
+      <label for="Prothrombin">Prothrombin:</label>
+      <input type="number" step="0.1" name="Prothrombin" id="Prothrombin" value="11"><br>
+      <label for="Stage">Stage:</label>
+      <input type="number" name="Stage" id="Stage" value="1"><br>
+      <label for="Ascites">Ascites (1=Yes,0=No):</label>
+      <input type="number" name="Ascites" id="Ascites" value="0" min="0" max="1"><br>
+      <label for="Hepatomegaly">Hepatomegaly (1=Yes,0=No):</label>
+      <input type="number" name="Hepatomegaly" id="Hepatomegaly" value="0" min="0" max="1"><br>
+      <label for="Spiders">Spiders (1=Yes,0=No):</label>
+      <input type="number" name="Spiders" id="Spiders" value="0" min="0" max="1"><br>
+      <label for="Edema">Edema (1=Yes,0=No):</label>
+      <input type="number" name="Edema" id="Edema" value="0" min="0" max="1">
+    </fieldset>
+
+    <!-- Hepatitis C Parameters -->
+    <fieldset>
+      <legend>Hepatitis C Parameters</legend>
+      <label for="ALB">ALB:</label>
+      <input type="number" step="0.1" name="ALB" id="ALB" value="4.0"><br>
+      <label for="ALP">ALP:</label>
+      <input type="number" name="ALP" id="ALP" value="70"><br>
+      <label for="ALT">ALT:</label>
+      <input type="number" name="ALT" id="ALT" value="45"><br>
+      <label for="AST">AST:</label>
+      <input type="number" name="AST" id="AST" value="38"><br>
+      <label for="BIL_hep">BIL (Hepatitis C):</label>
+      <input type="number" step="0.1" name="BIL_hep" id="BIL_hep" value="0.8"><br>
+      <label for="CHE">CHE:</label>
+      <input type="number" name="CHE" id="CHE" value="8000"><br>
+      <label for="CHOL_hep">CHOL (Hepatitis C):</label>
+      <input type="number" name="CHOL_hep" id="CHOL_hep" value="180"><br>
+      <label for="CREA">CREA:</label>
+      <input type="number" step="0.1" name="CREA" id="CREA" value="0.9"><br>
+      <label for="GGT">GGT:</label>
+      <input type="number" name="GGT" id="GGT" value="30"><br>
+      <label for="PROT_hep">PROT (Hepatitis C):</label>
+      <input type="number" step="0.1" name="PROT_hep" id="PROT_hep" value="7.0">
+    </fieldset>
+
+    <input type="submit" value="Predict">
+  </form>
+
+  {% if result %}
+  <div class="results">
+    <h2>Prediction Results</h2>
+    <ul>
+      {% for disease, info in result.items() %}
+        {% if disease != 'Overall Risk Score' %}
+          <li>
+            <strong>{{ disease }}:</strong>
+            Risk Level: {{ info.Risk }},
+            Probability: {{ (info.Probability * 100)|round(2) }}%
+          </li>
+        {% else %}
+          <li><strong>{{ disease }}:</strong> {{ (info * 100)|round(2) }}%</li>
+        {% endif %}
+      {% endfor %}
+    </ul>
+  </div>
+  {% endif %}
+
+</body>
+</html>