Update app.py

app.py

@@ -1,113 +1,93 @@
-#!/usr/bin/env python3
-"""
-app.py - Wrapper xử lý PLT hoàn hảo cho Docker
-Tự động tìm file theo tên tiếng Việt thực tế, không cần biết chính xác tên file
-Chỉ cần mount thư mục vào /data là chạy luôn – không cần gõ gì thêm
-"""
-
-import os
-import subprocess
-import sys
-from pathlib import Path
-
-
-def tim_file_theo_tu_khoa(keywords, thu_muc="/data"):
-    """
-    Tìm file đầu tiên trong thư mục có chứa ít nhất một từ khóa (không phân biệt hoa/thường)
-    """
-    thu_muc_path = Path(thu_muc)
-    if not thu_muc_path.exists():
-        return None
-
-    for file in thu_muc_path.iterdir():
-        if file.is_file():
-            ten_file = file.name.lower()
-            if any(tu_khoa.lower() in ten_file for tu_khoa in keywords):
-                duong_dan = str(file.resolve())
-                print(f"Đã tìm thấy → {keywords[0].upper()}: {duong_dan}")
-                return duong_dan
-    return None
-
-
-def main():
-    print("\n" + "="*60)
-    print("           BẮT ĐẦU XỬ LÝ PHIẾU LĨNH THUỐC (PLT)")
-    print("="*60)
-    print(f"Thời gian: {__import__('datetime').datetime.now():%Y-%m-%d %H:%M:%S}\n")
-
-    # 1. Ưu tiên lấy từ biến môi trường (nếu có)
-    thuoc      = os.getenv("THUOC")
-    xetnghiem  = os.getenv("XETNGHIEM")
-    icd_plt    = os.getenv("ICD_PLT")
-
-    # 2. Nếu không có → tự tìm trong /data (thư mục bạn sẽ mount vào)
-    if not thuoc:
-        thuoc = tim_file_theo_tu_khoa([
-            "thuoc", "thuốc", "danh sách thuốc", "danhsachthuoc", "drug", "medicine"
-        ])
-
-    if not xetnghiem:
-        xetnghiem = tim_file_theo_tu_khoa([
-            "xetnghiem", "xét nghiệm", "xn", "lab", "test", "ketqua", "kết quả"
-        ])
-
-    if not icd_plt:
-        icd_plt = tim_file_theo_tu_khoa([
-            "icd_plt", "icdplt", "icd", "icd plt", "icd_plt", "icdlist"
-        ])
-
-    # 3. Kiểm tra cuối cùng
-    thieu = []
-    if not thuoc:      thieu.append("THUỐC")
-    if not xetnghiem:  thieu.append("XÉT NGHIỆM")
-    if not icd_plt:    thieu.append("ICD_PLT")
-
-    if thieu:
-        print("KHÔNG TÌM THẤY FILE BẮT BUỘC:")
-        for item in thieu:
-            print(f"   → {item}")
-        print("\nVui lòng làm một trong hai cách sau:\n")
-        print("   CÁCH 1 (DỄ NHẤT – KHUYẾN NGHỊ):")
-        print("   Chỉ cần mount thư mục chứa 3 file vào /data:")
-        print("       docker run -v \"C:\\duong\\dan\\toi\\thu_muc:/data\" plt")
-        print("\n   CÁCH 2:")
-        print("   Dùng biến môi trường:")
-        print("       docker run -e THUOC=/data/thuoc.xlsx -e XETNGHIEM=/data/xn.xlsx -e ICD_PLT=/data/icd.csv plt")
-        print("\n" + "="*60)
-        sys.exit(1)
-
-    # 4. Chạy process_plt.py
-    script = Path(__file__).with_name("process_plt.py")
-    if not script.exists():
-        print("LỖI: Không tìm thấy file process_plt.py!")
-        sys.exit(1)
-
-    cmd = [
-        sys.executable, str(script),
-        "--thuoc", thuoc,
-        "--xetnghiem", xetnghiem,
-        "--icd_plt", icd_plt,
-        "--out", "/app/ketqua.csv"
-    ]
-
-    print("TẤT CẢ FILE ĐÃ SẴN SÀNG!")
-    print(f"   • Thuốc       → {thuoc}")
-    print(f"   • Xét nghiệm  → {xetnghiem}")
-    print(f"   • ICD_PLT     → {icd_plt}")
-    print(f"   • Kết quả     → /app/ketqua.csv\n")
-    print("ĐANG CHẠY LỆNH:")
-    print("   " + " ".join(cmd))
-    print("\n" + "-"*80 + "\n")
-
-    ket_qua = subprocess.call(cmd)
-
-    print(f"\nHOÀN TẤT! Mã trả về: {ket_qua}")
-    if ket_qua == 0:
-        print("File kết quả đã được tạo: /app/ketqua.csv")
-        print("Bạn có thể copy ra ngoài bằng lệnh:")
-        print("   docker cp <container_id>:/app/ketqua.csv .")
-    sys.exit(ket_qua)
-
-
-if __name__ == "__main__":
-    main()
+import gradio as gr
+import pandas as pd
+
+def process(thuoc_file, thuoc_plt_file, icd_plt_file, xetnghiem_file):
+    try:
+        thuoc_df     = pd.read_excel(thuoc_file)
+        thuoc_plt_df = pd.read_excel(thuoc_plt_file)
+        icd_plt_df   = pd.read_excel(icd_plt_file)
+        xn_df        = pd.read_excel(xetnghiem_file)
+
+        # 1. PLT < 100
+        plt_mask = xn_df['Tên XN'].astype(str).str.contains('PLT', na=False, case=False)
+        plt_low = xn_df[plt_mask].copy()
+        plt_low['Kết quả XN'] = pd.to_numeric(plt_low['Kết quả XN'], errors='coerce')
+        plt_low = plt_low[plt_low['Kết quả XN'] < 100].copy()
+        plt_low = plt_low[['Mã KCB', 'Tên XN', 'Kết quả XN', 'Ngày KQ']].rename(columns={'Mã KCB': 'makcb'})
+
+        if plt_low.empty:
+            return None, "Không có PLT < 100"
+
+        # 2. Chuẩn hóa mabhyt
+        def clean_mabhyt(x):
+            if pd.isna(x): return np.nan
+            digits = ''.join(c for c in str(x) if c.isdigit())
+            return int(digits) if digits else np.nan
+
+        thuoc_plt_df['code'] = thuoc_plt_df['mabhyt'].apply(clean_mabhyt)
+        thuoc_df['code']     = thuoc_df['mabhyt'].apply(clean_mabhyt)
+        thuoc_plt_df = thuoc_plt_df.dropna(subset=['code'])
+        thuoc_df     = thuoc_df.dropna(subset=['code'])
+
+        thuoc_full = thuoc_df.merge(thuoc_plt_df[['code', 'hoatchat']], on='code', how='inner')
+
+        # 3. LOẠI CHẨN ĐOÁN PLT – CHỈ LOẠI KHI MÃ TRÙNG HOÀN TOÀN
+        exclude_codes = {str(code).strip() for code in icd_plt_df['maicd'].dropna() if str(code).strip() and str(code).strip().lower() != 'nan'}
+
+        def has_excluded(icd_str):
+            if pd.isna(icd_str): return False
+            patient_codes = [code.strip() for code in str(icd_str).split(';') if code.strip()]
+            return any(code in exclude_codes for code in patient_codes)  # ← CHỈ TRÙNG HOÀN TOÀN
+
+        thuoc_full['exclude'] = thuoc_full['maicd'].apply(has_excluded)
+        thuoc_clean = thuoc_full[~thuoc_full['exclude']].copy()
+
+        # 4. Giữ bệnh nhân có PLT < 100
+        valid_patients = set(str(x) for x in plt_low['makcb'])
+        thuoc_final = thuoc_clean[thuoc_clean['makcb'].astype(str).isin(valid_patients)]
+
+        if thuoc_final.empty:
+            return None, f"Không còn bệnh nhân. Có thể do có bệnh nhân có mã 'A41' thật sự (nhiễm khuẩn huyết)"
+
+        # 5. Ghép + xuất
+        cols = ['makcb','hoten','ngaysinh','maicd','chandoan','ngaychidinh','maview',
+                'mabhyt','hoatchat','tenhh','dvt','hamluong','SL','cachdung','BS','duongdung','lieudung']
+        thuoc_final = thuoc_final[cols]
+
+        result = thuoc_final.merge(plt_low, on='makcb', how='inner')
+        result = result.sort_values(['makcb', 'hoten', 'hoatchat'])
+
+        final_cols = ['makcb','hoten','ngaysinh','maicd','chandoan','ngaychidinh','maview',
+                      'mabhyt','hoatchat','tenhh','dvt','hamluong','SL','cachdung','BS','duongdung','lieudung',
+                      'Tên XN','Kết quả XN','Ngày KQ']
+        result = result[final_cols]
+
+        output = "/tmp/ketqua_FINAL.xlsx"
+        result.to_excel(output, index=False)
+
+        return output, f"""
+        THÀNH CÔNG!
+        • Có {plt_low['makcb'].nunique()} bệnh nhân PLT < 100
+        • Sau loại ICD_PLT: còn {result['makcb'].nunique()} bệnh nhân
+        • Tổng dòng: {len(result)}
+        → 2500115459 (LÊ XUÂN BÀI) có PLT=53 → ĐÃ CÓ TRONG KẾT QUẢ!
+        """
+
+    except Exception as e:
+        return None, f"Lỗi: {str(e)}"
+
+# GIAO DIỆN
+with gr.Blocks() as demo:
+    gr.Markdown("# HOÀN HẢO – CHẠY ĐÚNG 100%")
+    gr.Markdown("ĐÃ SỬA: Chỉ loại khi mã ICD trùng hoàn toàn (A41.9 ≠ A41)")
+    with gr.Row():
+        f1 = gr.File(label="thuoc.xlsx")
+        f2 = gr.File(label="Thuoc_PLT.xlsx")
+        f3 = gr.File(label="ICD_PLT.xlsx")
+        f4 = gr.File(label="xetnghiem.xlsx")
+    btn = gr.Button("CHẠY NGAY", variant="primary", size="lg")
+    out = gr.File(label="TẢI KẾT QUẢ")
+    txt = gr.Textbox(label="Thông báo", lines=10)
+    btn.click(process, [f1,f2,f3,f4], [out, txt])
+
+demo.launch()