Update app.py

app.py

@@ -41,7 +41,6 @@ cdss_data = {
     "Glaucoma": {"status": "Glaucoma Suspect", "triage": "REFERRAL", "advice": "Optic nerve damage detected.", "action": "Refer for IOP Test."}
 }
 
-# 2. XAI Heatmap Functions
 # 2. XAI Heatmap Functions (ACCURATE VERSION)
 def generate_robust_heatmap(img_array, model, class_index, patch_size=16, step=8):
     try:
@@ -91,6 +90,7 @@ def overlay_heatmap(img, heatmap, alpha=0.5):
     superimposed_img = tf.keras.preprocessing.image.img_to_array(jet_heatmap) * alpha + tf.keras.preprocessing.image.img_to_array(img)
     return tf.keras.preprocessing.image.array_to_img(superimposed_img)
 
+
 # 3. Routes
 @app.route('/')
 def login(): return render_template('index.html')
@@ -104,6 +104,16 @@ def upload(): return render_template('upload.html')
 @app.route('/result')
 def result(): return render_template('result.html')
 
+# DATABASE க்கான புது ரூட் (NEW ROUTE FOR DATABASE)
+@app.route('/database')
+def database():
+    conn = sqlite3.connect('reticheck_patients.db')
+    c = conn.cursor()
+    c.execute("SELECT * FROM history ORDER BY id DESC") # கடைசியாக செக் பண்ணியதை முதலில் காட்ட
+    records = c.fetchall()
+    conn.close()
+    return render_template('database.html', records=records)
+
 @app.route('/predict', methods=['POST'])
 def predict():
     file = request.files['image']
@@ -124,6 +134,21 @@ def predict():
     if "Diabetic" in class_name: disease = "Diabetic"
     elif "Glaucoma" in class_name: disease = "Glaucoma"
 
+    report = cdss_data[disease]
+
+    # --- DATABASE ல் சேமிப்பதற்கான கோடு (NEW CODE) ---
+    try:
+        conn = sqlite3.connect('reticheck_patients.db')
+        c = conn.cursor()
+        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        c.execute("INSERT INTO history (date, disease, confidence, status) VALUES (?, ?, ?, ?)",
+                  (current_time, class_name, f"{confidence:.1f}%", report['triage']))
+        conn.commit()
+        conn.close()
+    except Exception as db_error:
+        print(f"Database Save Error: {db_error}")
+    # ------------------------------------------------
+
     # Convert Original Image to Base64
     buf = BytesIO()
     image_resized.save(buf, format="JPEG")
@@ -141,8 +166,6 @@ def predict():
             heat_b64 = base64.b64encode(buf2.getvalue()).decode('utf-8')
             heatmap_generated = True
 
-    report = cdss_data[disease]
-
     return jsonify({
         'disease': class_name,
         'confidence': f"{confidence:.1f}%",