Create app.py

app.py

@@ -0,0 +1,57 @@
+from transformers import pipeline
+import gradio as gr
+import trafilatura
+
+# 載入輕量摘要模型（CPU 友好）
+summarizer = pipeline(
+    "summarization",
+    model="sshleifer/distilbart-cnn-12-6",
+    device=-1  # 強制使用 CPU
+)
+
+def summarize_url(url):
+    if not url or not url.startswith("http"):
+        return "請輸入有效的網址（需包含 http:// 或 https://）"
+    
+    try:
+        # 抓取網頁
+        downloaded = trafilatura.fetch_url(url)
+        if downloaded is None:
+            return "無法載入該網頁，請確認網址是否正確或網站是否允許訪問。"
+        
+        # 提取乾淨正文
+        text = trafilatura.extract(
+            downloaded,
+            include_comments=False,
+            include_tables=False,
+            output_format="txt",
+            max_length=2000  # 限制長度以加速處理
+        )
+        
+        if not text or len(text.strip()) < 100:
+            return "無法提取足夠的文字內容（可能不是文章頁面）。"
+        
+        # 摘要（限制輸入長度）
+        input_text = text[:2000]
+        summary = summarizer(
+            input_text,
+            max_length=130,
+            min_length=40,
+            do_sample=False,
+            truncation=True
+        )
+        return summary[0]['summary_text']
+    
+    except Exception as e:
+        return f"處理時發生錯誤：{str(e)}"
+
+# Gradio 介面
+with gr.Blocks(title="網頁重點整理 Agent") as demo:
+    gr.Markdown("# 🌐 網頁重點整理 Agent\n輸入一篇新聞、部落格或文章的網址，自動幫你生成重點摘要（CPU 版本）")
+    url_input = gr.Textbox(label="請輸入網址", placeholder="https://example.com/article")
+    output = gr.Textbox(label="重點摘要", lines=6)
+    btn = gr.Button("開始整理")
+    btn.click(fn=summarize_url, inputs=url_input, outputs=output)
+    gr.Markdown("💡 提示：僅支援文字為主的網頁（如新聞、技術文章），不支援登入頁、影片頁或動態內容過多的網站。")
+
+demo.launch()