Update app.py

app.py

@@ -4,77 +4,100 @@ import numpy as np
 import pandas as pd
 from PIL import Image
 
-# 建立 PaddleOCR 實例
+# 初始化 OCR
 ocr = PaddleOCR(use_angle_cls=True, lang='ch')
 
 # 權限密碼
 ADMIN_PASSWORD = "admin123"
 
-# 儲存 DataFrame
-global_df = pd.DataFrame(columns=["門市", "日期", "時段", "人員"])
-
-# 工時換算（可依需求更改）
+# 排班工時對照表
 TIME_SLOTS = {
     "早(07-15)": 8,
     "早(06-14)": 8,
+    "早(08-18)": 10,
     "晚(15-23)": 8,
+    "晚(17-25)": 8,
     "晚(18-23)": 5,
     "夜(23-07)": 8,
-    "晚(17-25)": 8,
-    "早(08-18)": 10,
-    "晚(18-23)": 5
 }
 
+# 全域 DataFrame
+global_df = pd.DataFrame(columns=["醫院", "日期", "時段", "人員", "工時"])
+
+
+# OCR 文本解析為 DataFrame
 def extract_info_from_text(text):
-    """
-    模擬從 OCR 文本解析結構化資料，可依實際格式進階開發
-    """
     import re
     rows = []
     for line in text.splitlines():
         match = re.findall(r"(安學|天期|永華東|鎮山|頂美)\s*(\d+月\d+日)\s*(早.*?|晚.*?|夜.*?)\s*([^\s]+)", line)
         for m in match:
-            rows.append({"門市": m[0], "日期": m[1], "時段": m[2], "人員": m[3]})
+            時段 = m[2]
+            工時 = 0
+            for k in TIME_SLOTS:
+                if k in 時段:
+                    工時 = TIME_SLOTS[k]
+                    break
+            rows.append({
+                "醫院": m[0],
+                "日期": m[1],
+                "時段": 時段,
+                "人員": m[3],
+                "工時": 工時
+            })
     return pd.DataFrame(rows)
 
+
+# 管理員上傳 + OCR 辨識
 def admin_upload(pwd, image):
     if pwd != ADMIN_PASSWORD:
         return "❌ 密碼錯誤", None
 
-    img_np = np.array(image)
-    result = ocr.ocr(img_np, cls=True)
-    full_text = "\n".join([line[1][0] for line in result[0]])
+    try:
+        img_np = np.array(image)
+        result = ocr.ocr(img_np)
 
-    df = extract_info_from_text(full_text)
+        if not result or not result[0]:
+            return "❌ OCR 未辨識出任何文字", None
 
-    global global_df
-    global_df = pd.concat([global_df, df], ignore_index=True)
+        full_text = "\n".join([line[1][0] for line in result[0] if line and line[1]])
 
-    return "✅ OCR 辨識完成", df
+        df = extract_info_from_text(full_text)
 
+        if df.empty:
+            return "⚠️ 無法解析表格內容", None
+
+        global global_df
+        global_df = pd.concat([global_df, df], ignore_index=True)
+
+        return "✅ OCR 辨識完成並儲存", df
+
+    except Exception as e:
+        return f"❌ 發生錯誤：{str(e)}", None
+
+
+# 查詢人員排班紀錄
 def query_by_staff(staff_name):
     df = global_df[global_df["人員"] == staff_name]
     if df.empty:
         return f"未找到 {staff_name} 的值班紀錄", None
     return "", df
 
+
+# 查詢人員總工時
 def calc_total_hours(staff_name):
     df = global_df[global_df["人員"] == staff_name]
-    total = 0
-    for _, row in df.iterrows():
-        time_label = row["時段"]
-        for key, val in TIME_SLOTS.items():
-            if key in time_label:
-                total += val
+    total = df["工時"].sum()
     return f"{staff_name} 一週總值班時數為：{total} 小時"
 
-# 建立 Gradio 介面
+
+# Gradio 介面
 with gr.Blocks() as demo:
-    gr.Markdown("## 📋 排班表 OCR 辨識系統")
+    gr.Markdown("## 🏥 醫院值班表 OCR 辨識系統")
 
-    with gr.Tab("🔐 管理者上傳 (admin123)"):
-        admin_pwd = gr.Textbox(label="輸入管理者密碼", type="password")
-        admin_img = gr.Image(type="pil", label="上傳排班表圖片")
+    with gr.Tab("🔐 管理員上傳"):
+        admin_pwd = gr.Textbox(label="輸入密碼", type="password")
+        admin_img = gr.Image(type="pil", label="上傳排班圖片")
         admin_btn = gr.Button("辨識並儲存")
         admin_status = gr.Textbox(label="狀態")
         admin_df = gr.Dataframe()
@@ -83,9 +106,9 @@ with gr.Blocks() as demo:
                         inputs=[admin_pwd, admin_img],
                         outputs=[admin_status, admin_df])
 
-    with gr.Tab("🔍 查詢值班紀錄"):
+    with gr.Tab("📋 查詢個人紀錄"):
         staff_input = gr.Textbox(label="輸入人員姓名")
-        staff_btn = gr.Button("查詢每日紀錄")
+        staff_btn = gr.Button("查詢紀錄")
         staff_status = gr.Textbox(label="狀態")
         staff_result = gr.Dataframe()
 
@@ -93,13 +116,13 @@ with gr.Blocks() as demo:
                         inputs=staff_input,
                         outputs=[staff_status, staff_result])
 
-    with gr.Tab("📊 整週總時數"):
-        staff_total = gr.Textbox(label="輸入人員姓名")
+    with gr.Tab("⏱️ 查詢總時數"):
+        total_input = gr.Textbox(label="輸入人員姓名")
         total_btn = gr.Button("計算總時數")
-        total_output = gr.Textbox(label="總時數")
+        total_result = gr.Textbox(label="總時數")
 
         total_btn.click(fn=calc_total_hours,
-                        inputs=staff_total,
-                        outputs=total_output)
+                        inputs=total_input,
+                        outputs=total_result)
 
 demo.launch()