Update app.py

app.py

@@ -1,116 +1,29 @@
-# ==========================================
-# 核心修復：偽裝螢幕環境，防止雲端伺服器崩潰
-# ==========================================
-import sys
-import types
-
-# 建立一個假的 screeninfo 模組
-fake_screeninfo = types.ModuleType("screeninfo")
-fake_screeninfo.ScreenInfoError = Exception
-
-# 偽裝 get_monitors 函數，讓它回傳一個假的螢幕物件（包含 width 和 height 屬性）
-class FakeMonitor:
-    width = 1920
-    height = 1080
-fake_screeninfo.get_monitors = lambda: [FakeMonitor()]
-
-# 強行注入到 Python 的全域模組緩存中
-sys.modules["screeninfo"] = fake_screeninfo
-# ==========================================
-
 import os
+import sys
 import shutil
 import subprocess
 import gradio as gr
 
 OMR_DIR = "OMRChecker"
+MOCK_DIR = os.path.abspath("mock_libs")
 
-# 1. 啟動時檢查並自動克隆 OMRChecker 官方專案
-if not os.path.exists(OMR_DIR):
-    print("正在從 GitHub 複製 OMRChecker 核心專案...")
-    subprocess.run(["git", "clone", "https://github.com/udayraj123/OMRChecker.git"], check=True)
-    # 安裝該專案可能缺失的其他微量依賴
-    subprocess.run([sys.executable, "-m", "pip", "install", "-r", f"{OMR_DIR}/requirements.txt"], check=True)
-
-def process_omr(image_file, template_content):
-    if image_file is None or not template_content:
-        return None, None, "錯誤：請務必上傳答案卡圖片並填寫 Template JSON 配置。"
-
-    # 定義 OMRChecker 的工作目錄結構
-    job_dir = os.path.join(OMR_DIR, "inputs", "hf_space_job")
-    outputs_dir = os.path.join(OMR_DIR, "outputs")
-
-    # 清理舊的暫存輸入與輸出，避免前後兩次請求互相干擾
-    if os.path.exists(job_dir):
-        shutil.rmtree(job_dir)
-    os.makedirs(job_dir, exist_ok=True)
-    
-    if os.path.exists(outputs_dir):
-        shutil.rmtree(outputs_dir)
-
-    # 將上傳的圖片複製到 OMR 要求的輸入目錄
-    target_image_path = os.path.join(job_dir, "sheet.jpg")
-    shutil.copy(image_file, target_image_path)
-
-    # 將前端/API 傳入的 Template JSON 寫入對應位置
-    with open(os.path.join(job_dir, "template.json"), "w", encoding="utf-8") as f:
-        f.write(template_content)
-        
-    # 保險起見，也在 inputs 根目錄放一份
-    with open(os.path.join(OMR_DIR, "inputs", "template.json"), "w", encoding="utf-8") as f:
-        f.write(template_content)
+# =======================================================
+# 核心修復：在硬碟建立實體偽裝庫，並透過 PYTHONPATH 注入子程序
+# =======================================================
+os.makedirs(MOCK_DIR, exist_ok=True)
+with open(os.path.join(MOCK_DIR, "screeninfo.py"), "w", encoding="utf-8") as f:
+    f.write('''\
+class ScreenInfoError(Exception):
+    pass
 
-    # 2. 執行 OMRChecker CLI 核心
-    result = subprocess.run(
-        [sys.executable, "main.py"],
-        cwd=OMR_DIR,
-        capture_output=True,
-        text=True
-    )
-
-    # 3. 搜尋並收集 OMRChecker 產生的輸出結果 (CSV 與 標記圖片)
-    csv_file = None
-    output_image = None
-    
-    if os.path.exists(outputs_dir):
-        for root, dirs, files in os.walk(outputs_dir):
-            for file in files:
-                if file.endswith(".csv"):
-                    csv_file = os.path.join(root, file)
-                elif file.endswith((".jpg", ".jpeg", ".png")) and "results" in root.lower():
-                    output_image = os.path.join(root, file)
-
-    # 彙整 CLI 的標準輸出日誌
-    logs = f"--- STDOUT ---\n{result.stdout}\n\n--- STDERR ---\n{result.stderr}"
-    
-    return csv_file, output_image, logs
-
-# 4. 建立 Gradio 介面
-interface = gr.Interface(
-    fn=process_omr,
-    inputs=[
-        gr.Image(type="filepath", label="1. 上傳 OMR 答案卡圖片 (JPG/PNG)"),
-        gr.Textbox(lines=12, label="2. 填入 Template JSON 配置", placeholder='{"columns": [...]}')
-    ],
-    outputs=[
-        gr.File(label="下載辨識結果 (CSV)"),
-        gr.Image(label="視覺化劃記檢視圖片"),
-        gr.Textbox(label="系統執行日誌 (Logs)")
-    ],
-    title="OMRChecker 雲端辨識系統 API",
-    description="本系統基於 udayraj123/OMRChecker。除了網頁操作外，亦可透過頁尾的 'Use via API' 連結進行自動化調用。"
-)
-
-if __name__ == "__main__":
-    interface.launch()
-
-import os
-import sys
-import shutil
-import subprocess
-import gradio as gr
+class FakeMonitor:
+    width = 1920
+    height = 1080
 
-OMR_DIR = "OMRChecker"
+def get_monitors():
+    return [FakeMonitor()]
+''')
+# =======================================================
 
 # 1. 啟動時檢查並自動克隆 OMRChecker 官方專案
 if not os.path.exists(OMR_DIR):
@@ -147,12 +60,17 @@ def process_omr(image_file, template_content):
     with open(os.path.join(OMR_DIR, "inputs", "template.json"), "w", encoding="utf-8") as f:
         f.write(template_content)
 
-    # 2. 執行 OMRChecker CLI 核心
+    # 配置環境變數，強迫子程序優先載��我們的 mock_libs
+    current_env = os.environ.copy()
+    current_env["PYTHONPATH"] = MOCK_DIR + os.pathsep + current_env.get("PYTHONPATH", "")
+
+    # 2. 執行 OMRChecker CLI 核心 (帶入偽裝環境變數)
     result = subprocess.run(
         [sys.executable, "main.py"],
         cwd=OMR_DIR,
         capture_output=True,
-        text=True
+        text=True,
+        env=current_env
     )
 
     # 3. 搜尋並收集 OMRChecker 產生的輸出結果 (CSV 與 標記圖片)
@@ -172,7 +90,7 @@ def process_omr(image_file, template_content):
     
     return csv_file, output_image, logs
 
-# 4. 建立 Gradio 介面（這會自動產生對應的 API 接口）
+# 4. 建立 Gradio 介面
 interface = gr.Interface(
     fn=process_omr,
     inputs=[