Fix: single rewrite btn, 24h TTL for wall/shorts, AI ask uses article content"

ai_runtime_patch_fast.py

@@ -1,5 +1,5 @@
-"""Patch over c1e2703: persistent source_details after reload, topic rewrite, short comments saved."""
-import re, threading, time, json, os
+"""Patch over c1e2703: single rewrite button, 24h TTL for AI wall/shorts, AI ask uses content."""
+import re, threading, time, json, os, asyncio
 import ai_runtime_final6 as f6
 from ai_runtime_final6 import app, rt, f5, HTMLResponse, JSONResponse, Request, Query
 import html as html_lib
@@ -8,20 +8,28 @@ def clean(s):return re.sub(r"\s+"," ",html_lib.unescape(str(s or ""))).strip()
 
 DATA_DIR="/data" if os.path.isdir('/data') else "/app/data"
 SHORT_COMMENTS_FILE=os.path.join(DATA_DIR,'short_comments.json')
+TTL_24H=86400
 
-def _load_comments():
+def _load_json(path,default):
     try:
-        if os.path.exists(SHORT_COMMENTS_FILE):
-            with open(SHORT_COMMENTS_FILE,'r',encoding='utf-8') as f:return json.load(f)
+        if os.path.exists(path):
+            with open(path,'r',encoding='utf-8') as f:return json.load(f)
     except Exception:pass
-    return {}
-def _save_comments(db):
+    return default
+def _save_json(path,data):
     try:
-        os.makedirs(os.path.dirname(SHORT_COMMENTS_FILE),exist_ok=True);tmp=SHORT_COMMENTS_FILE+'.tmp'
-        with open(tmp,'w',encoding='utf-8') as f:json.dump(db,f,ensure_ascii=False)
-        os.replace(tmp,SHORT_COMMENTS_FILE)
+        os.makedirs(os.path.dirname(path),exist_ok=True);tmp=path+'.tmp'
+        with open(tmp,'w',encoding='utf-8') as f:json.dump(data,f,ensure_ascii=False)
+        os.replace(tmp,path)
     except Exception:pass
 
+def _cleanup_old_posts():
+    """Remove AI wall posts and shorts older than 24h."""
+    now=int(time.time())
+    posts=f5.base._load_ai_wall()
+    fresh=[p for p in posts if now-int(p.get('ts') or 0)<TTL_24H]
+    if len(fresh)<len(posts):f5.base._save_ai_wall(fresh)
+
 # ===== BACKGROUND PREFETCH =====
 _bg_home_cache={"t":0,"d":[]}
 _bg_shorts_cache={"t":0,"d":[]}
@@ -42,6 +50,7 @@ def _bg_refresh():
             vid=v.get('id') or ''
             if vid and vid not in seen:seen.add(vid);out.append(v)
         if out:_bg_shorts_cache.update({"t":time.time(),"d":out[:40]})
+        _cleanup_old_posts()
     except Exception:pass
     finally:_bg_loading=False
 
@@ -58,6 +67,8 @@ app.router.routes=[r for r in app.router.routes if not (
     (getattr(r,'path',None)=='/api/topic_post' and 'POST' in getattr(r,'methods',set())) or
     (getattr(r,'path',None)=='/api/short/comments' and 'GET' in getattr(r,'methods',set())) or
     (getattr(r,'path',None)=='/api/short/comment' and 'POST' in getattr(r,'methods',set())) or
+    (getattr(r,'path',None)=='/api/article/ask' and 'POST' in getattr(r,'methods',set())) or
+    (getattr(r,'path',None)=='/api/ai_wall' and 'GET' in getattr(r,'methods',set())) or
     (getattr(r,'path',None)=='/' and 'GET' in getattr(r,'methods',set()))
 )]
 
@@ -81,23 +92,61 @@ def api_shorts_instant(refresh:int=Query(default=0)):
     if hasattr(f6,'api_shorts_final6'):return f6.api_shorts_final6(refresh=refresh)
     return JSONResponse([])
 
-# ===== SHORT COMMENTS (persistent) =====
+@app.get('/api/ai_wall')
+def api_ai_wall_with_ttl():
+    """Return AI wall posts, filtering out posts older than 24h."""
+    now=int(time.time());posts=f5.base._load_ai_wall()
+    fresh=[p for p in posts if now-int(p.get('ts') or 0)<TTL_24H]
+    return JSONResponse({'posts':fresh})
+
+# ===== SHORT COMMENTS =====
 @app.get('/api/short/comments')
 def get_short_comments(id:str=Query(...)):
-    db=_load_comments()
+    db=_load_json(SHORT_COMMENTS_FILE,{})
     return JSONResponse({'comments':db.get(id,[])})
 
 @app.post('/api/short/comment')
 async def post_short_comment(request:Request):
     body=await request.json();vid=str(body.get('id','')).strip();text=clean(body.get('text',''))
     if not vid or not text:return JSONResponse({'error':'missing id or text'},status_code=400)
-    db=_load_comments()
-    comments=db.get(vid,[])
-    comments.insert(0,{'text':text[:300],'ts':int(time.time())})
-    db[vid]=comments[:100]
-    _save_comments(db)
+    db=_load_json(SHORT_COMMENTS_FILE,{})
+    comments=db.get(vid,[]);comments.insert(0,{'text':text[:300],'ts':int(time.time())})
+    db[vid]=comments[:100];_save_json(SHORT_COMMENTS_FILE,db)
     return JSONResponse({'comments':db[vid]})
 
+# ===== ARTICLE ASK AI (based on article content) =====
+@app.post('/api/article/ask')
+async def article_ask(request:Request):
+    body=await request.json();url=clean(body.get('url',''));question=clean(body.get('question',''));context=clean(body.get('context',''))
+    if not question:return JSONResponse({'error':'missing question'},status_code=400)
+    title='';raw=''
+    if url:
+        try:
+            data=f5.base.scrape_any_url(url) if hasattr(f5.base,'scrape_any_url') else None
+            if data:title=data.get('title','');raw=(data.get('summary','')+'\n'+data.get('text','')).strip()
+        except Exception:pass
+    if not raw and context:raw=context[:12000]
+    if not raw:raw=question
+    prompt=f"""Bạn là trợ lý đọc hiểu VNEWS. Trả lời câu hỏi DỰA TRÊN NỘI DUNG BÀI VIẾT bên dưới.
+
+BẮT BUỘC:
+- Chỉ dùng thông tin CÓ TRONG bài viết để trả lời.
+- Nếu bài viết không có thông tin liên quan đến câu hỏi, nói rõ "Bài viết không đề cập đến vấn đề này."
+- Không bịa thêm chi tiết ngoài bài.
+- Trích dẫn/tham chiếu các đoạn cụ thể trong bài khi trả lời.
+
+Tiêu đề bài: {title}
+Nội dung bài:
+{raw[:12000]}
+
+Câu hỏi: {question}
+
+Trả lời bằng tiếng Việt, chi tiết, dựa sát nội dung bài.
+"""
+    ans=await f5.base.qwen_generate(prompt,max_tokens=1200)
+    if not ans:ans='AI chưa trả lời được. Thử hỏi cụ thể hơn.'
+    return JSONResponse({'answer':ans,'title':title})
+
 # ===== TOPIC POST =====
 @app.post('/api/topic_post')
 async def topic_post_focused(request:Request):
@@ -122,7 +171,6 @@ QUY TẮC:
 4. Không mở đầu bằng "Dưới đây là".
 5. Cuối bài có "Nguồn tham khảo".
 """
-    import asyncio
     text=None
     try:text=await asyncio.wait_for(f5.base.qwen_generate(prompt,image_url=img,max_tokens=1700),timeout=35)
     except Exception:text=None
@@ -137,102 +185,91 @@ QUY TẮC:
 
 # ===== FRONTEND =====
 PATCH_INJECT=r'''
-<style>.short-cmt-panel{position:fixed;bottom:0;left:0;right:0;max-height:55vh;background:#181818;border-radius:16px 16px 0 0;z-index:99999;padding:14px;display:none;overflow-y:auto}.short-cmt-panel.active{display:block}.short-cmt-panel textarea{width:100%;background:#222;border:1px solid #444;color:#eee;border-radius:10px;padding:9px;margin:6px 0;min-height:60px}.short-cmt-panel button{background:#2d8659;border:0;color:#fff;border-radius:10px;padding:8px 12px;margin:4px}.cmt-item{background:#222;border-radius:8px;padding:7px;margin:5px 0;color:#ccc;font-size:12px}.source-detail-box{margin-top:14px;background:#151515;border:1px solid #2b2b2b;border-radius:10px;padding:10px}.source-detail-item{background:#202020;border-radius:8px;padding:9px;margin:7px 0}.source-detail-title{font-size:12px;font-weight:700;color:#eee}.source-detail-meta{font-size:10px;color:#888;margin:3px 0}.source-detail-content{font-size:12px;color:#bbb;line-height:1.5;white-space:pre-wrap;max-height:160px;overflow:auto}.source-detail-item a{color:#5cb87a;font-size:11px;text-decoration:none;cursor:pointer}.source-detail-item img{width:100%;aspect-ratio:16/9;object-fit:cover;border-radius:6px;margin-bottom:6px;background:#222}</style>
+<style>.short-cmt-panel{position:fixed;bottom:0;left:0;right:0;max-height:55vh;background:#181818;border-radius:16px 16px 0 0;z-index:99999;padding:14px;display:none;overflow-y:auto}.short-cmt-panel.active{display:block}.short-cmt-panel textarea{width:100%;background:#222;border:1px solid #444;color:#eee;border-radius:10px;padding:9px;margin:6px 0;min-height:60px}.short-cmt-panel button{background:#2d8659;border:0;color:#fff;border-radius:10px;padding:8px 12px;margin:4px}.cmt-item{background:#222;border-radius:8px;padding:7px;margin:5px 0;color:#ccc;font-size:12px}.source-detail-box{margin-top:14px;background:#151515;border:1px solid #2b2b2b;border-radius:10px;padding:10px}.source-detail-item{background:#202020;border-radius:8px;padding:9px;margin:7px 0}.source-detail-title{font-size:12px;font-weight:700;color:#eee}.source-detail-meta{font-size:10px;color:#888;margin:3px 0}.source-detail-content{font-size:12px;color:#bbb;line-height:1.5;white-space:pre-wrap;max-height:160px;overflow:auto}.source-detail-item a{color:#5cb87a;font-size:11px;text-decoration:none;cursor:pointer}.source-detail-item img{width:100%;aspect-ratio:16/9;object-fit:cover;border-radius:6px;margin-bottom:6px;background:#222}.article-ai-ask{margin-top:12px;background:#141414;border:1px solid #2a2a2a;border-radius:10px;padding:10px}.article-ai-ask textarea{width:100%;min-height:60px;background:#222;border:1px solid #444;color:#eee;border-radius:10px;padding:9px}.article-ai-ask button{background:#2d8659;border:0;color:#fff;border-radius:10px;padding:8px 12px;margin-top:6px}.article-ai-answer{white-space:pre-wrap;color:#ccc;font-size:13px;line-height:1.55;margin-top:8px}</style>
 <div id="short-cmt-panel" class="short-cmt-panel"></div>
 <script>
 (function(){
 function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
 
-// === 1) Source details persistent after reload ===
+// === Source details render (persistent after reload) ===
 function renderSourceDetails(post,container){
 let details=post.source_details||[];if(!details.length)return;
 let box=document.createElement('div');box.className='source-detail-box';
-box.innerHTML='<h3 style="font-size:14px;color:#5cb87a;margin-bottom:8px">📚 Bài viết nguồn</h3>'+details.map((s,i)=>{
-let imgHtml=s.img?`<img src="${esc(s.img)}" loading="lazy" onerror="this.style.display='none'">`:'';
-return `<div class="source-detail-item">${imgHtml}<div class="source-detail-title">${i+1}. ${esc(s.title)}</div><div class="source-detail-meta">${esc(s.via||'')}</div><div class="source-detail-content">${esc((s.content||'').slice(0,600))}</div>${s.url?`<a onclick="event.preventDefault();if(typeof readArticle==='function')readArticle('${esc(s.url)}');else window.open('${esc(s.url)}','_blank')">📖 Đọc trên VNEWS</a>`:''}</div>`}).join('');
+box.innerHTML='<h3 style="font-size:14px;color:#5cb87a;margin-bottom:8px">📚 Bài viết nguồn</h3>'+details.map((s,i)=>`<div class="source-detail-item"><div class="source-detail-title">${i+1}. ${esc(s.title)}</div><div class="source-detail-meta">${esc(s.via||'')}</div><div class="source-detail-content">${esc((s.content||'').slice(0,600))}</div>${s.url?`<a onclick="event.preventDefault();if(typeof readArticle==='function')readArticle('${esc(s.url)}')">📖 Đọc trên VNEWS</a>`:''}</div>`).join('');
 container.appendChild(box);
-// Fetch images for sources that don't have img yet.
-details.forEach((s,i)=>{if(s.img||!s.url)return;fetch('/api/article?url='+encodeURIComponent(s.url)).then(r=>r.json()).then(d=>{if(d&&(d.og_image||d.img)){let items=box.querySelectorAll('.source-detail-item');if(items[i]){let img=document.createElement('img');img.src=d.og_image||d.img;img.loading='lazy';img.onerror=function(){this.style.display='none'};items[i].prepend(img);}}}).catch(()=>{});});}
+details.forEach((s,i)=>{if(!s.url)return;fetch('/api/article?url='+encodeURIComponent(s.url)).then(r=>r.json()).then(d=>{if(d&&(d.og_image||d.img)){let items=box.querySelectorAll('.source-detail-item');if(items[i]){let img=document.createElement('img');img.src=d.og_image||d.img;img.loading='lazy';img.onerror=function(){this.style.display='none'};items[i].prepend(img);}}}).catch(()=>{});});}
 
-// === 2) Patch AI wall read to show source_details + rewrite button ===
+// === AI Wall read with source_details + rewrite + ask AI ===
 async function readAIWallPost(i){
 let wall=(await fetch('/api/ai_wall').then(r=>r.json()).catch(()=>({posts:[]}))).posts||[];
-let p=wall[i];if(!p)return;
-showView('view-article');
-let h=`<button class="back-btn" onclick="switchCat('home')">← Quay lại</button><div class="article-view" data-post-id="${p.id}"><span class="badge badge-ai">AI</span><h1 class="article-title">${esc(p.title)}</h1>${p.img?`<img class="article-img" src="${p.img}">`:''}`;
+let p=wall[i];if(!p)return;showView('view-article');
+let h=`<button class="back-btn" onclick="switchCat('home')">← Quay lại</button><div class="article-view"><span class="badge badge-ai">AI</span><h1 class="article-title">${esc(p.title)}</h1>${p.img?`<img class="article-img" src="${p.img}">`:''}`;
 if(p.sources&&p.sources.length){h+=`<div class="article-summary"><b>Nguồn:</b> ${p.sources.map(s=>esc(s.via||s.title||'')).join(', ')}</div>`;}
 h+=`<p class="article-p" style="white-space:pre-wrap">${esc(p.text)}</p>`;
-h+=`<div class="article-actions"><button class="primary" onclick="rewriteTopicPost(${i})">🤖 AI viết lại & đăng tường</button><button onclick="doShare('${esc(p.title)}','${location.origin}/aw?post=${p.id}','${esc(p.img||'')}')">📤 Chia sẻ</button></div>`;
+let firstUrl=(p.source_details||[]).find(s=>s.url)?.url||(p.sources||[]).find(s=>s.url)?.url||'';
+h+=`<div class="article-actions">${firstUrl?`<button class="primary" onclick="rewriteFromUrl('${esc(firstUrl)}')">🤖 Tóm tắt AI đăng tường</button>`:''}<button onclick="doShare('${esc(p.title)}','${location.origin}','${esc(p.img||'')}')">📤 Chia sẻ</button></div>`;
+h+=`<div class="article-ai-ask"><h3 style="font-size:14px;color:#5cb87a;margin-bottom:6px">🤖 Hỏi AI về bài viết</h3><textarea id="article-ai-q" placeholder="Hỏi về nội dung bài viết này..."></textarea><button onclick="askAIWall(${i})">Hỏi AI</button><div id="article-ai-ans" class="article-ai-answer"></div></div>`;
 h+=`</div>`;
 document.getElementById('view-article').innerHTML=h;
-// Render source details from saved data.
-let art=document.querySelector('.article-view');
-if(art&&p.source_details&&p.source_details.length)renderSourceDetails(p,art);
+let art=document.querySelector('.article-view');if(art&&p.source_details)renderSourceDetails(p,art);
 window.scrollTo(0,0);}
 
-window.readAIWallPost=readAIWallPost;
-window.aiReadWallPatched=window.aiReadWall=window.readWallPost=function(i){readAIWallPost(i)};
+window.readAIWallPost=readAIWallPost;window.aiReadWallPatched=window.aiReadWall=window.readWallPost=function(i){readAIWallPost(i)};
 
-// === 3) Rewrite topic post using its source URLs ===
-window.rewriteTopicPost=async function(i){
+window.rewriteFromUrl=async function(url){
+if(!url)return alert('Không có URL nguồn');
+let btn=event?.target;if(btn){btn.disabled=true;btn.textContent='Đang tóm tắt...';}
+try{let r=await fetch('/api/url_wall',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url})});let j=await r.json();if(!r.ok||j.error)throw new Error(j.error||'Lỗi');alert('Đã tóm tắt và đăng lên Tường AI.');}catch(e){alert(e.message)}finally{if(btn){btn.disabled=false;btn.textContent='🤖 Tóm tắt AI đăng tường';}}};
+
+window.askAIWall=async function(i){
+let q=document.getElementById('article-ai-q')?.value.trim();if(!q)return alert('Nhập câu hỏi');
+let ans=document.getElementById('article-ai-ans');ans.textContent='Đang hỏi AI...';
 let wall=(await fetch('/api/ai_wall').then(r=>r.json()).catch(()=>({posts:[]}))).posts||[];
-let p=wall[i];if(!p)return alert('Bài không tồn tại');
-let urls=(p.source_details||[]).map(s=>s.url).filter(Boolean);
-if(!urls.length&&p.sources)urls=p.sources.map(s=>s.url).filter(Boolean);
-if(!urls.length)return alert('Bài này không có link nguồn để rewrite.');
-let btn=event?.target;if(btn){btn.disabled=true;btn.textContent='Đang rewrite...';}
-try{
-let r=await fetch('/api/url_wall',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url:urls[0]})});
-let j=await r.json();
-if(!r.ok||j.error)throw new Error(j.error||'Lỗi');
-alert('Đã rewrite bài nguồn đầu tiên và đăng lên Tường AI.');
-if(typeof loadPatchedWall==='function')loadPatchedWall();
-}catch(e){alert(e.message)}finally{if(btn){btn.disabled=false;btn.textContent='🤖 AI viết lại & đăng tường';}}}
-
-// === 4) Short comments: persistent + display ===
+let p=wall[i]||{};let context=p.text||'';let url=(p.source_details||[]).find(s=>s.url)?.url||(p.sources||[]).find(s=>s.url)?.url||'';
+try{let r=await fetch('/api/article/ask',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url,question:q,context})});let j=await r.json();ans.textContent=j.answer||'Không có trả lời';}catch(e){ans.textContent='Lỗi: '+e.message}};
+
+// === Short comments ===
 window.openShortComments=async function(id){
 let panel=document.getElementById('short-cmt-panel');
 let j=await fetch('/api/short/comments?id='+encodeURIComponent(id)).then(r=>r.json()).catch(()=>({comments:[]}));
 let cmts=j.comments||[];
 panel.innerHTML=`<h3 style="color:#5cb87a;font-size:14px">💬 Bình luận</h3><div id="cmt-list">${cmts.map(c=>`<div class="cmt-item">${esc(c.text)}</div>`).join('')||'<div class="cmt-item" style="color:#777">Chưa có bình luận</div>'}</div><textarea id="cmt-text" placeholder="Nhập bình luận..."></textarea><button onclick="submitShortCmt('${esc(id)}')">Gửi</button><button onclick="document.getElementById('short-cmt-panel').classList.remove('active')">Đóng</button>`;
 panel.classList.add('active');}
-
 window.submitShortCmt=async function(id){
 let text=document.getElementById('cmt-text')?.value.trim();if(!text)return;
 let j=await fetch('/api/short/comment',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({id,text})}).then(r=>r.json()).catch(()=>({comments:[]}));
-let cmts=j.comments||[];
-document.getElementById('cmt-list').innerHTML=cmts.map(c=>`<div class="cmt-item">${esc(c.text)}</div>`).join('');
+document.getElementById('cmt-list').innerHTML=(j.comments||[]).map(c=>`<div class="cmt-item">${esc(c.text)}</div>`).join('');
 document.getElementById('cmt-text').value='';}
 
-// === 5) Patch shorts feed to show comment count + open panel ===
-function patchShortCommentButtons(){
-document.querySelectorAll('.tiktok-slide').forEach(sl=>{
-if(sl.dataset.cmtPatched)return;sl.dataset.cmtPatched='1';
-let id=sl.dataset.id||'';if(!id)return;
-let right=sl.querySelector('.tiktok-right,.short-action-panel');
-if(!right)return;
-let existing=right.querySelector('[data-cmt-btn]');if(existing)return;
-let btn=document.createElement('button');btn.className='short-action-btn';btn.setAttribute('data-cmt-btn','1');
-btn.innerHTML='<div class="ico" style="width:42px;height:42px;border-radius:50%;background:rgba(0,0,0,.45);display:flex;align-items:center;justify-content:center;font-size:20px">💬</div><span>BL</span>';
-btn.onclick=function(e){e.stopPropagation();openShortComments(id);};
-right.appendChild(btn);});}
-
-// === 6) Add rewrite to regular articles ===
-function addRewriteButton(){let art=document.querySelector('#view-article .article-view');if(!art||art.querySelector('[data-rewrite]'))return;let actions=art.querySelector('.article-actions');if(!actions){actions=document.createElement('div');actions.className='article-actions';art.appendChild(actions);}let btn=document.createElement('button');btn.className='primary';btn.setAttribute('data-rewrite','1');btn.textContent='🤖 AI viết lại & đăng tường';btn.onclick=function(){if(typeof rewriteCurrentArticle==='function')rewriteCurrentArticle();else alert('Chức năng rewrite chưa sẵn sàng.')};actions.insertBefore(btn,actions.firstChild);}
-
-// === 7) Source links open in-app ===
-function patchSourceLinksInApp(){document.querySelectorAll('.source-detail-item a[onclick]').forEach(a=>{});/* Already using onclick in renderSourceDetails */}
-
-// === 8) Patch readArticle for rewrite button ===
-let oldReadArticle=window.readArticle;
-if(oldReadArticle){window.readArticle=async function(){let ret=await oldReadArticle.apply(this,arguments);setTimeout(addRewriteButton,600);return ret;}}
-
-// === 9) Defer hot topics ===
+// === Patch shorts to add comment button ===
+function patchShortButtons(){document.querySelectorAll('.tiktok-slide').forEach(sl=>{if(sl.dataset.cmtDone)return;sl.dataset.cmtDone='1';let id=sl.dataset.id||'';if(!id)return;let right=sl.querySelector('.tiktok-right,.short-action-panel');if(!right)return;if(right.querySelector('[data-cmt]'))return;let btn=document.createElement('button');btn.className='short-action-btn';btn.setAttribute('data-cmt','1');btn.innerHTML='<div class="ico" style="width:42px;height:42px;border-radius:50%;background:rgba(0,0,0,.45);display:flex;align-items:center;justify-content:center;font-size:20px">💬</div><span>BL</span>';btn.onclick=function(e){e.stopPropagation();openShortComments(id);};right.appendChild(btn);});}
+
+// === Only ONE rewrite button for regular articles (not duplicate) ===
+function addSingleRewriteBtn(){let art=document.querySelector('#view-article .article-view');if(!art)return;
+// Remove any duplicates first.
+let btns=art.querySelectorAll('[data-rewrite]');if(btns.length>1){for(let i=1;i<btns.length;i++)btns[i].remove();}
+if(btns.length>=1)return;
+let actions=art.querySelector('.article-actions');if(!actions)return;
+let btn=document.createElement('button');btn.className='primary';btn.setAttribute('data-rewrite','1');btn.textContent='🤖 Tóm tắt AI đăng tường';
+btn.onclick=function(){if(typeof rewriteCurrentArticle==='function')rewriteCurrentArticle();else alert('Chức năng rewrite chưa sẵn sàng.')};
+actions.insertBefore(btn,actions.firstChild);}
+
+// === Ask AI box in regular articles ===
+function addArticleAskBox(){let art=document.querySelector('#view-article .article-view');if(!art||art.querySelector('.article-ai-ask'))return;
+let box=document.createElement('div');box.className='article-ai-ask';
+box.innerHTML='<h3 style="font-size:14px;color:#5cb87a;margin-bottom:6px">🤖 Hỏi AI về bài viết</h3><textarea id="article-ai-question" placeholder="Hỏi về nội dung bài viết..."></textarea><button onclick="askArticleAI()">Hỏi AI</button><div id="article-ai-answer" class="article-ai-answer"></div>';
+art.appendChild(box);}
+window.askArticleAI=async function(){let q=document.getElementById('article-ai-question')?.value.trim();if(!q)return alert('Nhập câu hỏi');let ans=document.getElementById('article-ai-answer');ans.textContent='Đang hỏi AI...';let url=(window._currentArticle&&window._currentArticle.url)||'';let context=document.querySelector('.article-view')?.innerText?.slice(0,12000)||'';try{let r=await fetch('/api/article/ask',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url,question:q,context})});let j=await r.json();ans.textContent=j.answer||'Không có trả lời';}catch(e){ans.textContent='Lỗi: '+e.message}}
+
+// === Patch readArticle ===
+let oldRA=window.readArticle;
+if(oldRA){window.readArticle=async function(){let ret=await oldRA.apply(this,arguments);setTimeout(()=>{addSingleRewriteBtn();addArticleAskBox();},600);return ret;}}
+
+// === Defer non-critical ===
 let _hotLoaded=false;function deferHot(){if(_hotLoaded)return;_hotLoaded=true;setTimeout(()=>{if(typeof ensureHotTopics==='function')ensureHotTopics();if(typeof ensureNewsShortsHome==='function')ensureNewsShortsHome();},4000);}
 if(document.readyState==='complete')deferHot();else window.addEventListener('load',deferHot);
 
-setInterval(()=>{addRewriteButton();patchShortCommentButtons();},1500);
-setTimeout(()=>{addRewriteButton();patchShortCommentButtons();},1000);
+setInterval(()=>{addSingleRewriteBtn();patchShortButtons();addArticleAskBox();},1500);
 })();
 </script>
 '''