Update app.py

app.py

@@ -1,14 +1,70 @@
+# app.py  — CPU/免費空間穩定版（Caption only）
+import os
+import gc
 import gradio as gr
+from PIL import Image
+from transformers import pipeline
 
-def ping(txt):
-    return "OK: " + (txt or "")
+# 降低雜訊與多執行緒開銷
+os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "true"
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
+os.environ["PIP_DISABLE_PIP_VERSION_CHECK"] = "1"
 
-with gr.Blocks(title="Sanity Check") as demo:
-    gr.Markdown("# ✅ Build sanity check")
-    inp = gr.Textbox(label="Input")
-    out = gr.Textbox(label="Output")
-    btn = gr.Button("Run")
-    btn.click(ping, inputs=inp, outputs=out, concurrency_limit=1)
+# 嘗試偵測 torch（若沒裝也能跑，僅用 CPU）
+try:
+    import torch  # type: ignore
+    _HAS_TORCH = True
+except Exception:
+    torch = None  # type: ignore
+    _HAS_TORCH = False
+
+# 懶載入：第一次呼叫才建立管線
+_captioner = None
+
+def get_device_id():
+    if _HAS_TORCH:
+        try:
+            return 0 if torch.cuda.is_available() else -1
+        except Exception:
+            return -1
+    return -1
+
+def get_captioner():
+    global _captioner
+    if _captioner is None:
+        _captioner = pipeline(
+            task="image-to-text",
+            model="Salesforce/blip-image-captioning-base",
+            device=get_device_id(),
+        )
+    return _captioner
+
+def do_caption(image: Image.Image):
+    if image is None:
+        return "請先上傳圖片。"
+    pipe = get_captioner()
+    out = pipe(image)
+    # 釋放一些暫存（避免長時間使用記憶體升高）
+    gc.collect()
+    return out[0]["generated_text"]
+
+def clear_all():
+    return None, ""
+
+# ===== UI =====
+with gr.Blocks(title="Vision Language AI 應用（Caption 版）") as demo:
+    gr.Markdown("# 🎨 Vision Language AI 應用展示 · Caption")
+    gr.Markdown("上傳圖片後按下按鈕，產生文字描述。此版本僅啟用 **Image Captioning**，以確保在免費 CPU 環境穩定執行。")
+
+    with gr.Row():
+        img = gr.Image(type="pil", label="上傳圖片", height=320)
+        out = gr.Textbox(label="AI 產生的描述", lines=6, show_copy_button=True)
+    with gr.Row():
+        run_btn = gr.Button("🚀 產生描述", variant="primary")
+        clear_btn = gr.Button("🧹 清除", variant="secondary")
+
+    run_btn.click(fn=do_caption, inputs=img, outputs=out, concurrency_limit=1)
+    clear_btn.click(fn=clear_all, inputs=None, outputs=[img, out], concurrency_limit=1)
 
 demo.queue(max_size=10)