Update app.py

app.py

@@ -1,98 +1,106 @@
+# app.py  –  ĐÃ SỬA HOÀN CHỈNH, CÓ Ô UPLOAD FILE
 import gradio as gr
 import pandas as pd
 import numpy as np
-from datetime import datetime
+import os
+
+def process(upload_thuoc, upload_thuoc_plt, upload_icd_plt, upload_xetnghiem):
+    if None in [upload_thuoc, upload_thuoc_plt, upload_icd_plt, upload_xetnghiem]:
+        return None, "Vui lòng upload đủ 4 file!"
 
-def process_data():
     try:
-        # Đọc file
-        thuoc_df = pd.read_excel("thuoc.xlsx")
-        thuoc_plt = pd.read_excel("Thuoc_PLT.xlsx")
-        icd_plt = pd.read_excel("ICD_PLT.xlsx")
-        xetnghiem_df = pd.read_excel("xetnghiem.xlsx")
+        # Đọc file từ upload
+        thuoc_df       = pd.read_excel(upload_thuoc)
+        thuoc_plt      = pd.read_excel(upload_thuoc_plt)
+        icd_plt        = pd.read_excel(upload_icd_plt)
+        xetnghiem_df   = pd.read_excel(upload_xetnghiem)
 
         # ==================== 1. LỌC PLT < 100 ====================
-        plt_df = xetnghiem_df[xetnghiem_df['Tên XN'].str.contains('PLT', case=False, na=False)].copy()
+        plt_mask = xetnghiem_df['Tên XN'].str.contains('PLT', case=False, na=False)
+        plt_df = xetnghiem_df[plt_mask].copy()
         plt_df['Kết quả XN'] = pd.to_numeric(plt_df['Kết quả XN'], errors='coerce')
         plt_low = plt_df[plt_df['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 tìm thấy bệnh nhân nào có PLT < 100"
+            return None, "Không có bệnh nhân nào PLT < 100"
 
         # ==================== 2. Chuẩn hóa mabhyt ====================
-        def normalize_mabhyt(x):
+        def norm_mabhyt(x):
             if pd.isna(x): return np.nan
             return int(str(x).replace('.', ''))
 
-        thuoc_plt['mabhyt_norm'] = thuoc_plt['mabhyt'].apply(normalize_mabhyt)
-        thuoc_df['mabhyt_norm'] = thuoc_df['mabhyt'].apply(normalize_mabhyt)
+        thuoc_plt['mabhyt_norm'] = thuoc_plt['mabhyt'].apply(norm_mabhyt)
+        thuoc_df['mabhyt_norm']  = thuoc_df['mabhyt'].apply(norm_mabhyt)
 
-        thuoc_full = thuoc_df.merge(thuoc_plt[['mabhyt_norm', 'hoatchat']], on='mabhyt_norm', how='left')
+        thuoc_full = thuoc_df.merge(thuoc_plt[['mabhyt_norm', 'hoatchat']], 
+                                  on='mabhyt_norm', how='left')
 
         # ==================== 3. Loại bệnh nhân có chẩn đoán trong ICD_PLT ====================
         exclude_icds = set(icd_plt['maicd'].dropna().astype(str).str.strip())
 
-        def has_excluded_icd(icd_str):
+        def has_excluded(icd_str):
             if pd.isna(icd_str): return False
-            codes = [c.strip() for c in str(icd_str).split(';')]
-            return any(code in exclude_icds for code in codes)
+            return any(c.strip() in exclude_icds for c in str(icd_str).split(';'))
 
-        thuoc_full['exclude'] = thuoc_full['maicd'].apply(has_excluded_icd)
+        thuoc_full['exclude'] = thuoc_full['maicd'].apply(has_excluded)
         thuoc_clean = thuoc_full[~thuoc_full['exclude']].copy()
 
         # ==================== 4. Chỉ giữ bệnh nhân có PLT < 100 ====================
         valid_patients = set(plt_low['makcb'])
-        thuoc_final = thuoc_clean[thuoc_clean['makcb'].isin(valid_patients)].copy()
+        thuoc_final = thuoc_clean[thuoc_clean['makcb'].isin(valid_patients)]
 
         if thuoc_final.empty:
-            return None, "Không còn bệnh nhân nào sau khi loại chẩn đoán PLT"
+            return None, "Sau khi loại chẩn đoán PLT → không còn bệnh nhân nào"
 
-        # Các cột thuốc cần giữ
-        thuoc_cols = ['makcb','hoten','ngaysinh','maicd','chandoan','ngaychidinh','maview',
+        cols_thuoc = ['makcb','hoten','ngaysinh','maicd','chandoan','ngaychidinh','maview',
                       'mabhyt','hoatchat','tenhh','dvt','hamluong','SL','cachdung','BS','duongdung','lieudung']
-        thuoc_final = thuoc_final[thuoc_cols]
+        thuoc_final = thuoc_final[cols_thuoc]
 
-        # ==================== 5. Ghép cartesian thuốc + PLT < 100 ====================
+        # ==================== 5. Ghép cartesian ====================
         result = thuoc_final.merge(plt_low, on='makcb', how='inner')
+        result = result.sort_values(['makcb', 'ngaychidinh', 'hoatchat', 'Ngày KQ'])
 
-        # Sắp xếp đẹp
-        result = result.sort_values(['makcb', 'hoten', 'ngaychidinh', 'hoatchat', 'Ngày KQ'])
-
-        # Đúng thứ tự 20 cột như yêu cầu
         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]
 
-        # Lưu file kết quả
-        output_file = "KetQua_Thuoc_Va_PLT_duoi_100.xlsx"
-        result.to_excel(output_file, index=False)
+        # Lưu file
+        output_path = "/tmp/KetQua_PLT_duoi_100_FULL.xlsx"
+        result.to_excel(output_path, index=False)
 
         summary = f"""
         HOÀN TẤT!
-        Tổng bệnh nhân có PLT < 100: {plt_low['makcb'].nunique()}
-        Sau khi loại chẩn đoán PLT: còn {result['makcb'].nunique()} bệnh nhân
-        Tổng số dòng kết quả (thuốc × PLT): {len(result):,} dòng
-        Đã lưu file: {output_file}
+        • Bệnh nhân có PLT < 100               : {plt_low['makcb'].nunique()}
+        • Sau khi loại chẩn đoán PLT            : {result['makcb'].nunique()} bệnh nhân
+        • Tổng dòng kết quả (thuốc × PLT)       : {len(result):,} dòng
         """
-        return output_file, summary
+        return output_path, summary
 
     except Exception as e:
         return None, f"Lỗi: {str(e)}"
 
-# Giao diện Gradio
-with gr.Blocks(title="Lọc bệnh nhân dùng thuốc + PLT < 100") as demo:
-    gr.Markdown("# Lọc bệnh nhân có PLT < 100 và loại chẩn đoán PLT")
-    gr.Markdown("Upload đủ 4 file: thuoc.xlsx, Thuoc_PLT.xlsx, ICD_PLT.xlsx, xetnghiem.xlsx")
+# ==================== GIAO DIỆN GRADIO ====================
+with gr.Blocks(title="Lọc PLT < 100 + Loại chẩn đoán PLT") as demo:
+    gr.Markdown("# Lọc bệnh nhân có PLT < 100 và loại bỏ chẩn đoán liên quan tiểu cầu")
     
     with gr.Row():
-        btn = gr.Button("Chạy xử lý", variant="primary", size="lg")
-    
+        thuoc       = gr.File(label="1. thuoc.xlsx", file_types=[".xlsx"])
+        thuoc_plt   = gr.File(label="2. Thuoc_PLT.xlsx", file_types=[".xlsx"])
     with gr.Row():
-        output_file = gr.File(label="Tải kết quả Excel")
-        output_text = gr.Textbox(label="Thông báo", lines=10)
+        icd_plt     = gr.File(label="3. ICD_PLT.xlsx", file_types=[".xlsx"])
+        xetnghiem   = gr.File(label="4. xetnghiem.xlsx", file_types=[".xlsx"])
+
+    btn = gr.Button("CHẠY XỬ LÝ", variant="primary", size="lg")
     
-    btn.click(process_data, outputs=[output_file, output_text])
+    output_file = gr.File(label="TẢI KẾT QUẢ EXCEL")
+    output_text = gr.Textbox(label="Thông báo chi tiết", lines=8)
+
+    btn.click(
+        fn=process,
+        inputs=[thuoc, thuoc_plt, icd_plt, xetnghiem],
+        outputs=[output_file, output_text]
+    )
 
 demo.launch()