Create app.py

app.py

@@ -0,0 +1,105 @@
+import gradio as gr
+from paddleocr import PaddleOCR
+import numpy as np
+import pandas as pd
+from PIL import Image
+
+# 建立 PaddleOCR 實例
+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,
+    "晚(15-23)": 8,
+    "晚(18-23)": 5,
+    "夜(23-07)": 8,
+    "晚(17-25)": 8,
+    "早(08-18)": 10,
+    "晚(18-23)": 5
+}
+
+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]})
+    return pd.DataFrame(rows)
+
+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]])
+
+    df = extract_info_from_text(full_text)
+
+    global global_df
+    global_df = pd.concat([global_df, df], ignore_index=True)
+
+    return "✅ OCR 辨識完成", df
+
+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
+    return f"{staff_name} 一週總值班時數為：{total} 小時"
+
+# 建立 Gradio 介面
+with gr.Blocks() as demo:
+    gr.Markdown("## 📋 排班表 OCR 辨識系統")
+
+    with gr.Tab("🔐 管理者上傳 (admin123)"):
+        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()
+
+        admin_btn.click(fn=admin_upload,
+                        inputs=[admin_pwd, admin_img],
+                        outputs=[admin_status, admin_df])
+
+    with gr.Tab("🔍 查詢值班紀錄"):
+        staff_input = gr.Textbox(label="輸入人員姓名")
+        staff_btn = gr.Button("查詢每日紀錄")
+        staff_status = gr.Textbox(label="狀態")
+        staff_result = gr.Dataframe()
+
+        staff_btn.click(fn=query_by_staff,
+                        inputs=staff_input,
+                        outputs=[staff_status, staff_result])
+
+    with gr.Tab("📊 整週總時數"):
+        staff_total = gr.Textbox(label="輸入人員姓名")
+        total_btn = gr.Button("計算總時數")
+        total_output = gr.Textbox(label="總時數")
+
+        total_btn.click(fn=calc_total_hours,
+                        inputs=staff_total,
+                        outputs=total_output)
+
+demo.launch()