Clean rewrite text, remove junk phrases, better prompt"

app_entry.py

@@ -1,4 +1,4 @@
-"""Wrapper: kill old renderers FIRST, then inject everything."""
+"""Wrapper: kill old renderers FIRST, highlight TikTok feed, clean rewrite, single wall."""
 from ai_runtime_patch_fast import *
 from ai_runtime_patch_fast import app, f5, f6, rt, PATCH_INJECT, _scrape, _domain, clean, _bg, _bg_home, _bg_shorts
 from patch_extra import EXTRA_FIX
@@ -25,7 +25,6 @@ def _homepage_fast():
         return JSONResponse(_bg_home['d'])
     threading.Thread(target=_bg,daemon=True).start()
     return JSONResponse([])
-
 @app.get('/api/shorts')
 def _shorts_fast(refresh:int=Query(default=0)):
     if _bg_shorts['d']:
@@ -41,6 +40,15 @@ def _extract_title(text):
         first=re.sub(r'^[#*\-•\d\.\)\s]+','',lines[0]).strip()
         if 10<=len(first)<=120:return first
     return lines[0][:100] if lines else 'Bài viết AI'
+
+def _clean_rewrite_text(text):
+    """Remove junk phrases from AI rewrite output."""
+    if not text:return text
+    junk=['xem trên vnews','xem trên VNEWS','Xem trên VNEWS','📖 Xem trên VNEWS','đọc trên vnews','Đọc trên VNEWS','📖 Đọc trên','Mở nguồn gốc','mở nguồn gốc']
+    for j in junk:text=text.replace(j,'')
+    text=re.sub(r'\n{3,}','\n\n',text)
+    return text.strip()
+
 def _source_image(sources, details):
     for s in (details or []):
         if s.get('url'):
@@ -70,8 +78,10 @@ async def _tp(request:Request):
     try:text=await asyncio.wait_for(f5.base.qwen_generate(f'Viết bài tiếng Việt VỀ: "{topic}"\nNGUỒN:\n{sb[:18000]}\nCHỈ viết về "{topic}". 5-8 đoạn. Cuối có nguồn.',image_url=img,max_tokens=1700),timeout=35)
     except:pass
     if not text or len(text)<300:text=f"{topic}: tổng hợp\n\n"+'\n'.join([f"• {d['title']}: {d.get('content','')[:300]}" for d in (det or [])[:6]])+"\n\nNguồn: "+', '.join(sorted({d.get('via','') for d in (det or []) if d.get('via')}))
+    text=_clean_rewrite_text(text)
     post=f5.base.make_post(topic,text,img,'','topic_focused',sources=[s for s in src if s.get('url')]);post['images']=[img];post['source_details']=det
     ps=f5.base._load_ai_wall();ps.insert(0,post);f5.base._save_ai_wall(ps);return JSONResponse({'post':post})
+
 @app.post('/api/rewrite_share')
 @app.post('/api/url_wall')
 async def _rw(request:Request):
@@ -81,14 +91,28 @@ async def _rw(request:Request):
     if len(raw)<50:raw=ctx[:14000]
     if len(raw)<50:return JSONResponse({'error':'Không đọc được bài'},status_code=422)
     img=_ensure_img(img)
+    prompt=f"""Tóm tắt bài viết thành bản tin ngắn gọn. Dòng đầu tiên phải là tiêu đề mới hấp dẫn.
+
+Tiêu đề gốc (tham khảo): {title}
+Nội dung bài:
+{raw[:14000]}
+
+Yêu cầu:
+- Dòng 1: Tiêu đề MỚI ngắn gọn, hấp dẫn (không copy tiêu đề gốc).
+- Tiếp: 4-6 ý chính tóm tắt nội dung.
+- Cuối: ghi nguồn.
+- KHÔNG viết "xem trên VNEWS", "đọc trên VNEWS", "mở nguồn gốc" hay bất kỳ cụm điều hướng nào."""
     text=None
-    try:text=await asyncio.wait_for(f5.base.qwen_generate(f'Tóm tắt đăng Tường AI. Dòng 1: TIÊU ĐỀ MỚI.\n\nTiêu đề gốc: {title}\nNội dung:\n{raw[:14000]}\n\nDòng 1: Tiêu đề mới. Tiếp: 4-6 ý. Cuối: nguồn.',image_url=img,max_tokens=1000),timeout=30)
+    try:text=await asyncio.wait_for(f5.base.qwen_generate(prompt,image_url=img,max_tokens=1000),timeout=30)
     except:pass
     if not text or len(text)<80:text=f"{title}\n\n{raw[:1200]}\n\nNguồn: {_domain(url)}"
+    text=_clean_rewrite_text(text)
     ai_title=_extract_title(text);lines=text.strip().split('\n')
     body_text='\n'.join(lines[1:]).strip() if lines and lines[0].strip()==ai_title else text
+    body_text=_clean_rewrite_text(body_text)
     post=f5.base.make_post(ai_title,body_text,img,url,'rewrite',sources=[{'title':title,'url':url,'via':_domain(url)}])
     ps=f5.base._load_ai_wall();ps.insert(0,post);f5.base._save_ai_wall(ps);return JSONResponse({'post':post})
+
 @app.post('/api/topic/rewrite')
 async def _tr(request:Request):
     b=await request.json();pid=str(b.get('post_id','')).strip()
@@ -103,43 +127,43 @@ async def _tr(request:Request):
         if not best_img and uimg and len(uimg)>20:best_img=uimg
     ac='\n---\n'.join(parts) if parts else (p.get('text') or '')
     img=_ensure_img(best_img or p.get('img',''))
+    prompt=f"""Viết lại thành bản tóm tắt mới. Dòng đầu là tiêu đề mới hấp dẫn.
+
+Chủ đề gốc: {p.get('title','')}
+Nguồn:
+{ac[:16000]}
+
+Yêu cầu:
+- Dòng 1: Tiêu đề MỚI.
+- Tiếp: 4-6 ý chính.
+- Cuối: nguồn tham khảo.
+- KHÔNG viết "xem trên VNEWS" hay bất kỳ cụm điều hướng nào."""
     text=None
-    try:text=await asyncio.wait_for(f5.base.qwen_generate(f'Viết lại tóm tắt. Dòng 1: TIÊU ĐỀ MỚI.\n\nChủ đề: {p.get("title","")}\nNguồn:\n{ac[:16000]}\n\nDòng 1: Tiêu đề mới. Tiếp: 4-6 ý. Cuối: nguồn.',image_url=img,max_tokens=1200),timeout=35)
+    try:text=await asyncio.wait_for(f5.base.qwen_generate(prompt,image_url=img,max_tokens=1200),timeout=35)
     except:pass
     if not text or len(text)<100:text=f"Tóm tắt: {p.get('title','')}\n\n{ac[:1500]}\n\nNguồn: VNEWS AI"
+    text=_clean_rewrite_text(text)
     ai_title=_extract_title(text);lines=text.strip().split('\n')
     body_text='\n'.join(lines[1:]).strip() if lines and lines[0].strip()==ai_title else text
+    body_text=_clean_rewrite_text(body_text)
     np=f5.base.make_post(ai_title,body_text,img,'','rewrite_topic',sources=p.get('sources',[]));np['images']=[img]
     all_p=f5.base._load_ai_wall();all_p.insert(0,np);f5.base._save_ai_wall(all_p);return JSONResponse({'post':np})
 
-# === CRITICAL: Script that runs BEFORE all old injects to prevent slide destruction ===
 PRE_KILL_SCRIPT = r'''
 <script>
-// PRE-KILL: Override destructive functions BEFORE old inject code defines and calls them
 Object.defineProperty(window,'renderTopicWallE',{get:function(){return function(){}},set:function(){},configurable:true});
 Object.defineProperty(window,'renderAIShortHome',{get:function(){return function(){}},set:function(){},configurable:true});
 Object.defineProperty(window,'renderAIShorts7',{get:function(){return function(){}},set:function(){},configurable:true});
 </script>
 '''
 
-HIGHLIGHT_FIX = r'''
-<style>
-.vp-wrap{aspect-ratio:1/1!important;width:100%!important;max-height:100vh!important;background:#000!important;position:relative!important}
-.vp-wrap video,.vp-wrap iframe{width:100%!important;height:100%!important;object-fit:contain!important}
-.vp-wrap .vp-ratio-btn{display:none!important}
-</style>
-'''
-
 @app.get('/')
 async def _index_final():
     html=f5.f4.f3.f2.f1._load_index_html()
-    # PRE-KILL goes FIRST before any old inject can run
     body=PRE_KILL_SCRIPT
     body+=getattr(rt.old,'PATCH_INJECT','')+f5.f4.f3.f2.f1.FINAL_INJECT+f5.f4.f3.FINAL3_INJECT+f5.f4.FINAL4_INJECT+f5.FINAL5_INJECT
     body+=getattr(f6,'FINAL6_INJECT','')
     body+=getattr(f6,'FINAL6_FAST_HOME_INJECT','')
-    # NO FINAL6E_INJECT
     body+=PATCH_INJECT
     body+=EXTRA_FIX
-    body+=HIGHLIGHT_FIX
     return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)