Add guarded hotfix patch for safe app

safe_patch.py

@@ -0,0 +1,223 @@
+"""Guarded hotfixes for VNEWS Safe App.
+
+Never raise during import/apply. All patches are best-effort.
+"""
+import re
+import time
+import requests
+from urllib.parse import quote
+
+
+def apply(app):
+    try:
+        from fastapi import Request, Query
+        from fastapi.responses import JSONResponse, HTMLResponse
+        import ai_ext as base
+    except Exception:
+        return
+
+    shorts_cache = {"t": 0, "d": []}
+    channels = ["baodantri7941", "baosuckhoedoisongboyte"]
+
+    def clean(s):
+        import html as html_lib
+        return re.sub(r"\s+", " ", html_lib.unescape(s or "")).strip()
+
+    bad_topic_lines = [
+        "Chỉ xuất bản bài viết cuối cùng",
+        "không nhắc lại yêu cầu",
+        "không liệt kê chỉ dẫn",
+        "Không sao chép nguyên văn",
+        "hãy tổng hợp và diễn đạt lại",
+        "Bài có tiêu đề",
+        "sapo",
+        "các đoạn phân tích",
+        "Nhiệm vụ: viết một bài báo tiếng Việt hoàn chỉnh",
+    ]
+
+    def clean_topic_output(text):
+        if not text:
+            return text
+        lines = []
+        for ln in str(text).splitlines():
+            low = ln.strip().lower()
+            if any(b.lower() in low for b in bad_topic_lines):
+                continue
+            if low.startswith(("yêu cầu:", "nhiệm vụ:", "bắt buộc:", "đầu ra:", "chỉ xuất bản")):
+                continue
+            lines.append(ln)
+        return re.sub(r"\n{3,}", "\n\n", "\n".join(lines).strip())
+
+    def fallback_shorts():
+        hard = [
+            ("Lu_iCQ5YwNM", "Công an lập hồ sơ xử lý người phụ nữ chửi bới, tát tài xế ô tô | Dân trí", "baodantri7941"),
+            ("CwWvijF8BOA", "Chú rể bật khóc nhận món quà bí mật người cha quá cố gửi 26 năm trước | Dân trí", "baodantri7941"),
+            ("tvPewsc2ph4", "Tính năng ẩn trên iPhone giúp giảm mỏi mắt | Dân trí", "baodantri7941"),
+            ("7Pd6vZ2Lz1M", "Hành động ấm lòng trong tìm kiếm học sinh tử vong ở sông Lô | SKĐS", "baosuckhoedoisongboyte"),
+            ("SlHLt_ZyPiE", "Xử phạt người đàn ông xóa số điện thoại cứu hộ trên cao tốc Bắc - Nam | SKĐS", "baosuckhoedoisongboyte"),
+            ("IUOprcJyYr4", "Phụ nữ táo bón có phải do lười ăn rau? | SKĐS", "baosuckhoedoisongboyte"),
+        ]
+        return [
+            {
+                "id": vid,
+                "title": title,
+                "channel": ch,
+                "link": "https://www.youtube.com/watch?v=" + vid,
+                "img": "https://i.ytimg.com/vi/" + vid + "/hqdefault.jpg",
+                "source": "yt",
+            }
+            for vid, title, ch in hard
+        ]
+
+    def yt_html(handle, count=18):
+        try:
+            headers = getattr(base, "HEADERS", {"User-Agent": "Mozilla/5.0"})
+            html = requests.get(f"https://www.youtube.com/@{handle}/shorts", headers=headers, timeout=6).text
+            ids, out = [], []
+            for m in re.finditer(r'"videoId":"([A-Za-z0-9_-]{11})"', html):
+                vid = m.group(1)
+                if vid in ids:
+                    continue
+                ids.append(vid)
+                snip = html[max(0, m.start() - 900):m.start() + 1600]
+                mt = re.search(r'"title":\{"runs":\[\{"text":"([^"]+)"', snip) or re.search(r'"accessibilityText":"([^"]+)"', snip)
+                title = clean(mt.group(1).replace('\\n', ' ')) if mt else "YouTube Short"
+                out.append({
+                    "id": vid,
+                    "title": title,
+                    "channel": handle,
+                    "link": "https://www.youtube.com/watch?v=" + vid,
+                    "img": "https://i.ytimg.com/vi/" + vid + "/hqdefault.jpg",
+                    "source": "yt",
+                })
+                if len(out) >= count:
+                    break
+            return out
+        except Exception:
+            return []
+
+    def fresh_shorts():
+        seen, out = set(), []
+        for ch in channels:
+            for v in yt_html(ch, 18):
+                if v["id"] not in seen:
+                    seen.add(v["id"])
+                    out.append(v)
+        for v in fallback_shorts():
+            if v["id"] not in seen:
+                seen.add(v["id"])
+                out.append(v)
+        return out[:40]
+
+    def remove_routes(paths):
+        app.router.routes = [r for r in app.router.routes if getattr(r, "path", None) not in set(paths)]
+
+    remove_routes(["/api/shorts", "/api/topic_post", "/api/rewrite_topic"])
+
+    @app.get("/api/shorts")
+    def api_shorts(refresh: int = Query(default=0)):
+        now = time.time()
+        if not refresh and shorts_cache["d"] and now - shorts_cache["t"] < 600:
+            return JSONResponse(shorts_cache["d"])
+        data = fresh_shorts()
+        shorts_cache.update({"t": now, "d": data})
+        return JSONResponse(data)
+
+    @app.post("/api/topic_post")
+    async def topic_post(request: Request):
+        body = await request.json()
+        topic = clean(body.get("topic", ""))
+        if not topic:
+            return JSONResponse({"error": "missing topic"}, status_code=400)
+        try:
+            img = base.pollinations_image_url(topic)
+        except Exception:
+            img = "https://image.pollinations.ai/prompt/" + quote("Vietnamese news editorial illustration " + topic) + "?width=1024&height=576&nologo=true"
+        prompt = f"""Viết một bài báo tiếng Việt hoàn chỉnh, chất lượng cao về chủ đề: {topic}
+
+Hãy sử dụng kiến thức tổng hợp của bạn để tạo nội dung có giá trị thực sự cho độc giả.
+
+Phong cách:
+- Như một bài báo/tạp chí đã hoàn thiện, không phải dàn ý.
+- Không nhắc lại yêu cầu của người dùng.
+- Không liệt kê chỉ dẫn viết bài.
+- Không sao chép nguyên văn nguồn nào.
+
+Nội dung cần có:
+- Tiêu đề cụ thể.
+- Sapo ngắn, hấp dẫn.
+- Các đoạn phân tích bối cảnh, nguyên nhân, tác động, ví dụ và nhận định.
+- Nếu chủ đề là thể thao như World Cup, hãy nói về ý nghĩa giải đấu, lịch sử, tác động tới bóng đá, đội tuyển/cầu thủ, kinh tế - truyền thông và cảm xúc người hâm mộ.
+- Nếu thiếu dữ kiện thời sự mới, hãy diễn đạt thận trọng và tập trung vào kiến thức nền.
+- Cuối bài có mục Nguồn tham khảo ngắn: Qwen2.5-VL / kiến thức tổng hợp.
+"""
+        try:
+            text = await base.qwen_generate(prompt, image_url=img, max_tokens=1800)
+        except Exception:
+            text = ""
+        if not text:
+            text = f"{topic}\n\n{topic} là một chủ đề có nhiều lớp ý nghĩa, từ bối cảnh lịch sử đến tác động xã hội, truyền thông và đời sống người hâm mộ. Bài viết này tổng hợp các kiến thức nền và những điểm đáng chú ý nhất để người đọc hiểu rõ hơn về chủ đề.\n\nNguồn tham khảo: Qwen2.5-VL / kiến thức tổng hợp."
+        text = clean_topic_output(text)
+        post = base.make_post(topic, text, img, "", "topic_qwen", sources=[{"title": "Qwen2.5-VL / kiến thức tổng hợp", "url": "", "via": "Qwen2.5-VL"}])
+        post["images"] = [img]
+        posts = base._load_ai_wall()
+        posts.insert(0, post)
+        base._save_ai_wall(posts)
+        return JSONResponse({"post": post})
+
+    @app.post("/api/rewrite_topic")
+    async def rewrite_topic(request: Request):
+        body = await request.json()
+        post_id = str(body.get("id", "")).strip()
+        posts = base._load_ai_wall()
+        p = next((x for x in posts if str(x.get("id")) == post_id), None)
+        if not p:
+            return JSONResponse({"error": "post not found"}, status_code=404)
+        prompt = f"""Viết lại bài sau theo phong cách báo chí phân tích, mạch lạc, hấp dẫn. Chỉ xuất bản bản viết lại cuối cùng.
+
+Tiêu đề: {p.get('title','')}
+Nội dung:
+{(p.get('text') or '')[:12000]}
+"""
+        try:
+            txt = await base.qwen_generate(prompt, image_url=p.get("img") or None, max_tokens=1600)
+        except Exception:
+            txt = ""
+        if not txt:
+            txt = p.get("text", "")
+        txt = clean_topic_output(txt)
+        new = dict(p)
+        new["id"] = str(int(time.time() * 1000))
+        new["text"] = txt
+        new["kind"] = "topic_rewrite"
+        posts.insert(0, new)
+        base._save_ai_wall(posts)
+        return JSONResponse({"post": new})
+
+    # Patch homepage without replacing the whole index route.
+    @app.middleware("http")
+    async def inject_hotfix(request, call_next):
+        response = await call_next(request)
+        try:
+            if request.url.path == "/" and response.headers.get("content-type", "").startswith("text/html"):
+                body = b""
+                async for chunk in response.body_iterator:
+                    body += chunk
+                html = body.decode("utf-8", "ignore")
+                inject = r'''
+<script>
+(function(){
+function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
+function cleanTopicText(){document.querySelectorAll('.wall-text,.article-p,.rewrite-text').forEach(el=>{let t=el.textContent||'';let bad=['Chỉ xuất bản bài viết cuối cùng','Không sao chép nguyên văn','Bài có tiêu đề','Nhiệm vụ: viết một bài báo tiếng Việt hoàn chỉnh'];if(bad.some(b=>t.includes(b))){el.textContent=t.split('\n').filter(l=>!bad.some(b=>l.includes(b))).join('\n')}})}
+async function ensureFastShorts(){let home=document.getElementById('view-home');if(!home||document.getElementById('shorts-hotfix-fast'))return;let sh=await fetch('/api/shorts?refresh=1').then(r=>r.json()).catch(()=>[]);if(!sh.length)return;let wrap=document.createElement('div');wrap.id='shorts-hotfix-fast';wrap.className='slider-wrap';let h='<div class="slider-header"><span class="slider-label">📱 Shorts Dân trí & SKĐS</span><span class="slider-note">mới nhất / tải nhanh</span></div><div class="slider-track">';sh.slice(0,24).forEach((a,i)=>{h+=`<div class="slider-item shorts-item" onclick="openShortsHotfix(${i})"><div class="slider-thumb shorts-thumb">${a.img?`<img src="${a.img}">`:''}<div class="card-play">▶</div></div><div class="slider-title">${esc(a.title)}</div></div>`});h+='</div>';wrap.innerHTML=h;(document.querySelector('.ai-compose')||home.firstChild).after(wrap);window.__hotfixShorts=sh;}
+window.openShortsHotfix=function(start){let arts=window.__hotfixShorts||[];if(!arts.length)return alert('Chưa tải được Shorts');let ordered=start>0?arts.slice(start).concat(arts.slice(0,start)):arts;showView('view-tiktok');let h='<button class="back-btn" onclick="switchCat(\'home\')">← Shorts</button><div class="tiktok-container"><div class="tiktok-feed" id="tiktok-feed">';ordered.forEach((v,i)=>{let src='https://www.youtube.com/embed/'+v.id+'?autoplay=1&rel=0&playsinline=1';h+=`<div class="tiktok-slide"><iframe src="${src}" allowfullscreen allow="accelerometer;autoplay;clipboard-write;encrypted-media;gyroscope;picture-in-picture"></iframe><div class="tiktok-bottom"><span class="badge badge-fpt">YT</span><p class="tiktok-title">${esc(v.title)}</p></div><span class="tiktok-counter">${i+1}/${ordered.length}</span></div>`});h+='</div></div>';document.getElementById('view-tiktok').innerHTML=h;}
+setInterval(()=>{cleanTopicText();if(document.getElementById('view-home')?.classList.contains('active'))ensureFastShorts();},1200);setTimeout(ensureFastShorts,800);
+})();
+</script>
+'''
+                if "</body>" in html:
+                    html = html.replace("</body>", inject + "</body>")
+                return HTMLResponse(html)
+        except Exception:
+            pass
+        return response