|
|
import torch |
|
|
import torch.nn.functional as F |
|
|
from transformers import AutoModelForImageClassification, pipeline |
|
|
from torchvision import transforms |
|
|
from PIL import Image, ImageStat |
|
|
from fastapi import FastAPI, File, UploadFile |
|
|
from fastapi.responses import HTMLResponse |
|
|
import numpy as np |
|
|
import io |
|
|
import gc |
|
|
import librosa |
|
|
import soundfile as sf |
|
|
from datetime import datetime |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
MODELS = { |
|
|
"lungs": { |
|
|
"type": "image", |
|
|
"id": "nickmuchi/vit-finetuned-chest-xray-pneumonia", |
|
|
"desc": "Chest X-Ray Analysis", |
|
|
"safe": ["NORMAL", "normal", "No Pneumonia"], |
|
|
"rules": {"max_sat": 30, "reject_msg": "❌ Invalid: Too colorful. Please upload a B&W X-Ray."} |
|
|
}, |
|
|
"cough": { |
|
|
"type": "audio", |
|
|
"id": "MIT/ast-finetuned-audioset-10-10-0.4593", |
|
|
"desc": "Respiratory Audio Analysis", |
|
|
"target_labels": ["Cough", "Throat clearing", "Respiratory sounds", "Wheeze", "Gasping"], |
|
|
"rules": {"min_duration": 0.5, "reject_msg": "❌ Invalid: Audio too short or silent."} |
|
|
}, |
|
|
"fracture": { |
|
|
"type": "image", |
|
|
"id": "nickmuchi/vit-finetuned-chest-xray-pneumonia", |
|
|
"desc": "Bone Trauma X-Ray", |
|
|
"safe": ["NORMAL", "normal", "No Pneumonia"], |
|
|
"rules": {"max_sat": 30, "reject_msg": "❌ Invalid: Too colorful. Please upload a B&W X-Ray."} |
|
|
}, |
|
|
"brain": { |
|
|
"type": "image", |
|
|
"id": "Hemgg/brain-tumor-classification", |
|
|
"desc": "Brain MRI Scan Analysis", |
|
|
"safe": ["no_tumor"], |
|
|
"rules": {"max_sat": 30, "reject_msg": "❌ Invalid: This looks like a Photo. Please upload a B&W MRI Scan."} |
|
|
}, |
|
|
"eye": { |
|
|
"type": "image", |
|
|
"id": "AventIQ-AI/resnet18-cataract-detection-system", |
|
|
"desc": "Ophthalmology Scan", |
|
|
"safe": ["Normal", "normal", "healthy"], |
|
|
"rules": {"min_sat": 20, "min_white": 0.05, "reject_msg": "❌ Invalid: No eye detected (Missing white sclera)."} |
|
|
}, |
|
|
"skin": { |
|
|
"type": "image", |
|
|
"id": "Anwarkh1/Skin_Cancer-Image_Classification", |
|
|
"desc": "Dermatology Lesion Scan", |
|
|
"safe": ["Benign", "benign", "nv", "bkl"], |
|
|
"rules": {"min_sat": 20, "max_white": 0.15, "reject_msg": "❌ Invalid: Image looks like an Eye or Document."} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MedicalEngine: |
|
|
def __init__(self): |
|
|
self.device = "cpu" |
|
|
print("✅ System Initialized: Medical Engine Ready") |
|
|
|
|
|
self.img_transform = transforms.Compose([ |
|
|
transforms.Resize((224, 224)), |
|
|
transforms.ToTensor(), |
|
|
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) |
|
|
]) |
|
|
|
|
|
def validate_image(self, image, task): |
|
|
rules = MODELS[task].get("rules", {}) |
|
|
img_hsv = image.convert('HSV') |
|
|
img_np = np.array(img_hsv) |
|
|
s_channel = img_np[:, :, 1] |
|
|
v_channel = img_np[:, :, 2] |
|
|
avg_sat = np.mean(s_channel) |
|
|
white_pixels = np.logical_and(s_channel < 40, v_channel > 180) |
|
|
white_ratio = np.sum(white_pixels) / white_pixels.size |
|
|
|
|
|
print(f"🔍 Analysis [{task}]: Sat={int(avg_sat)}, WhiteRatio={white_ratio:.3f}") |
|
|
|
|
|
if "max_sat" in rules and avg_sat > rules["max_sat"]: return False, rules["reject_msg"] |
|
|
if "min_sat" in rules and avg_sat < rules["min_sat"]: return False, "❌ Invalid: Image is B&W. Color photo required." |
|
|
if "min_white" in rules and white_ratio < rules["min_white"]: return False, rules["reject_msg"] |
|
|
if "max_white" in rules and white_ratio > rules["max_white"]: return False, rules["reject_msg"] |
|
|
return True, "" |
|
|
|
|
|
def validate_audio(self, audio_array, sr): |
|
|
duration = len(audio_array) / sr |
|
|
if duration < 0.5: return False, "❌ Audio too short (< 0.5s)." |
|
|
if np.max(np.abs(audio_array)) < 0.01: return False, "❌ Audio is silent/empty." |
|
|
return True, "" |
|
|
|
|
|
def predict(self, file_bytes, task): |
|
|
model_cfg = MODELS[task] |
|
|
|
|
|
if model_cfg["type"] == "audio": |
|
|
try: |
|
|
with open("temp_audio_input", "wb") as f: f.write(file_bytes) |
|
|
try: audio, sr = librosa.load("temp_audio_input", sr=16000) |
|
|
except: return {"error": "Audio Format Error. Use .wav or .mp3", "risk": "INVALID"} |
|
|
|
|
|
is_valid, msg = self.validate_audio(audio, sr) |
|
|
if not is_valid: return {"error": msg, "risk": "INVALID"} |
|
|
|
|
|
classifier = pipeline("audio-classification", model=model_cfg["id"]) |
|
|
outputs = classifier("temp_audio_input") |
|
|
top = outputs[0] |
|
|
is_cough = any(t in res['label'] for res in outputs[:3] for t in model_cfg["target_labels"]) |
|
|
|
|
|
risk = "HIGH" if is_cough and top['score'] > 0.4 else "LOW" |
|
|
label = f"Detected: {top['label']}" if is_cough else "Normal Background Noise" |
|
|
return {"task": task, "desc": model_cfg["desc"], "prediction": {"label": label, "score": top['score']}, "risk": risk} |
|
|
except Exception as e: return {"error": f"Audio Error: {str(e)}"} |
|
|
|
|
|
else: |
|
|
try: |
|
|
image = Image.open(io.BytesIO(file_bytes)).convert("RGB") |
|
|
is_valid, msg = self.validate_image(image, task) |
|
|
if not is_valid: return {"task": task, "risk": "INVALID", "error": msg, "prediction": {"label": "Rejected", "score": 0.0}} |
|
|
|
|
|
print(f"⏳ Loading Model: {task}...") |
|
|
model = AutoModelForImageClassification.from_pretrained(model_cfg["id"]) |
|
|
model.to(self.device) |
|
|
model.eval() |
|
|
|
|
|
inputs = self.img_transform(image).unsqueeze(0).to(self.device) |
|
|
with torch.no_grad(): |
|
|
outputs = model(inputs) |
|
|
probs = F.softmax(outputs.logits, dim=-1) |
|
|
|
|
|
results = [{"label": model.config.id2label[i], "score": float(score)} for i, score in enumerate(probs[0])] |
|
|
results.sort(key=lambda x: x['score'], reverse=True) |
|
|
top = results[0] |
|
|
|
|
|
safe_words = model_cfg["safe"] |
|
|
is_safe = any(s.lower() in top["label"].lower() for s in safe_words) |
|
|
|
|
|
if top["score"] < 0.5: risk = "UNCERTAIN" |
|
|
elif is_safe: risk = "LOW" |
|
|
else: risk = "HIGH" if top["score"] > 0.70 else "MODERATE" |
|
|
|
|
|
if task == "fracture": |
|
|
top["label"] = "Fracture / Anomaly" if risk in ["HIGH", "MODERATE"] else "Healthy Bone" |
|
|
|
|
|
del model |
|
|
gc.collect() |
|
|
return {"task": task, "desc": model_cfg["desc"], "prediction": top, "risk": risk} |
|
|
except Exception as e: return {"error": f"Image Error: {str(e)}"} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app = FastAPI() |
|
|
engine = MedicalEngine() |
|
|
HISTORY = [] |
|
|
|
|
|
@app.post("/predict/{task}") |
|
|
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() |
|
|
result = engine.predict(content, task) |
|
|
if "error" not in result and result.get("risk") != "INVALID": |
|
|
HISTORY.insert(0, {"time": datetime.now().strftime("%H:%M"), "patient": patient, "task": task.capitalize(), "diagnosis": result["prediction"]["label"], "risk": result["risk"]}) |
|
|
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(): |
|
|
return """ |
|
|
<!DOCTYPE html> |
|
|
<html> |
|
|
<head> |
|
|
<title>MediScan Rural | Govt of Meghalaya</title> |
|
|
<script src="https://cdn.tailwindcss.com"></script> |
|
|
<link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css" rel="stylesheet"> |
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0"> |
|
|
<style> |
|
|
body { background: #f8fafc; background-image: radial-gradient(#cbd5e1 1px, transparent 1px); background-size: 24px 24px; } |
|
|
.glass-header { background: rgba(255, 255, 255, 0.9); backdrop-filter: blur(10px); border-bottom: 1px solid rgba(255,255,255,0.5); } |
|
|
.glass-card { background: rgba(255, 255, 255, 0.7); backdrop-filter: blur(20px); border: 1px solid rgba(255, 255, 255, 0.5); box-shadow: 0 8px 32px 0 rgba(31, 38, 135, 0.07); } |
|
|
.icon-btn { transition: all 0.3s cubic-bezier(0.34, 1.56, 0.64, 1); background: rgba(255,255,255,0.8); backdrop-filter: blur(5px); } |
|
|
.icon-btn:hover { transform: translateY(-5px) scale(1.05); box-shadow: 0 15px 30px -5px rgba(59, 130, 246, 0.15); border-color: #93c5fd; } |
|
|
.icon-btn.active { background: #eff6ff; border-color: #3b82f6; box-shadow: 0 0 0 4px rgba(59, 130, 246, 0.15); transform: scale(0.98); } |
|
|
.icon-sticker { filter: drop-shadow(0 4px 6px rgba(0,0,0,0.1)); } |
|
|
</style> |
|
|
</head> |
|
|
<body class="font-sans text-slate-800 min-h-screen flex flex-col"> |
|
|
|
|
|
<nav class="glass-header sticky top-0 z-50"> |
|
|
<div class="container mx-auto px-4 py-3 flex justify-between items-center"> |
|
|
<div class="flex items-center gap-4"> |
|
|
<img src="https://upload.wikimedia.org/wikipedia/commons/thumb/a/a9/Seal_of_Meghalaya.svg/150px-Seal_of_Meghalaya.svg.png" class="h-14 drop-shadow-md"> |
|
|
<div> |
|
|
<h2 class="text-[10px] md:text-xs font-bold text-slate-500 uppercase tracking-[0.2em]" data-translate="govt_title">GOVERNMENT OF MEGHALAYA</h2> |
|
|
<h1 class="text-xl md:text-2xl font-extrabold text-slate-900 tracking-tight">MediScan <span class="text-blue-600">AI</span></h1> |
|
|
<p class="text-[10px] font-bold text-green-600 flex items-center gap-1.5 mt-0.5"> |
|
|
<span class="relative flex h-2 w-2"> |
|
|
<span class="animate-ping absolute inline-flex h-full w-full rounded-full bg-green-400 opacity-75"></span> |
|
|
<span class="relative inline-flex rounded-full h-2 w-2 bg-green-500"></span> |
|
|
</span> |
|
|
<span data-translate="online_status">Online Node: Shillong HQ</span> |
|
|
</p> |
|
|
</div> |
|
|
</div> |
|
|
<div class="flex flex-col items-end gap-2"> |
|
|
<select id="lang-select" onchange="changeLanguage()" class="bg-slate-100 border border-slate-200 text-slate-700 text-xs rounded-lg focus:ring-blue-500 focus:border-blue-500 block p-1.5 shadow-sm outline-none"> |
|
|
<option value="en">🇬🇧 English</option> |
|
|
<option value="hi">🇮🇳 हिंदी (Hindi)</option> |
|
|
<option value="as">🇮🇳 অসমীয়া (Assamese)</option> |
|
|
<option value="bn">🇧🇩 বাংলা (Bengali)</option> |
|
|
<option value="kha">🌲 Khasi</option> |
|
|
<option value="gar">⛰️ Garo</option> |
|
|
</select> |
|
|
</div> |
|
|
</div> |
|
|
</nav> |
|
|
|
|
|
<div class="container mx-auto mt-8 p-4 max-w-5xl flex-grow"> |
|
|
|
|
|
<div class="grid grid-cols-3 md:grid-cols-6 gap-4 mb-8"> |
|
|
<button onclick="setTask('lungs', 'image')" id="btn-lungs" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-lungs text-blue-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_lungs">Lungs</span> |
|
|
</button> |
|
|
<button onclick="setTask('cough', 'audio')" id="btn-cough" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-head-side-cough text-teal-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_cough">Cough</span> |
|
|
</button> |
|
|
<button onclick="setTask('fracture', 'image')" id="btn-fracture" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-bone text-slate-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_bone">Fracture</span> |
|
|
</button> |
|
|
<button onclick="setTask('brain', 'image')" id="btn-brain" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-brain text-purple-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_brain">Brain</span> |
|
|
</button> |
|
|
<button onclick="setTask('eye', 'image')" id="btn-eye" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-eye text-indigo-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_eye">Eye</span> |
|
|
</button> |
|
|
<button onclick="setTask('skin', 'image')" id="btn-skin" class="icon-btn rounded-2xl p-4 flex flex-col items-center justify-center gap-3 border border-white/60 shadow-sm group"> |
|
|
<i class="fas fa-hand-dots text-orange-500 text-5xl mb-1 icon-sticker transition-transform group-hover:scale-110"></i> |
|
|
<span class="text-xs font-extrabold text-slate-600 uppercase tracking-wide" data-translate="btn_skin">Skin</span> |
|
|
</button> |
|
|
</div> |
|
|
|
|
|
<div class="glass-card rounded-3xl overflow-hidden relative"> |
|
|
<div id="scope-box" class="hidden bg-slate-50/50 border-b border-slate-100 p-3 text-center backdrop-blur-sm"> |
|
|
<p class="text-[10px] font-bold text-slate-400 uppercase tracking-wider 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 class="p-8"> |
|
|
<h2 id="header-text" class="text-2xl font-bold text-slate-800 mb-8 text-center">Select a Category</h2> |
|
|
|
|
|
<div id="inputs" class="opacity-50 pointer-events-none transition-all duration-500 mb-8 max-w-2xl mx-auto"> |
|
|
<div class="grid grid-cols-3 gap-4 mb-4"> |
|
|
<div class="col-span-2"> |
|
|
<label class="text-[10px] font-bold text-slate-400 uppercase tracking-wider mb-1 block" data-translate="lbl_name">Patient Name</label> |
|
|
<input type="text" id="p-name" class="w-full bg-white/80 border border-slate-200 p-3 rounded-xl focus:ring-2 focus:ring-blue-500 outline-none transition-all"> |
|
|
</div> |
|
|
<div> |
|
|
<label class="text-[10px] font-bold text-slate-400 uppercase tracking-wider mb-1 block" data-translate="lbl_age">Age</label> |
|
|
<input type="number" id="p-age" class="w-full bg-white/80 border border-slate-200 p-3 rounded-xl focus:ring-2 focus:ring-blue-500 outline-none transition-all"> |
|
|
</div> |
|
|
</div> |
|
|
|
|
|
<div onclick="document.getElementById('file-input').click()" class="border-2 border-dashed border-slate-300 rounded-2xl p-10 text-center cursor-pointer hover:bg-blue-50/50 hover:border-blue-400 transition-all duration-300 group relative overflow-hidden bg-white/60"> |
|
|
<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 duration-300"> |
|
|
<div class="w-20 h-20 bg-blue-100 rounded-full flex items-center justify-center mx-auto mb-4 text-blue-600 shadow-sm"> |
|
|
<i id="upload-icon" class="fas fa-cloud-upload-alt text-3xl"></i> |
|
|
</div> |
|
|
<p id="upload-text" class="text-slate-600 font-bold" data-translate="txt_upload">Tap to upload</p> |
|
|
<p class="text-xs text-slate-400 mt-1">Supported: JPG, PNG, WAV, MP3</p> |
|
|
</div> |
|
|
<div class="relative z-10"> |
|
|
<img id="img-preview" class="hidden mx-auto max-h-64 rounded-xl shadow-lg object-contain bg-black/5"> |
|
|
<audio id="audio-preview" controls class="hidden w-full mt-4"></audio> |
|
|
</div> |
|
|
</div> |
|
|
</div> |
|
|
|
|
|
<button id="run-btn" onclick="analyze()" class="hidden w-full max-w-md mx-auto bg-gradient-to-r from-blue-600 to-indigo-700 hover:from-blue-700 hover:to-indigo-800 text-white font-bold py-4 rounded-xl shadow-lg shadow-blue-200 transition-all transform hover:scale-[1.02] flex items-center justify-center gap-3 animate-pulse-soft"> |
|
|
<i class="fas fa-microscope text-xl"></i> <span class="text-lg" data-translate="btn_run">Run Diagnosis</span> |
|
|
</button> |
|
|
|
|
|
<div id="loader" class="hidden text-center py-8"> |
|
|
<div class="relative w-16 h-16 mx-auto mb-4"> |
|
|
<div class="absolute inset-0 border-4 border-slate-100 rounded-full"></div> |
|
|
<div class="absolute inset-0 border-4 border-blue-500 rounded-full border-t-transparent animate-spin"></div> |
|
|
</div> |
|
|
<p class="text-sm font-bold text-slate-600 animate-pulse" data-translate="txt_analyzing">Analyzing data...</p> |
|
|
</div> |
|
|
|
|
|
<div id="result-box" class="hidden mt-10 border-t border-slate-100 pt-8"> |
|
|
<div class="flex flex-col md:flex-row justify-between items-start gap-6 mb-6"> |
|
|
<div> |
|
|
<p class="text-xs font-bold text-slate-400 uppercase tracking-widest mb-1" data-translate="lbl_result">Analysis Result</p> |
|
|
<h1 id="res-label" class="text-3xl md:text-4xl font-extrabold text-slate-800 tracking-tight">--</h1> |
|
|
<p class="text-sm text-slate-500 mt-2 font-medium bg-slate-100/80 inline-block px-3 py-1 rounded-lg border border-slate-200"> |
|
|
<span data-translate="lbl_conf">AI Confidence</span>: <span id="res-conf" class="font-mono text-slate-800">--</span> |
|
|
</p> |
|
|
</div> |
|
|
<span id="res-badge" class="px-5 py-2.5 rounded-xl text-sm font-bold uppercase shadow-sm tracking-wide">--</span> |
|
|
</div> |
|
|
|
|
|
<div id="alert-box" class="hidden p-6 rounded-2xl border border-l-4 shadow-sm flex flex-col md:flex-row items-start md:items-center gap-5 transition-all bg-white/80 backdrop-blur"> |
|
|
<div class="p-4 bg-slate-50 rounded-full shrink-0"> |
|
|
<i id="alert-icon" class="fas fa-info-circle text-2xl text-slate-600"></i> |
|
|
</div> |
|
|
<div class="flex-grow"> |
|
|
<strong class="block text-lg mb-1 text-slate-800" data-translate="lbl_action">Action Required</strong> |
|
|
<span id="alert-text" class="text-sm text-slate-600 leading-relaxed">--</span> |
|
|
</div> |
|
|
<button id="doctor-btn" onclick="findDoctor()" class="hidden px-6 py-3 bg-white text-slate-800 font-bold text-sm rounded-xl shadow-md border border-slate-100 hover:bg-slate-50 hover:shadow-lg transition flex items-center gap-2 whitespace-nowrap"> |
|
|
<i class="fas fa-map-marker-alt text-red-500 text-lg"></i> <span>Find Nearest Doctor</span> |
|
|
</button> |
|
|
</div> |
|
|
|
|
|
<div class="mt-8 flex items-center justify-between text-[10px] text-slate-400 border-t border-slate-50 pt-4"> |
|
|
<span class="flex items-center gap-2"><i class="fas fa-database text-blue-400"></i> <span data-translate="lbl_sync">Ayushman Bharat Sync</span></span> |
|
|
<span id="sync-msg" class="text-yellow-500 font-bold flex items-center gap-1"><i class="fas fa-sync fa-spin"></i> Pending...</span> |
|
|
</div> |
|
|
</div> |
|
|
</div> |
|
|
</div> |
|
|
|
|
|
<div class="mt-12 mb-12"> |
|
|
<div class="flex justify-between items-center mb-6"> |
|
|
<h3 class="text-sm font-bold text-slate-500 uppercase tracking-widest flex items-center gap-2"><i class="fas fa-history"></i> Recent Patients</h3> |
|
|
<button onclick="clearHistory()" class="text-[10px] font-bold text-red-500 hover:text-red-700 bg-red-50 hover:bg-red-100 px-4 py-2 rounded-lg transition-colors uppercase tracking-wide">Clear History</button> |
|
|
</div> |
|
|
<div class="glass-card rounded-xl shadow-sm overflow-hidden border border-slate-200/50"> |
|
|
<table class="w-full text-sm text-left"> |
|
|
<thead class="bg-slate-50/80 text-slate-500 font-bold uppercase text-[10px] tracking-wider"> |
|
|
<tr><th class="px-6 py-4">Time</th><th class="px-6 py-4">Patient</th><th class="px-6 py-4">Category</th><th class="px-6 py-4">Diagnosis</th><th class="px-6 py-4">Risk</th></tr> |
|
|
</thead> |
|
|
<tbody id="history-table" class="divide-y divide-slate-100"></tbody> |
|
|
</table> |
|
|
</div> |
|
|
</div> |
|
|
</div> |
|
|
|
|
|
<script> |
|
|
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", online_status: "Online Node: Shillong HQ", |
|
|
btn_lungs: "Lungs", btn_cough: "Cough", btn_bone: "Fracture", btn_brain: "Brain", btn_eye: "Eye", btn_skin: "Skin", |
|
|
lbl_name: "Patient Name", lbl_age: "Age / ID", txt_upload: "Tap to upload Scan/Photo", btn_run: "Run Diagnosis", |
|
|
txt_analyzing: "Analyzing medical data...", lbl_result: "Analysis Result", lbl_conf: "AI Confidence", |
|
|
lbl_action: "Medical Action Required", lbl_scope: "Scope of Detection", lbl_sync: "Ayushman Bharat Sync" |
|
|
}, |
|
|
hi: { |
|
|
govt_title: "मेघालय सरकार", online_status: "ऑनलाइन नोड: शिलांग", |
|
|
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_scope: "जांच का दायरा", lbl_sync: "सरकारी सिंक" |
|
|
}, |
|
|
as: { |
|
|
govt_title: "মেঘালয় চৰকাৰ", online_status: "অনলাইন ন'ড: শ্বিলং", |
|
|
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_scope: "পৰিসৰ", lbl_sync: "ডাটাবেচ" |
|
|
}, |
|
|
bn: { |
|
|
govt_title: "মেঘালয় সরকার", online_status: "অনলাইন নোড: শিলং", |
|
|
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_scope: "শনাক্তকরণের পরিসর", lbl_sync: "সরকারি সিঙ্ক" |
|
|
}, |
|
|
kha: { |
|
|
govt_title: "SORKAR MEGHALAYA", online_status: "Online: Shillong", |
|
|
btn_lungs: "Phopsa", btn_cough: "Jyrhoh", btn_bone: "Shyieng", btn_brain: "Jabieng", btn_eye: "Khmat", btn_skin: "Doh", |
|
|
lbl_name: "Kyrteng", lbl_age: "Rta", txt_upload: "Thep dur", btn_run: "Leh Test", |
|
|
txt_analyzing: "Check...", lbl_result: "Result", lbl_conf: "Jingshisha", |
|
|
lbl_action: "Leh kane", lbl_scope: "Jingshem", lbl_sync: "Sorkar Sync" |
|
|
}, |
|
|
gar: { |
|
|
govt_title: "MEGHALAYA SORKARI", online_status: "Online: Shillong", |
|
|
btn_lungs: "Ka'sop", btn_cough: "Gusuk", btn_bone: "Greng", btn_brain: "Taning", btn_eye: "Mikron", btn_skin: "Bigil", |
|
|
lbl_name: "Bimung", lbl_age: "Bilsi", txt_upload: "Gata", btn_run: "Porikka", |
|
|
txt_analyzing: "Niyenga...", lbl_result: "Result", lbl_conf: "Bebera'ani", |
|
|
lbl_action: "Kam", lbl_scope: "Am·sandiani", lbl_sync: "Sorkar Sync" |
|
|
} |
|
|
}; |
|
|
|
|
|
let currTask = null, currType = 'image', currFile = null, currLang = 'en'; |
|
|
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]; |
|
|
}); |
|
|
if (currTask) { |
|
|
let taskKey = "btn_" + currTask; |
|
|
document.getElementById('header-text').innerHTML = `Upload <span class="uppercase text-blue-600">${t[taskKey]}</span>`; |
|
|
} |
|
|
} |
|
|
|
|
|
function setTask(task, type) { |
|
|
currTask = task; currType = type; |
|
|
let t = TRANSLATIONS[currLang]; |
|
|
|
|
|
document.querySelectorAll('.icon-btn').forEach(b => { |
|
|
b.classList.remove('active', 'border-blue-500', 'ring-2', 'ring-blue-200'); |
|
|
b.classList.add('border-white/60'); |
|
|
}); |
|
|
let btn = document.getElementById('btn-'+task); |
|
|
btn.classList.add('active', 'border-blue-500', 'ring-2', 'ring-blue-200'); |
|
|
|
|
|
document.getElementById('header-text').innerHTML = `Upload <span class="uppercase text-blue-600">${t["btn_" + task]}</span>`; |
|
|
|
|
|
let scopeTags = document.getElementById('scope-tags'); |
|
|
scopeTags.innerHTML = ""; |
|
|
MODEL_SCOPES[task].forEach(d => { |
|
|
let tag = document.createElement("span"); |
|
|
tag.className = "px-3 py-1 bg-white text-slate-600 text-[10px] font-bold uppercase rounded-full border border-slate-200 tracking-wide shadow-sm"; |
|
|
tag.innerText = d; |
|
|
scopeTags.appendChild(tag); |
|
|
}); |
|
|
document.getElementById('scope-box').classList.remove('hidden'); |
|
|
|
|
|
let input = document.getElementById('file-input'); |
|
|
let icon = document.getElementById('upload-icon'); |
|
|
let txt = document.getElementById('upload-text'); |
|
|
|
|
|
if (type === 'audio') { |
|
|
input.accept = ".wav, .mp3, audio/*"; |
|
|
icon.className = "fas fa-microphone-alt text-4xl text-teal-500 mb-2"; |
|
|
txt.innerHTML = "Tap to upload Audio<br><span class='text-xs text-slate-400'>.wav, .mp3</span>"; |
|
|
} else { |
|
|
input.accept = "image/*"; |
|
|
icon.className = "fas fa-cloud-upload-alt text-4xl text-blue-400 mb-2"; |
|
|
txt.innerText = t['txt_upload']; |
|
|
} |
|
|
|
|
|
document.getElementById('inputs').classList.remove('opacity-50', 'pointer-events-none'); |
|
|
document.getElementById('result-box').classList.add('hidden'); |
|
|
document.getElementById('run-btn').classList.add('hidden'); |
|
|
document.getElementById('placeholder').classList.remove('hidden'); |
|
|
document.getElementById('img-preview').classList.add('hidden'); |
|
|
document.getElementById('audio-preview').classList.add('hidden'); |
|
|
currFile = null; |
|
|
} |
|
|
|
|
|
function showPreview(event) { |
|
|
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; aud.classList.remove('hidden'); |
|
|
document.getElementById('img-preview').classList.add('hidden'); |
|
|
} else { |
|
|
let img = document.getElementById('img-preview'); |
|
|
img.src = url; 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'); |
|
|
} |
|
|
} |
|
|
|
|
|
async function analyze() { |
|
|
if (!currTask || !currFile) return; |
|
|
let pname = document.getElementById('p-name').value; |
|
|
let page = document.getElementById('p-age').value; |
|
|
if (!pname) { alert("Please enter Patient Name first."); return; } |
|
|
|
|
|
document.getElementById('run-btn').classList.add('hidden'); |
|
|
document.getElementById('loader').classList.remove('hidden'); |
|
|
document.getElementById('result-box').classList.add('hidden'); |
|
|
|
|
|
let formData = new FormData(); |
|
|
formData.append("file", currFile); |
|
|
|
|
|
try { |
|
|
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'); |
|
|
|
|
|
updateHistoryTable(); |
|
|
|
|
|
if (data.risk === "INVALID") { |
|
|
updateResultUI("Rejected", "--", "INVALID INPUT", "bg-slate-200 text-slate-600"); |
|
|
showAlert("Input Error", data.error, "bg-slate-100 border-slate-200 text-slate-800", "fa-exclamation-triangle text-slate-400"); |
|
|
return; |
|
|
} |
|
|
|
|
|
if (data.error) { alert(data.error); document.getElementById('run-btn').classList.remove('hidden'); return; } |
|
|
|
|
|
let conf = (data.prediction.score * 100).toFixed(1) + "%"; |
|
|
updateResultUI(data.prediction.label, conf, data.risk + " RISK", ""); |
|
|
|
|
|
let btnDoc = document.getElementById('doctor-btn'); |
|
|
|
|
|
if (data.risk === "HIGH") { |
|
|
document.getElementById('res-badge').className = "px-4 py-2 rounded-full text-sm font-bold uppercase shadow-sm tracking-wide bg-red-100 text-red-700"; |
|
|
showAlert("Critical Issue Detected", "Immediate referral to District Hospital recommended.", "bg-red-50 border-red-100 text-red-900", "fa-ambulance text-red-500"); |
|
|
btnDoc.classList.remove('hidden'); |
|
|
} else if (data.risk === "MODERATE") { |
|
|
document.getElementById('res-badge').className = "px-4 py-2 rounded-full text-sm font-bold uppercase shadow-sm tracking-wide bg-yellow-100 text-yellow-800"; |
|
|
showAlert("Moderate Risk", "Consult local PHC doctor for further tests.", "bg-yellow-50 border-yellow-100 text-yellow-900", "fa-user-md text-yellow-600"); |
|
|
btnDoc.classList.remove('hidden'); |
|
|
} else { |
|
|
document.getElementById('res-badge').className = "px-4 py-2 rounded-full text-sm font-bold uppercase shadow-sm tracking-wide bg-green-100 text-green-700"; |
|
|
document.getElementById('alert-box').classList.add('hidden'); |
|
|
btnDoc.classList.add('hidden'); |
|
|
} |
|
|
|
|
|
setTimeout(() => { |
|
|
document.getElementById('sync-msg').innerHTML = "<i class='fas fa-check-circle'></i> Synced to Govt Database"; |
|
|
document.getElementById('sync-msg').className = "text-green-600 font-bold flex items-center gap-1"; |
|
|
}, 2000); |
|
|
|
|
|
} catch (e) { |
|
|
alert("Connection Failed."); |
|
|
document.getElementById('loader').classList.add('hidden'); |
|
|
document.getElementById('run-btn').classList.remove('hidden'); |
|
|
} |
|
|
} |
|
|
|
|
|
// FIND NEAREST DOCTOR (Corrected URL) |
|
|
function findDoctor() { |
|
|
if (navigator.geolocation) { |
|
|
navigator.geolocation.getCurrentPosition((pos) => { |
|
|
let lat = pos.coords.latitude; |
|
|
let lon = pos.coords.longitude; |
|
|
window.open(`https://www.google.com/maps/search/hospitals+near+me/@${lat},${lon},13z`, '_blank'); |
|
|
}, () => { |
|
|
window.open('https://www.google.com/maps/search/hospitals+near+me', '_blank'); |
|
|
}); |
|
|
} else { |
|
|
window.open('https://www.google.com/maps/search/hospitals+near+me', '_blank'); |
|
|
} |
|
|
} |
|
|
|
|
|
function updateResultUI(label, conf, badgeText, badgeClass) { |
|
|
document.getElementById('res-label').innerText = label; |
|
|
document.getElementById('res-conf').innerText = conf; |
|
|
document.getElementById('res-badge').innerText = badgeText; |
|
|
if(badgeClass) document.getElementById('res-badge').className = `px-4 py-2 rounded-full text-sm font-bold uppercase shadow-sm tracking-wide ${badgeClass}`; |
|
|
} |
|
|
|
|
|
function showAlert(title, msg, classes, iconClass) { |
|
|
let a = document.getElementById('alert-box'); |
|
|
a.className = `p-5 rounded-xl border border-l-4 shadow-sm flex flex-col md:flex-row items-start md:items-center gap-4 transition-all ${classes}`; |
|
|
a.classList.remove('hidden'); |
|
|
document.getElementById('alert-text').innerText = msg; |
|
|
document.querySelector('[data-translate="lbl_action"]').innerText = title; |
|
|
document.getElementById('alert-icon').className = `fas ${iconClass} text-2xl`; |
|
|
} |
|
|
|
|
|
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-slate-400 text-center italic"><td colspan="5" class="py-6">No records found.</td></tr>'; |
|
|
return; |
|
|
} |
|
|
data.forEach(row => { |
|
|
let color = row.risk === "HIGH" ? "text-red-600 font-bold bg-red-50" : row.risk === "MODERATE" ? "text-yellow-600 font-bold bg-yellow-50" : "text-green-600 bg-green-50"; |
|
|
let tr = ` |
|
|
<tr class="bg-white border-b border-slate-50 hover:bg-slate-50 transition-colors"> |
|
|
<td class="px-6 py-4 text-slate-500 font-mono text-xs">${row.time}</td> |
|
|
<td class="px-6 py-4 font-bold text-slate-700">${row.patient}</td> |
|
|
<td class="px-6 py-4 text-slate-500 text-xs uppercase tracking-wide">${row.task}</td> |
|
|
<td class="px-6 py-4 text-slate-800 font-medium">${row.diagnosis}</td> |
|
|
<td class="px-6 py-4"><span class="px-2 py-1 rounded text-[10px] uppercase tracking-wider ${color}">${row.risk}</span></td> |
|
|
</tr>`; |
|
|
tbody.innerHTML += tr; |
|
|
}); |
|
|
} catch(e) {} |
|
|
} |
|
|
|
|
|
async function clearHistory() { |
|
|
if(confirm("Clear local patient history?")) { await fetch("/reset_history", { method: "POST" }); updateHistoryTable(); } |
|
|
} |
|
|
</script> |
|
|
</body> |
|
|
</html> |
|
|
""" |
