Fix: add /api/topic/rewrite handler and natural AI ask prompt

ai_runtime_patch_fast.py

@@ -51,7 +51,9 @@ def _bg_refresh():
 
 @app.on_event("startup")
 async def startup_prefetch():threading.Thread(target=_bg_refresh,daemon=True).start()
-threading.Thread(target=lambda:[time.sleep(600) or _bg_refresh() for _ in iter(int,1)],daemon=True).start()
+def _periodic():
+    while True:time.sleep(600);_bg_refresh()
+threading.Thread(target=_periodic,daemon=True).start()
 
 # ===== OVERRIDE ENDPOINTS =====
 app.router.routes=[r for r in app.router.routes if not (
@@ -104,7 +106,7 @@ async def post_short_comment(request:Request):
     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 — natural, friendly, based on content =====
+# ===== ARTICLE ASK AI — natural and friendly =====
 @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',''))
@@ -117,49 +119,49 @@ async def article_ask(request:Request):
         except Exception:pass
     if not raw and context:raw=context[:12000]
     if not raw:raw=question
-    prompt=f"""Bạn là trợ lý thông minh của VNEWS. Người dùng đang đọc một bài viết/video và muốn hỏi bạn.
+    prompt=f"""Bạn là trợ lý thông minh của VNEWS. Người dùng đang đọc một bài viết/xem video và muốn hỏi bạn.
 
-Nội dung bài viết/video:
+Nội dung bài viết/video mà người dùng đang xem:
+---
 Tiêu đề: {title}
-Nội dung: {raw[:10000]}
+{raw[:10000]}
+---
 
-Câu hỏi của người dùng: {question}
+Câu hỏi: {question}
 
-Hãy trả lời bằng tiếng Việt một cách TỰ NHIÊN, thân thiện và hữu ích:
-- Dựa vào nội dung bài viết/video ở trên làm ngữ cảnh chính.
-- Giải thích rõ ràng, dễ hiểu như đang trò chuyện.
-- Nếu câu hỏi liên quan đến bài, hãy trả lời chi tiết dựa trên thông tin trong bài.
-- Nếu câu hỏi vượt ra ngoài bài nhưng liên quan đến chủ đề, bạn vẫn có thể bổ sung kiến thức chung để giúp người đọc hiểu hơn, nhưng nói rõ đó là kiến thức bổ sung.
-- Tránh trả lời kiểu "bài viết không đề cập" một cách khô khan. Thay vào đó hãy nói "Trong bài này chưa nói đến phần đó, nhưng theo những gì tôi biết thì..." nếu bạn có thể bổ sung.
+Hãy trả lời bằng tiếng Việt, tự nhiên và thân thiện như đang trò chuyện:
+- Lấy thông tin từ nội dung bài viết/video ở trên làm cơ sở chính.
+- Giải thích dễ hiểu, có ví dụ nếu cần.
+- Nếu câu hỏi liên quan đến bài → trả lời chi tiết từ nội dung bài.
+- Nếu câu hỏi mở rộng hơn → bổ sung kiến thức chung nhưng nói rõ "ngoài bài viết, theo hiểu biết chung thì..."
+- Không trả lời khô khan, máy móc. Hãy viết như một người bạn am hiểu đang giải thích.
 """
     ans=await f5.base.qwen_generate(prompt,max_tokens=1200)
     if not ans:ans='AI chưa trả lời được lúc này. Bạn thử hỏi lại nhé.'
     return JSONResponse({'answer':ans,'title':title})
 
-# ===== TOPIC REWRITE (dedicated endpoint for topic posts) =====
+# ===== TOPIC REWRITE — takes post_id, rewrites using all source content =====
 @app.post('/api/topic/rewrite')
 async def topic_rewrite(request:Request):
-    """Rewrite/summarize a topic post's content and source articles into a new wall post."""
     body=await request.json();post_id=str(body.get('post_id','')).strip()
     if not post_id:return JSONResponse({'error':'missing post_id'},status_code=400)
     posts=f5.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)
-    # Collect all available content: post text + source details content
-    all_content=p.get('text') or ''
+    if not p:return JSONResponse({'error':'Không tìm thấy bài viết.'},status_code=404)
+    all_content=(p.get('text') or '')
     for s in (p.get('source_details') or []):
-        all_content+='\n\n'+s.get('content','')
+        all_content+='\n\n--- Nguồn: '+s.get('title','')+' ---\n'+s.get('content','')
     title=p.get('title') or 'Bài viết AI'
-    prompt=f"""Bạn là biên tập viên VNEWS. Hãy VIẾT LẠI bài dưới đây thành một bản tóm tắt mới, ngắn gọn, dễ đọc, đăng lên Tường AI.
+    prompt=f"""Bạn là biên tập viên VNEWS. Viết lại nội dung bên dưới thành BẢN TÓM TẮT MỚI, hấp dẫn, đăng lên Tường AI.
 
 Tiêu đề gốc: {title}
-Nội dung gốc và nguồn:
+Nội dung gốc và nguồn tham khảo:
 {all_content[:16000]}
 
 Yêu cầu:
-- Viết lại bằng ngôn ngữ tự nhiên, mạch lạc.
-- Giữ đúng sự thật, không bịa.
+- Viết lại tự nhiên, mạch lạc, không sao chép nguyên văn.
 - 1 tiêu đề mới hấp dẫn.
-- 2 đoạn ngắn hoặc 4-5 ý chính.
+- 1 đoạn mở đầu + 3-5 ý chính.
+- Giữ sự thật, không bịa.
 - Cuối bài ghi nguồn tham khảo.
 """
     text=None
@@ -169,7 +171,7 @@ Yêu cầu:
         text=f"Tóm tắt: {title}\n\n{all_content[:1500]}\n\nNguồn tham khảo: VNEWS AI"
     new_post=f5.base.make_post('Rewrite: '+title,text,p.get('img',''),'','rewrite_topic',sources=p.get('sources',[]))
     new_post['images']=p.get('images',[])
-    posts=f5.base._load_ai_wall();posts.insert(0,new_post);f5.base._save_ai_wall(posts)
+    all_posts=f5.base._load_ai_wall();all_posts.insert(0,new_post);f5.base._save_ai_wall(all_posts)
     return JSONResponse({'post':new_post})
 
 # ===== TOPIC POST =====
@@ -235,7 +237,6 @@ h+=`</div>`;
 document.getElementById('view-article').innerHTML=h;
 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.rewriteTopicPost=async function(postId){
@@ -247,12 +248,10 @@ 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]||{};let context=(p.text||'');
-for(let s of (p.source_details||[]))context+='\n'+s.content;
+let p=wall[i]||{};let context=(p.text||'');for(let s of (p.source_details||[]))context+='\n'+s.content;
 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:context.slice(0,12000)})});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:[]}));
@@ -267,7 +266,6 @@ document.getElementById('cmt-text').value='';}
 
 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);});}
 
-// === Single rewrite + ask AI for regular articles ===
 function addSingleRewriteBtn(){let art=document.querySelector('#view-article .article-view');if(!art)return;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);}
 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 bất cứ điều gì về nội dung này..."></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}}