Add frontend safeJson injection for robust AI post creation

patch_runtime.py

@@ -1,17 +1,16 @@
 """Runtime patch layer for VNEWS.
-
 Keeps the current large app intact, but replaces fragile AI wall endpoints with
-stable JSON endpoints so the frontend never receives HTML/traceback and never
-throws `Unexpected token < in JSON`.
+stable JSON endpoints and injects frontend safeJson wrappers.
 """
 import hashlib
 import time
+import os
 from urllib.parse import quote
 
 import requests
 from bs4 import BeautifulSoup
 from fastapi import Request
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, HTMLResponse
 
 import main as _main
 
@@ -28,32 +27,24 @@ def _safe_text(v):
 
 
 def _ensure_article(url: str):
-    """Return article-like dict with title, summary, og_image, body.
-    Never raises.
-    """
     data = None
     try:
         if hasattr(_main, "_article_by_url"):
             data = _main._article_by_url(url)
     except Exception:
         data = None
-
     if not data:
         try:
             data = _main._scrape_generic_article(url) if hasattr(_main, "_scrape_generic_article") else None
         except Exception:
             data = None
-
     if not data:
         data = {"title": "", "summary": "", "og_image": "", "body": [], "url": url, "source": "generic"}
-
     title = _safe_text(data.get("title"))
     summary = _safe_text(data.get("summary"))
     img = _safe_text(data.get("og_image"))
     body = data.get("body") or []
-
-    # Last-resort OG scrape for title/summary/image.
-    if not title or not summary or not img:
+    if not title or not summary or not img or not body:
         try:
             r = requests.get(url, headers=getattr(_main, "HEADERS", {}), timeout=15)
             r.encoding = "utf-8"
@@ -73,22 +64,20 @@ def _ensure_article(url: str):
                     t = p.get_text(" ", strip=True)
                     if len(t) > 40:
                         ps.append({"type": "p", "text": t})
-                    if len(ps) >= 20:
+                    if len(ps) >= 30:
                         break
                 body = ps
         except Exception:
             pass
-
     if not summary and body:
         first = next((b.get("text", "") for b in body if b.get("type") == "p" and b.get("text")), "")
-        summary = first[:320]
+        summary = first[:360]
     if not title:
         title = url
     if not img:
         img = DEFAULT_IMG
     if not body and summary:
         body = [{"type": "p", "text": summary}]
-
     data.update({"title": title, "summary": summary, "og_image": img, "body": body, "url": url})
     return data
 
@@ -106,11 +95,20 @@ def _rewrite(data, tone="tu-nhien"):
     ps = [b.get("text", "") for b in data.get("body", []) if b.get("type") == "p" and b.get("text")]
     lead = summary or (ps[0] if ps else "")
     points = "\n".join(["• " + p[:220] + ("..." if len(p) > 220 else "") for p in ps[:5]])
-    return (f"Bản tin AI viết lại: {title}\n\n{lead}\n\nĐiểm chính:\n{points}").strip()
+    body = "\n\n".join(ps[:10])
+    return (f"Bản tin AI viết lại: {title}\n\n{lead}\n\n{body}\n\nĐiểm chính:\n{points}").strip()
+
+
+def _topic_image(topic):
+    try:
+        if hasattr(_main, "_image_for_topic"):
+            return _main._image_for_topic(topic)
+    except Exception:
+        pass
+    return "https://image.pollinations.ai/prompt/" + quote("editorial illustration Vietnamese news " + topic, safe="") + "?width=1024&height=576&nologo=true"
 
 
 def _save_post(post):
-    posts = []
     try:
         posts = _main._load_wall() if hasattr(_main, "_load_wall") else []
     except Exception:
@@ -124,7 +122,7 @@ def _save_post(post):
     return post
 
 
-_remove_routes(["/api/url_wall", "/api/topic_post", "/api/rewrite_share"])
+_remove_routes(["/api/url_wall", "/api/topic_post", "/api/rewrite_share", "/"])
 
 
 @app.post("/api/url_wall")
@@ -158,7 +156,6 @@ async def patched_url_wall(request: Request):
 
 @app.post("/api/rewrite_share")
 async def patched_rewrite_share(request: Request):
-    # Same robust behavior as url_wall, but accepts url/tone from article view.
     return await patched_url_wall(request)
 
 
@@ -183,21 +180,14 @@ async def patched_topic_post(request: Request):
             context = ""
         if not context:
             context = f"Chủ đề: {topic}"
-        prompt_data = {
-            "title": topic,
-            "summary": context[:420],
-            "og_image": getattr(_main, "_image_for_topic", lambda x: DEFAULT_IMG)(topic),
-            "body": [{"type": "p", "text": context}],
-            "source": "topic",
-            "url": "",
-        }
-        text = _rewrite(prompt_data, tone=tone)
+        data = {"title": topic, "summary": context[:420], "og_image": _topic_image(topic), "body": [{"type": "p", "text": context}], "source": "topic", "url": ""}
+        text = _rewrite(data, tone=tone)
         post = {
             "id": hashlib.md5((topic + str(time.time())).encode()).hexdigest()[:12],
             "url": "",
             "title": topic,
-            "summary": prompt_data["summary"],
-            "img": prompt_data["og_image"] or DEFAULT_IMG,
+            "summary": data["summary"],
+            "img": data["og_image"] or DEFAULT_IMG,
             "text": text or context,
             "source": "topic",
             "ts": int(time.time()),
@@ -206,3 +196,79 @@ async def patched_topic_post(request: Request):
         return JSONResponse({"post": post})
     except Exception as e:
         return JSONResponse({"error": "Không tạo được bài theo chủ đề", "detail": str(e)[:300]}, status_code=500)
+
+
+_FRONTEND_PATCH = r'''
+<script>
+(function(){
+  async function safeJson(res){
+    const text = await res.text();
+    try { return JSON.parse(text); }
+    catch(e){ return { error: (text || 'Server không trả JSON').slice(0,500) }; }
+  }
+  window.safeJson = safeJson;
+  window.createUrlPost = function(){
+    let inp=document.getElementById('ai-url-input');
+    let url=(inp&&inp.value||'').trim();
+    if(!url){ alert('Dán URL trước'); return; }
+    fetch('/api/url_wall',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url})})
+      .then(safeJson).then(j=>{
+        if(j&&j.post){
+          if(!j.post.img) j.post.img='https://s1.vnecdn.net/vnexpress/restruct/i/v9505/logo_default.jpg';
+          if(!j.post.text) j.post.text=j.post.summary||j.post.title||'Không lấy được nội dung tóm tắt.';
+          if(typeof prependWallPost==='function') prependWallPost(j.post);
+          alert('Đã tóm tắt URL và đăng lên tường');
+          if(inp) inp.value='';
+        } else alert((j&&j.error)||'Lỗi URL');
+      }).catch(e=>alert('Lỗi URL: '+e.message));
+  };
+  window.createTopicPost = function(){
+    let inp=document.getElementById('ai-topic-input');
+    let topic=(inp&&inp.value||'').trim();
+    if(!topic){ alert('Nhập chủ đề trước'); return; }
+    fetch('/api/topic_post',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({topic})})
+      .then(safeJson).then(j=>{
+        if(j&&j.post){
+          if(!j.post.img) j.post.img='https://s1.vnecdn.net/vnexpress/restruct/i/v9505/logo_default.jpg';
+          if(!j.post.text) j.post.text=j.post.summary||j.post.title||'Không lấy được nội dung.';
+          if(typeof prependWallPost==='function') prependWallPost(j.post);
+          alert('Đã tạo bài và đăng lên tường');
+          if(inp) inp.value='';
+        } else alert((j&&j.error)||'Lỗi tạo bài');
+      }).catch(e=>alert('Lỗi tạo bài: '+e.message));
+  };
+  window.rewriteCurrentArticle = function(){
+    if(!window._currentArticle && typeof _currentArticle!=='undefined') window._currentArticle=_currentArticle;
+    let ca = (typeof _currentArticle!=='undefined') ? _currentArticle : window._currentArticle;
+    if(!ca || !ca.url){ alert('Chưa có bài viết để rewrite'); return; }
+    let tone=document.getElementById('rewrite-tone')?.value||'nghiem-tuc';
+    let btn=document.querySelector('.article-actions button.primary');
+    if(btn){btn.textContent='Đang rewrite...';btn.disabled=true;}
+    fetch('/api/rewrite_share',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url:ca.url,tone})})
+      .then(safeJson).then(j=>{
+        if(j&&j.post){
+          if(!j.post.img) j.post.img='https://s1.vnecdn.net/vnexpress/restruct/i/v9505/logo_default.jpg';
+          if(!j.post.text) j.post.text=j.post.summary||j.post.title||'Không lấy được nội dung.';
+          let box=document.getElementById('rewrite-result');
+          if(box) box.innerHTML='<div class="rewrite-box"><div class="rewrite-title">Đã rewrite và đăng lên Tường AI</div><div class="rewrite-text">'+(j.post.text||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]))+'</div></div>';
+          if(typeof prependWallPost==='function') prependWallPost(j.post);
+          alert('Đã đăng lên Tường AI');
+        } else alert((j&&j.error)||'Không tạo được bài AI');
+      }).catch(e=>alert('Lỗi tạo bài AI: '+e.message))
+      .finally(()=>{if(btn){btn.textContent='🤖 AI viết lại & đăng tường';btn.disabled=false;}});
+  };
+})();
+</script>
+'''
+
+
+@app.get("/")
+async def patched_index():
+    try:
+        with open("/app/static/index.html", "r", encoding="utf-8") as f:
+            html = f.read()
+        if "window.safeJson" not in html:
+            html = html.replace("</body>", _FRONTEND_PATCH + "</body>")
+        return HTMLResponse(content=html)
+    except Exception as e:
+        return HTMLResponse(content=f"<pre>Index error: {str(e)}</pre>", status_code=500)