Update app.py

app.py

@@ -10,6 +10,7 @@ import io
 import gc
 import librosa
 import soundfile as sf
+from datetime import datetime
 
 # ==========================================
 # 1. CONFIGURATION
@@ -18,42 +19,42 @@ MODELS = {
     "lungs": {
         "type": "image",
         "id": "nickmuchi/vit-finetuned-chest-xray-pneumonia",
-        "desc": "Tuberculosis & Pneumonia (Chest X-Ray)",
+        "desc": "Chest X-Ray Analysis",
         "safe": ["NORMAL", "normal", "No Pneumonia"],
         "rules": {"max_sat": 30, "reject_msg": "Invalid: Too colorful. Upload B&W X-Ray."}
     },
     "cough": { 
         "type": "audio",
         "id": "MIT/ast-finetuned-audioset-10-10-0.4593",
-        "desc": "COPD & Respiratory Screening (Cough Audio)",
+        "desc": "Respiratory Audio Analysis",
         "target_labels": ["Cough", "Throat clearing", "Respiratory sounds", "Wheeze"],
         "rules": {"min_duration": 0.5, "reject_msg": "Invalid: Audio too short or silent."}
     },
     "fracture": {
         "type": "image",
         "id": "dima806/bone_fracture_detection",
-        "desc": "Bone Fracture Detection (X-Ray)",
+        "desc": "Bone Trauma X-Ray",
         "safe": ["normal", "healed"],
         "rules": {"max_sat": 30, "reject_msg": "Invalid: Too colorful. Upload B&W X-Ray."}
     },
     "brain": {
         "type": "image",
         "id": "Hemgg/brain-tumor-classification",
-        "desc": "Brain Tumor & Stroke Screening (MRI/CT)",
+        "desc": "Brain MRI Scan Analysis",
         "safe": ["no_tumor"],
         "rules": {"max_sat": 30, "reject_msg": "Invalid: Too colorful. Upload B&W MRI Scan."}
     },
     "eye": {
         "type": "image",
         "id": "AventIQ-AI/resnet18-cataract-detection-system", 
-        "desc": "Cataract Detection (Eye Photo)",
+        "desc": "Ophthalmology Scan",
         "safe": ["Normal", "normal", "healthy"],
         "rules": {"min_sat": 20, "min_white": 0.05, "reject_msg": "Invalid: No eye detected."}
     },
     "skin": {
         "type": "image",
         "id": "Anwarkh1/Skin_Cancer-Image_Classification",
-        "desc": "Dermatology Analysis (Lesion Photo)",
+        "desc": "Dermatology Lesion Scan",
         "safe": ["Benign", "benign", "nv", "bkl"],
         "rules": {"min_sat": 20, "max_white": 0.25, "reject_msg": "Invalid: Not a skin close-up."}
     }
@@ -79,7 +80,6 @@ class MedicalEngine:
         stat = ImageStat.Stat(img_hsv)
         avg_sat = stat.mean[1]
         
-        # Forensics
         img_np = np.array(img_hsv)
         white_pixels = np.logical_and(img_np[:,:,1] < 40, img_np[:,:,2] > 180)
         white_ratio = np.sum(white_pixels) / white_pixels.size
@@ -162,11 +162,39 @@ class MedicalEngine:
 app = FastAPI()
 engine = MedicalEngine()
 
+# IN-MEMORY HISTORY STORAGE
+HISTORY = []
+
 @app.post("/predict/{task}")
-async def predict_route(task: str, file: UploadFile = File(...)):
+async def predict_route(task: str, patient: str, age: str, file: UploadFile = File(...)):
     if task not in MODELS: return {"error": "Invalid Task"}
+    
     content = await file.read()
-    return engine.predict(content, task)
+    result = engine.predict(content, task)
+    
+    # Save to History if valid
+    if "error" not in result and result.get("risk") != "INVALID":
+        record = {
+            "time": datetime.now().strftime("%H:%M:%S"),
+            "patient": patient,
+            "age": age,
+            "task": task.capitalize(),
+            "diagnosis": result["prediction"]["label"],
+            "risk": result["risk"]
+        }
+        HISTORY.insert(0, record) # Add to top
+    
+    return result
+
+@app.get("/history")
+def get_history():
+    return HISTORY
+
+@app.post("/reset_history")
+def reset_history():
+    global HISTORY
+    HISTORY = []
+    return {"status": "cleared"}
 
 @app.get("/", response_class=HTMLResponse)
 def home():
@@ -194,20 +222,13 @@ def home():
                     </p>
                 </div>
             </div>
-            
             <div class="flex flex-col items-end gap-2">
                 <select id="lang-select" onchange="changeLanguage()" class="bg-gray-100 border border-gray-300 text-gray-700 text-sm rounded-lg focus:ring-blue-500 focus:border-blue-500 block p-2">
                     <option value="en">🇬🇧 English</option>
-                    <option value="hi">🇮🇳 हिंदी (Hindi)</option>
-                    <option value="as">🇮🇳 অসমীয়া (Assamese)</option>
+                    <option value="hi">🇮🇳 हिंदी</option>
                     <option value="kha">🌲 Khasi</option>
                     <option value="gar">⛰️ Garo</option>
                 </select>
-                
-                <div class="hidden md:flex items-center gap-2 bg-blue-50 px-3 py-1 rounded text-xs font-bold border border-blue-100 text-blue-800">
-                    <img src="https://upload.wikimedia.org/wikipedia/en/c/cf/Aadhaar_Logo.svg" class="h-4">
-                    <span data-translate="abha_link">ABHA Linked</span>
-                </div>
             </div>
         </div>
     </nav>
@@ -242,8 +263,13 @@ def home():
         </div>
 
         <div class="bg-white rounded-2xl shadow-xl p-6 border border-gray-100">
-            <h2 id="header-text" class="text-xl font-bold text-gray-700 mb-6 text-center border-b pb-4">Select a Category</h2>
+            <h2 id="header-text" class="text-xl font-bold text-gray-700 mb-2 text-center">Select a Category</h2>
             
+            <div id="scope-box" class="hidden mb-6 text-center">
+                <p class="text-xs font-bold text-gray-400 uppercase mb-2" data-translate="lbl_scope">Scope of Detection</p>
+                <div id="scope-tags" class="flex flex-wrap justify-center gap-2"></div>
+            </div>
+
             <div id="inputs" class="opacity-50 pointer-events-none transition-opacity mb-6">
                 <div class="grid grid-cols-2 gap-4 mb-4">
                     <div>
@@ -252,18 +278,16 @@ def home():
                     </div>
                     <div>
                         <label class="text-xs font-bold text-gray-400 uppercase" data-translate="lbl_age">Age / ID</label>
-                        <input type="text" class="w-full border p-2 rounded bg-gray-50 outline-none focus:border-blue-500">
+                        <input type="text" id="p-age" class="w-full border p-2 rounded bg-gray-50 outline-none focus:border-blue-500">
                     </div>
                 </div>
                 
                 <div onclick="document.getElementById('file-input').click()" class="border-2 border-dashed border-gray-300 rounded-xl p-8 text-center cursor-pointer hover:bg-blue-50 transition group">
                     <input type="file" id="file-input" class="hidden" onchange="showPreview(event)" onclick="this.value=null">
-                    
                     <div id="placeholder" class="group-hover:scale-105 transition-transform">
                         <i id="upload-icon" class="fas fa-cloud-upload-alt text-4xl text-gray-400 mb-2"></i>
                         <p id="upload-text" class="text-gray-500 font-medium" data-translate="txt_upload">Tap to upload</p>
                     </div>
-                    
                     <img id="img-preview" class="hidden mx-auto max-h-56 rounded shadow object-contain">
                     <audio id="audio-preview" controls class="hidden w-full mt-4"></audio>
                 </div>
@@ -295,18 +319,43 @@ def home():
                         <span id="alert-text">--</span>
                     </div>
                 </div>
-
-                <div class="mt-4 flex items-center justify-between text-xs text-gray-500 border-t pt-2">
-                    <span class="font-bold flex items-center gap-1"><i class="fas fa-database"></i> <span data-translate="lbl_sync">Govt Sync</span></span>
-                    <span id="sync-msg" class="text-yellow-600"><i class="fas fa-sync fa-spin"></i> Pending...</span>
-                </div>
             </div>
+        </div>
 
+        <div class="mt-8">
+            <div class="flex justify-between items-center mb-4">
+                <h3 class="text-lg font-bold text-gray-700 uppercase tracking-wide"><i class="fas fa-history mr-2"></i> Recent Patients</h3>
+                <button onclick="clearHistory()" class="text-xs text-red-500 hover:text-red-700 font-bold border border-red-200 px-3 py-1 rounded hover:bg-red-50">CLEAR HISTORY</button>
+            </div>
+            <div class="bg-white rounded-xl shadow overflow-hidden border border-gray-100">
+                <table class="w-full text-sm text-left">
+                    <thead class="bg-gray-50 text-gray-500 font-bold uppercase text-xs">
+                        <tr>
+                            <th class="px-6 py-3">Time</th>
+                            <th class="px-6 py-3">Patient</th>
+                            <th class="px-6 py-3">Category</th>
+                            <th class="px-6 py-3">Diagnosis</th>
+                            <th class="px-6 py-3">Risk</th>
+                        </tr>
+                    </thead>
+                    <tbody id="history-table" class="divide-y divide-gray-100">
+                        <tr class="text-gray-400 text-center"><td colspan="5" class="py-4">No records yet.</td></tr>
+                    </tbody>
+                </table>
+            </div>
         </div>
     </div>
 
     <script>
-        // TRANSLATION DICTIONARY
+        const MODEL_SCOPES = {
+            lungs: ["Pneumonia", "Tuberculosis", "Viral Infection", "Normal Lung"],
+            cough: ["COPD Signs", "Whooping Cough", "Wheezing", "Respiratory Infection"],
+            fracture: ["Bone Fracture", "Dislocation", "Healthy Bone Structure"],
+            brain: ["Glioma Tumor", "Meningioma Tumor", "Pituitary Tumor", "No Tumor"],
+            eye: ["Cataract", "Diabetic Retinopathy", "Glaucoma", "Normal Eye"],
+            skin: ["Melanoma", "Basal Cell Carcinoma", "Nevus (Mole)", "Benign Keratosis"]
+        };
+
         const TRANSLATIONS = {
             en: {
                 govt_title: "GOVERNMENT OF MEGHALAYA",
@@ -318,7 +367,7 @@ def home():
                 btn_run: "Run Diagnosis",
                 txt_analyzing: "Downloading AI Model & Analyzing...",
                 lbl_result: "Analysis Result", lbl_conf: "Confidence",
-                lbl_action: "Action Required", lbl_sync: "Govt Sync"
+                lbl_action: "Action Required", lbl_scope: "Scope of Detection"
             },
             hi: {
                 govt_title: "मेघालय सरकार",
@@ -330,21 +379,9 @@ def home():
                 btn_run: "जांच शुरू करें",
                 txt_analyzing: "विश्लेषण कर रहा है...",
                 lbl_result: "परिणाम", lbl_conf: "विश्वास स्तर",
-                lbl_action: "आवश्यक कार्रवाई", lbl_sync: "सरकारी सिंक"
-            },
-            as: {
-                govt_title: "মেঘালয় চৰকাৰ",
-                online_status: "অনলাইন ন'ড: শ্বিলং",
-                abha_link: "ABHA সংযুক্ত",
-                btn_lungs: "হাঁওফাঁও", btn_cough: "কাহ", btn_bone: "হাৰ ভঙা", btn_brain: "মগজু", btn_eye: "চকু", btn_skin: "ছাল",
-                lbl_name: "ৰোগীৰ নাম", lbl_age: "বয়স / আইডি",
-                txt_upload: "ফটো আপলোড কৰক",
-                btn_run: "পৰীক্ষা কৰক",
-                txt_analyzing: "বিশ্লেষণ চলি আছে...",
-                lbl_result: "ফলাফল", lbl_conf: "নিশ্চয়তা",
-                lbl_action: "পৰৱৰ্তী পদক্ষেপ", lbl_sync: "চৰকাৰী ডাটাবেচ"
+                lbl_action: "आवश्यक कार्रवाई", lbl_scope: "जांच का दायरा"
             },
-            kha: { // Khasi Language
+            kha: {
                 govt_title: "SORKAR MEGHALAYA",
                 online_status: "Online: Shillong",
                 abha_link: "ABHA Link",
@@ -354,9 +391,9 @@ def home():
                 btn_run: "Leh Test",
                 txt_analyzing: "Dang Check...",
                 lbl_result: "Result", lbl_conf: "Jingshisha",
-                lbl_action: "Leh ia kane", lbl_sync: "Sorkar Sync"
+                lbl_action: "Leh ia kane", lbl_scope: "Jingshem Jong Ka Test"
             },
-            gar: { // Garo Language
+            gar: {
                 govt_title: "MEGHALAYA SORKARI",
                 online_status: "Online: Shillong",
                 abha_link: "ABHA Link",
@@ -366,7 +403,7 @@ def home():
                 btn_run: "Porikka Ka'bo",
                 txt_analyzing: "Niyenga...",
                 lbl_result: "Result", lbl_conf: "Bebera'ani",
-                lbl_action: "Nangchongmotgipa Kam", lbl_sync: "Sorkar Sync"
+                lbl_action: "Nangchongmotgipa Kam", lbl_scope: "Am·sandiani"
             }
         };
 
@@ -375,16 +412,16 @@ def home():
         let currFile = null;
         let currLang = 'en';
 
+        // Load History on Start
+        updateHistoryTable();
+
         function changeLanguage() {
             currLang = document.getElementById('lang-select').value;
             let t = TRANSLATIONS[currLang];
-            
             document.querySelectorAll('[data-translate]').forEach(el => {
                 let key = el.getAttribute('data-translate');
                 if (t[key]) el.innerText = t[key];
             });
-            
-            // Refresh Header Text if task selected
             if (currTask) {
                 let taskKey = "btn_" + currTask;
                 document.getElementById('header-text').innerHTML = `Upload <span class="uppercase text-blue-600">${t[taskKey]}</span>`;
@@ -396,16 +433,24 @@ def home():
             currType = type;
             let t = TRANSLATIONS[currLang];
             
-            // Buttons
             document.querySelectorAll('button[id^="btn-"]').forEach(b => b.classList.remove('ring-2', 'ring-blue-400', 'border-blue-500', 'bg-blue-50'));
             let btn = document.getElementById('btn-'+task);
             btn.classList.add('ring-2', 'ring-blue-400', 'border-blue-500', 'bg-blue-50');
             
-            // Text
             let taskName = t["btn_" + task];
             document.getElementById('header-text').innerHTML = `Upload <span class="uppercase text-blue-600">${taskName}</span>`;
-            
-            // Input Logic
+
+            let scopeBox = document.getElementById('scope-box');
+            let scopeTags = document.getElementById('scope-tags');
+            scopeTags.innerHTML = "";
+            MODEL_SCOPES[task].forEach(disease => {
+                let tag = document.createElement("span");
+                tag.className = "px-2 py-1 bg-blue-50 text-blue-800 text-xs rounded border border-blue-100 font-semibold";
+                tag.innerText = disease;
+                scopeTags.appendChild(tag);
+            });
+            scopeBox.classList.remove('hidden');
+
             let input = document.getElementById('file-input');
             let icon = document.getElementById('upload-icon');
             let text = document.getElementById('upload-text');
@@ -419,7 +464,6 @@ def home():
             }
             text.innerText = t['txt_upload'];
 
-            // Reset UI
             document.getElementById('inputs').classList.remove('opacity-50', 'pointer-events-none');
             document.getElementById('result-box').classList.add('hidden');
             document.getElementById('run-btn').classList.add('hidden');
@@ -433,7 +477,6 @@ def home():
             if (event.target.files && event.target.files[0]) {
                 currFile = event.target.files[0];
                 let url = URL.createObjectURL(currFile);
-                
                 if (currType === 'audio') {
                     let aud = document.getElementById('audio-preview');
                     aud.src = url;
@@ -445,7 +488,6 @@ def home():
                     img.classList.remove('hidden');
                     document.getElementById('audio-preview').classList.add('hidden');
                 }
-                
                 document.getElementById('placeholder').classList.add('hidden');
                 document.getElementById('run-btn').classList.remove('hidden');
                 document.getElementById('result-box').classList.add('hidden');
@@ -454,7 +496,9 @@ def home():
 
         async function analyze() {
             if (!currTask || !currFile) return;
-            if (!document.getElementById('p-name').value) { alert("Please enter Patient Name."); return; }
+            let pname = document.getElementById('p-name').value;
+            let page = document.getElementById('p-age').value;
+            if (!pname) { alert("Please enter Patient Name."); return; }
 
             document.getElementById('run-btn').classList.add('hidden');
             document.getElementById('loader').classList.remove('hidden');
@@ -463,12 +507,17 @@ def home():
             let formData = new FormData();
             formData.append("file", currFile);
 
+            // Pass patient details to backend to save history
             try {
-                let res = await fetch("/predict/" + currTask, { method: "POST", body: formData });
+                let url = `/predict/${currTask}?patient=${encodeURIComponent(pname)}&age=${encodeURIComponent(page)}`;
+                let res = await fetch(url, { method: "POST", body: formData });
                 let data = await res.json();
 
                 document.getElementById('loader').classList.add('hidden');
                 document.getElementById('result-box').classList.remove('hidden');
+                
+                // Refresh History Table
+                updateHistoryTable();
 
                 if (data.risk === "INVALID") {
                     document.getElementById('res-label').innerText = "Rejected";
@@ -494,11 +543,6 @@ def home():
                     document.getElementById('alert-box').classList.add('hidden');
                 }
 
-                setTimeout(() => {
-                    document.getElementById('sync-msg').innerHTML = "<i class='fas fa-check-circle'></i> Synced!";
-                    document.getElementById('sync-msg').className = "text-green-600 font-bold";
-                }, 2000);
-
             } catch (e) {
                 alert("Connection Failed.");
                 console.error(e);
@@ -507,6 +551,41 @@ def home():
             }
         }
 
+        async function updateHistoryTable() {
+            try {
+                let res = await fetch("/history");
+                let data = await res.json();
+                let tbody = document.getElementById('history-table');
+                tbody.innerHTML = "";
+                
+                if(data.length === 0) {
+                    tbody.innerHTML = '<tr class="text-gray-400 text-center"><td colspan="5" class="py-4">No records yet.</td></tr>';
+                    return;
+                }
+
+                data.forEach(row => {
+                    let color = row.risk === "HIGH" ? "text-red-600 font-bold" : row.risk === "MODERATE" ? "text-yellow-600 font-bold" : "text-green-600";
+                    let tr = `
+                        <tr class="bg-white border-b hover:bg-gray-50">
+                            <td class="px-6 py-4 text-gray-500">${row.time}</td>
+                            <td class="px-6 py-4 font-bold text-gray-800">${row.patient}</td>
+                            <td class="px-6 py-4 text-gray-600">${row.task}</td>
+                            <td class="px-6 py-4 text-gray-800">${row.diagnosis}</td>
+                            <td class="px-6 py-4 ${color}">${row.risk}</td>
+                        </tr>
+                    `;
+                    tbody.innerHTML += tr;
+                });
+            } catch(e) { console.error("History fetch failed"); }
+        }
+
+        async function clearHistory() {
+            if(confirm("Delete all history?")) {
+                await fetch("/reset_history", { method: "POST" });
+                updateHistoryTable();
+            }
+        }
+
         function updateBadge(text, bg, color) {
             let b = document.getElementById('res-badge');
             b.innerText = text;