Update app.py

app.py

@@ -1,62 +1,45 @@
 import os
-os.system("python -m spacy download en_core_web_sm")
+import sys
+
+# 1. 強制修正環境變數，避免雲端安裝模型時權限不足
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/huggingface'
+os.environ['SPACY_CACHE'] = '/tmp/spacy'
+
+# 2. 在所有 import 之前，先確保 spacy 的模型已經下載完成
+try:
+    import spacy
+    if not spacy.util.is_package("en_core_web_sm"):
+        os.system(f"{sys.executable} -m spacy download en_core_web_sm")
+except Exception as e:
+    os.system(f"{sys.executable} -m spacy download en_core_web_sm")
 
 import gradio as gr
 from fastcoref import FCoref
 
-
-# 1. 初始化指代消解模型
-print("正在載入 AI 模型，請稍候...")
+# 3. 初始化模型 (這裡建議直接在啟動時下載)
 model = FCoref()
 
 def coref_chat(user_input):
-    # 執行模型推論
     preds = model.predict(texts=[user_input])
     clusters = preds[0].get_clusters()
     
     if not clusters:
         return user_input, "模型未偵測到明確的指代群組。"
     
-    # 整理輸出結果
     result_text = "🎯 AI 偵測到的連連看群組：\n"
     for i, cluster in enumerate(clusters):
-        result_text += f"【群組 {i+1}】(指的是同一個主體)：{cluster}\n"
+        result_text += f"【群組 {i+1}】：{cluster}\n"
     
     return user_input, result_text
 
-# 2. 建立 Gradio 網頁介面
-with gr.Blocks(title="指代消解 AI 助理") as demo:
-    gr.Markdown("# 🤖 指代消解 (Coreference Resolution) 聊天機器人")
-    gr.Markdown("輸入一段包含多個角色的對話，看 AI 如何幫你「連連看」！")
-    
+# 建立介面
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 指代消解 AI 助理 (Cloud Version)")
     with gr.Row():
-        with gr.Column():
-            txt_input = gr.Textbox(
-                label="請輸入測試文本", 
-                placeholder="例如: Professor James and Mary are in the lab. She smiled at him.",
-                lines=5
-            )
-            submit_btn = gr.Button("開始分析")
-            
-        with gr.Column():
-            txt_original = gr.Textbox(label="你輸入的文字", interactive=False)
-            txt_output = gr.Textbox(label="AI 分析結果", lines=8, interactive=False)
-
-    submit_btn.click(
-        fn=coref_chat, 
-        inputs=txt_input, 
-        outputs=[txt_original, txt_output]
-    )
-    
-    gr.Examples(
-        examples=[
-            ["Professor James and his assistant Mary are in the lab. The woman told the man that the experiment was a success. She smiled at him."],
-            ["Ben brought some pizza for his friends, and they all enjoyed it."]
-        ],
-        inputs=txt_input
-    )
+        txt_input = gr.Textbox(label="請輸入測試文本", lines=3)
+        submit_btn = gr.Button("開始分析")
+    txt_output = gr.Textbox(label="AI 分析結果", lines=5)
+    submit_btn.click(fn=coref_chat, inputs=txt_input, outputs=[txt_input, txt_output])
 
-# 3. 啟動機器人
 if __name__ == "__main__":
-    # share=True 會產生一個臨時網址，你可以傳給老師或組員看
     demo.launch()