Restore b9b6ba4 with safe runtime patch for topic cleanup and fast shorts

restore_b9b6ba4_runner.py

@@ -1,6 +1,5 @@
 import os
 import sys
-import shutil
 from huggingface_hub import snapshot_download
 
 REVISION = "b9b6ba4"
@@ -14,18 +13,180 @@ snapshot_dir = snapshot_download(
     local_dir_use_symlinks=False,
 )
 
-# Copy patch from current repo into the restored snapshot.
-patch_src = "/app/patch_b9b6ba4.py"
-patch_dst = os.path.join(snapshot_dir, "patch_b9b6ba4.py")
-if os.path.exists(patch_src):
-    shutil.copyfile(patch_src, patch_dst)
-
 os.chdir(snapshot_dir)
 sys.path.insert(0, snapshot_dir)
 
+# Runtime patch is written into the restored snapshot so the restored UI is used
+# (no placeholder index.html), while only the requested fixes are applied.
+patch_code = r'''
+import re, time, json, asyncio
+from fastapi import Request, Query
+from fastapi.responses import JSONResponse, HTMLResponse
+import ai_runtime_final6 as m
+
+app = m.app
+_SHORTS_CACHE = {"t": 0, "d": []}
+BAD_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',
+    'Bài có tiêu đề, sapo',
+    'Nhiệm vụ: viết một bài báo tiếng Việt hoàn chỉnh',
+]
+
+def _clean_topic_text(txt):
+    txt = str(txt or '')
+    out = []
+    for line in txt.splitlines():
+        s = line.strip()
+        if not s:
+            out.append(line); continue
+        low = s.lower()
+        if any(x.lower() in low for x in BAD_LINES):
+            continue
+        if re.match(r'^nhiệm vụ\s*:', low):
+            continue
+        # Drop prompt-like bullet instructions that accidentally leak into output.
+        if re.match(r'^[•\-*]\s*(chỉ xuất bản|không sao chép|bài có tiêu đề|nhiệm vụ)', low):
+            continue
+        out.append(line)
+    txt = '\n'.join(out)
+    txt = re.sub(r'\n{3,}', '\n\n', txt).strip()
+    return txt
+
+def _fallback_shorts():
+    out=[];seen=set()
+    try:
+        for v in getattr(m.f5.rt if hasattr(m,'f5') else m, 'SHORTS_FALLBACK', []) or []:
+            vid=v.get('id')
+            if vid and vid not in seen:
+                seen.add(vid);out.append(v)
+    except Exception:
+        pass
+    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')
+    ]
+    for vid,title,ch in hard:
+        if vid not in seen:
+            seen.add(vid);out.append({'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'})
+    return out
+
+def _fresh_shorts_fast():
+    seen=set();out=[]
+    handles=getattr(m, 'YOUTUBE_HANDLES', ['baodantri7941','baosuckhoedoisongboyte'])
+    for h in handles:
+        got=[]
+        try: got = m._yt_html(h, 24) or []
+        except Exception: got=[]
+        # yt-dlp can be slower; only use if html returns too few.
+        if len(got)<4:
+            try: got = (m._yt_ytdlp(h, 24) or []) + got
+            except Exception: pass
+        for v in got:
+            vid=v.get('id') or ''
+            if vid and vid not in seen:
+                seen.add(vid);out.append(v)
+    for v in _fallback_shorts():
+        vid=v.get('id') or ''
+        if vid and vid not in seen:
+            seen.add(vid);out.append(v)
+    return out[:50]
+
+# Remove only routes we override.
+_PATCH={('/api/topic_post','POST'),('/api/shorts','GET'),('/','GET')}
+app.router.routes=[r for r in app.router.routes if not any(getattr(r,'path',None)==p and method in getattr(r,'methods',set()) for p,method in _PATCH)]
+
+@app.get('/api/shorts')
+def api_shorts_patch(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_fast()
+    _SHORTS_CACHE.update({'t':now,'d':data})
+    return JSONResponse(data)
+
+@app.post('/api/topic_post')
+async def topic_post_patch(request:Request):
+    body=await request.json(); topic=m.clean(body.get('topic',''))
+    if not topic: return JSONResponse({'error':'missing topic'},status_code=400)
+    img=m._topic_image(topic)
+    # Use fast context if available; fall back safely.
+    try:
+        research=m._fast_context(topic)
+    except Exception:
+        research=m._web_research_context(topic)
+    context=research.get('context',''); sources=research.get('sources',[])
+    source_brief=context[:18000]
+    prompt=f'''Bạn là biên tập viên VNEWS. Hãy viết một bài báo tiếng Việt hoàn chỉnh về chủ đề: "{topic}" dựa trên dữ liệu nguồn bên dưới.
+
+DỮ LIỆU NGUỒN:
+{source_brief}
+
+YÊU CẦU NỘI DUNG:
+- Viết trực tiếp bài cuối cùng, không nhắc lại yêu cầu hay chỉ dẫn.
+- Tổng hợp và diễn đạt lại bằng ngôn ngữ báo chí tự nhiên.
+- Có tiêu đề mới, sapo ngắn, các đoạn bối cảnh/phân tích/tác động.
+- Không sao chép nguyên văn nguồn.
+- Cuối bài có mục Nguồn tham khảo ngắn.
+'''
+    try:
+        text=await asyncio.wait_for(m.f5.base.qwen_generate(prompt,image_url=img,max_tokens=1700),timeout=35)
+    except Exception:
+        text=None
+    if not text or len(str(text))<300:
+        vias=', '.join(sorted({s.get('via','') for s in sources if s.get('via')}))
+        titles='\n'.join('• '+s.get('title','') for s in sources[:6])
+        text=f'{topic}: những điểm đáng chú ý\n\n{topic} đang thu hút sự quan tâm qua nhiều nguồn tin. Tổng hợp nhanh cho thấy chủ đề này cần được nhìn từ bối cảnh, tác động và những diễn biến liên quan.\n\n{titles}\n\nNguồn tham khảo: {vias}'
+    text=_clean_topic_text(text)
+    src=[s for s in sources if s.get('url')]
+    post=m.f5.base.make_post(topic,text,img,'','topic_cleaned_b9b6ba4',sources=src)
+    post['images']=[img]
+    posts=m.f5.base._load_ai_wall(); posts.insert(0,post); m.f5.base._save_ai_wall(posts)
+    return JSONResponse({'post':post})
+
+EXTRA_JS = r'''
+<style>
+.topic-extra-actions{display:flex;gap:8px;flex-wrap:wrap;margin-top:10px}.topic-extra-actions button{background:#2d8659;border:0;color:#fff;border-radius:12px;padding:8px 12px;font-size:12px}.topic-extra-actions button.secondary{background:#222;border:1px solid #444}.topic-source-links{margin-top:10px;background:#151515;border:1px solid #2b2b2b;border-radius:10px;padding:10px}.topic-source-links a{display:block;color:#5cb87a;font-size:12px;margin:5px 0;text-decoration:none}.shorts-b9-wrap{margin:6px 4px;background:#1a1a1a;border:1px solid #2a2a2a;border-radius:8px;overflow:hidden}
+</style>
+<script>
+(function(){
+function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
+async function ensureFastShorts(){let home=document.getElementById('view-home');if(!home||document.getElementById('shorts-b9-fast'))return;let data=await fetch('/api/shorts').then(r=>r.json()).catch(()=>[]);if(!data.length)return;let wrap=document.createElement('div');wrap.id='shorts-b9-fast';wrap.className='shorts-b9-wrap';let h='<div class="slider-header"><span class="slider-label">📱 Shorts Dân trí & SKĐS</span><span class="slider-note">Tải nhanh</span></div><div class="slider-track">';data.slice(0,24).forEach((a,i)=>{h+=`<div class="slider-item shorts-item" onclick="openTikTok('shorts',${i})"><div class="slider-thumb shorts-thumb">${a.img?`<img src="${esc(a.img)}">`:''}<div class="card-play">▶</div></div><div class="slider-title">${esc(a.title)}</div></div>`});h+='</div>';wrap.innerHTML=h;let comp=document.querySelector('.ai-compose')||home.firstChild;if(comp)comp.after(wrap);else home.prepend(wrap);}
+function addTopicActions(){let art=document.querySelector('#view-article.active .article-view');if(!art||document.getElementById('topic-extra-actions'))return;let text=art.innerText||'';if(!/Nguồn tham khảo|Qwen|Tường AI|AI/i.test(text))return;let box=document.createElement('div');box.id='topic-extra-actions';box.className='topic-extra-actions';box.innerHTML='<button onclick="rewriteTopicArticleB9()">🤖 Rewrite AI & đăng tường</button><button class="secondary" onclick="openFirstTopicSourceB9()">🔗 Mở nguồn đầu tiên</button><div id="topic-source-links" class="topic-source-links" style="display:none"></div>';art.appendChild(box);}
+window.rewriteTopicArticleB9=function(){let title=document.querySelector('.article-title')?.innerText||'Bài chủ đề';let content=document.querySelector('.article-view')?.innerText||'';let payload={topic:title+'\n'+content.slice(0,2500)};fetch('/api/topic_post',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify(payload)}).then(r=>r.json()).then(j=>{if(j.post)alert('Đã rewrite AI và đăng lại lên Tường AI');else alert(j.error||'Lỗi rewrite')}).catch(e=>alert(e.message));};
+window.openFirstTopicSourceB9=function(){let a=document.querySelector('.article-summary a,.source-detail-box a,.topic-source-links a');if(a)a.click();else alert('Bài này chưa có link nguồn hiển thị.');};
+setInterval(()=>{ensureFastShorts();addTopicActions();},1500);setTimeout(()=>{ensureFastShorts();addTopicActions();},800);
+})();
+</script>
+'''
+
+@app.get('/')
+async def index_patch():
+    # Call restored final6 root if available to keep 100% restored UI and avoid placeholder.
+    try:
+        resp=await m.index_final6_fast_home()
+    except Exception:
+        try: resp=await m.index_final6()
+        except Exception:
+            html=m.f5.f4.f3.f2.f1._load_index_html()
+            return HTMLResponse(html.replace('</body>',EXTRA_JS+'</body>') if '</body>' in html else html+EXTRA_JS)
+    html=resp.body.decode('utf-8') if hasattr(resp,'body') else str(resp)
+    return HTMLResponse(html.replace('</body>',EXTRA_JS+'\n</body>') if '</body>' in html else html+EXTRA_JS)
+'''
+
+patch_path=os.path.join(snapshot_dir,'b9b6ba4_safe_patch.py')
+with open(patch_path,'w',encoding='utf-8') as f:
+    f.write(patch_code)
+
 cmd = [
     "uvicorn",
-    "patch_b9b6ba4:app",
+    "b9b6ba4_safe_patch:app",
     "--host",
     "0.0.0.0",
     "--port",