Update app.py

app.py

@@ -1,31 +1,60 @@
-from flask import Flask, request, jsonify
+import gradio as gr
+import trafilatura
 import requests
+from markdownify import markdownify as md
+from sumy.parsers.plaintext import PlaintextParser
+from sumy.nlp.tokenizers import Tokenizer
+from sumy.summarizers.text_rank import TextRankSummarizer
 
-app = Flask(__name__)
-
-@app.route("/proxy")
-def proxy():
-    target_url = request.args.get("url")
-    if not target_url:
-        return jsonify({"error": "url 파라미터가 필요합니다"}), 400
+# === 자동요약 함수 ===
+def auto_summarize(text, sentence_count=3):
+    try:
+        parser = PlaintextParser.from_string(text, Tokenizer("english"))
+        summarizer = TextRankSummarizer()
+        sents = [str(s) for s in summarizer(parser.document, sentence_count)]
+        return " ".join(sents).strip()
+    except Exception:
+        return text[:500]
 
+# === URL 파싱 + 모드별 출력 ===
+def parse_and_display(url, mode):
+    if not url.strip():
+        return "❌ URL을 입력하세요."
     try:
-        res = requests.get(target_url, timeout=10, headers={
-            "User-Agent": "Mozilla/5.0 (HTMLGrabber/1.0)"
-        })
-        res.raise_for_status()
-        html = res.text
+        headers = {"User-Agent": "Mozilla/5.0"}
+        r = requests.get(url, headers=headers, timeout=10)
+        r.raise_for_status()
+
+        html_content = trafilatura.extract(
+            r.text,
+            output_format="html",
+            include_tables=True,
+            favor_recall=True
+        )
+        if not html_content:
+            return "본문을 추출할 수 없습니다."
 
-        # HTML을 Markdown 코드블록 형태로 변환
-        code_block = f"```html\n{html}\n```"
+        markdown_text = md(html_content, heading_style="ATX")
+        summary_text = auto_summarize(markdown_text, sentence_count=3)
 
-        # CORS 허용
-        response = jsonify({"code_block": code_block})
-        response.headers["Access-Control-Allow-Origin"] = "*"
-        return response
+        if mode == "자동요약":
+            return f"📝 **자동요약**\n\n{summary_text}\n\n🔗 [원문 보기]({url})"
+        else:
+            return f"📄 **원문**\n\n{markdown_text}\n\n🔗 [원문 보기]({url})"
 
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        return f"에러: {str(e)}"
+
+# === Gradio UI ===
+with gr.Blocks() as demo:
+    gr.Markdown("## 🔗 링크 → 자동요약 / 원문 보기")
+    with gr.Row():
+        url_input = gr.Textbox(label="URL 입력", placeholder="https://example.com", scale=3)
+        mode_select = gr.Radio(["자동요약", "원문"], value="자동요약", label="기본 모드", scale=1)
+    output = gr.Markdown()
+    run_btn = gr.Button("추가")
+
+    run_btn.click(parse_and_display, inputs=[url_input, mode_select], outputs=output)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    demo.launch()