Final runtime overrides for URL-only AI, image gallery, and short layout

ai_runtime_final.py

@@ -0,0 +1,137 @@
+"""Final runtime overrides for VNEWS AI UI and short-video rendering."""
+import os, re
+import ai_runtime as rt
+from ai_runtime import app
+from fastapi.responses import HTMLResponse
+try:
+    from PIL import Image, ImageDraw, ImageFont
+except Exception:
+    Image = ImageDraw = ImageFont = None
+
+
+def clean(s):
+    import html as html_lib
+    return re.sub(r"\s+", " ", html_lib.unescape(s or "")).strip()
+
+
+def _strip_bullet_prefix(s):
+    return clean(re.sub(r'^[\s•\-\*·▪▫●○\d\.\)\(]+', '', s or ''))
+
+
+def _draw_center(draw, lines, font, y, fill, W, line_h):
+    for ln in lines:
+        try:
+            box=draw.textbbox((0,0),ln,font=font);tw=box[2]-box[0]
+        except Exception:
+            tw=len(ln)*24
+        draw.text((max(30,(W-tw)//2),y),ln,fill=fill,font=font)
+        y+=line_h
+    return y
+
+
+def final_make_frame(post,seg,idx,total,img_path,out_path):
+    """Frame layout requested by user:
+    - image at top
+    - source badge at image corner
+    - text centered in lower half
+    - no bullet/dot prefix in scene text
+    """
+    if Image is None:
+        return rt.make_frame(post,seg,idx,total,img_path,out_path)
+    W,H=1080,1920
+    hero_h=760
+    bg=Image.new('RGB',(W,H),(12,12,12))
+    try:
+        im=Image.open(img_path).convert('RGB')
+        ratio=im.width/max(1,im.height);tr=W/hero_h
+        if ratio>tr:
+            nh=hero_h;nw=int(nh*ratio)
+        else:
+            nw=W;nh=int(nw/ratio)
+        im=im.resize((nw,nh));left=(nw-W)//2;top=(nh-hero_h)//2
+        bg.paste(im.crop((left,top,left+W,top+hero_h)),(0,0))
+    except Exception:
+        pass
+    draw=ImageDraw.Draw(bg)
+    try:
+        fb=ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf',58)
+        ft=ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf',38)
+        fs=ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf',30)
+        fsmall=ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf',28)
+    except Exception:
+        fb=ft=fs=fsmall=None
+    # source badge, black rounded rectangle, no RGBA tuple to avoid Pillow RGB errors
+    badge='Nguồn: '+rt._source_badge(post)
+    try:
+        b=draw.textbbox((0,0),badge,font=fsmall);bw=b[2]-b[0];bh=b[3]-b[1]
+    except Exception:
+        bw=len(badge)*16;bh=34
+    bx=W-bw-42;by=24
+    draw.rounded_rectangle((bx-16,by-8,W-24,by+bh+14),radius=18,fill=(0,0,0))
+    draw.text((bx,by),badge,fill=(255,255,255),font=fsmall)
+
+    draw.rectangle((0,hero_h-20,W,H),fill=(12,12,12))
+    total=max(1,total)
+    total_w=total*38-14;start=(W-total_w)//2
+    for i in range(total):
+        fill=(92,184,122) if i==idx else (70,70,70)
+        draw.rounded_rectangle((start+i*38,820,start+i*38+24,832),radius=6,fill=fill)
+    brand='VNEWS AI SHORT'
+    try:
+        bb=draw.textbbox((0,0),brand,font=ft);tx=(W-(bb[2]-bb[0]))//2
+    except Exception:
+        tx=360
+    draw.text((tx,870),brand,fill=(110,231,143),font=ft)
+
+    seg=_strip_bullet_prefix(seg)
+    lines=rt.wrap_text(draw,seg,fb,W-120,8)
+    block_h=len(lines)*74
+    y=max(980,1250-block_h//2)
+    _draw_center(draw,lines,fb,y,(255,255,255),W,74)
+
+    title_lines=rt.wrap_text(draw,_strip_bullet_prefix(post.get('title','')),fs,W-120,3)
+    y2=1640
+    draw.line((80,y2-26,W-80,y2-26),fill=(70,70,70),width=2)
+    _draw_center(draw,title_lines,fs,y2,(220,220,220),W,42)
+    bg.save(out_path,quality=92)
+
+
+# Monkey-patch function used by ai_runtime.short_segments
+rt.make_frame = final_make_frame
+
+# Replace final index route to guarantee UI changes after all previous injections.
+app.router.routes=[r for r in app.router.routes if not (getattr(r,'path',None)=='/' and 'GET' in getattr(r,'methods',set()))]
+
+FINAL_INJECT = r'''
+<style>
+/* URL-only AI compose: remove topic controls completely */
+#ai-topic-input{display:none!important}
+button[onclick*="createTopicPost"],*[onclick*="createTopicPost"]{display:none!important}
+.ai-topic-row,.topic-row,.ai-compose-topic{display:none!important}
+.ai-url-only-note{font-size:11px;color:#888;margin:5px 0 8px}.ai-wall-gallery{display:grid;grid-template-columns:repeat(2,1fr);gap:6px;margin:10px 0}.ai-wall-gallery img{width:100%;aspect-ratio:16/9;object-fit:cover;border-radius:8px;background:#222}.ai-wall-gallery img:first-child{grid-column:1/-1}
+</style>
+<script>
+(function(){
+function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
+function hideTopicControls(){
+  document.querySelectorAll('#ai-topic-input').forEach(e=>{let row=e.closest('.ai-compose-topic,.topic-row,.ai-topic-row')||e.parentElement;if(row&&!row.querySelector('#ai-url-input'))row.style.display='none';else e.style.display='none';});
+  document.querySelectorAll('button,a').forEach(b=>{let t=(b.textContent||'').toLowerCase();let oc=b.getAttribute('onclick')||'';if(oc.includes('createTopicPost')||t.includes('chủ đề'))b.style.display='none';});
+  let url=document.getElementById('ai-url-input');if(url&&!document.getElementById('ai-url-only-note')){let n=document.createElement('div');n.id='ai-url-only-note';n.className='ai-url-only-note';n.textContent='Dán URL bài viết để AI tóm tắt và lấy tất cả ảnh trong bài.';url.insertAdjacentElement('afterend',n);} 
+}
+window.createTopicPost=function(){alert('Đã tắt ô nhập chủ đề. Vui lòng dán URL bài viết.');};
+window.createUrlPost=function(){let inp=document.getElementById('ai-url-input');let url=(inp&&inp.value||'').trim();if(!url)return alert('Dán URL trước');if(!/^https?:\/\//i.test(url))return alert('URL cần bắt đầu bằng http:// hoặc https://');fetch('/api/url_wall',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({url})}).then(r=>r.json().then(j=>({ok:r.ok,j}))).then(({ok,j})=>{if(ok&&j.post){if(inp)inp.value='';alert('Đã tóm tắt URL, lấy ảnh trong bài và đăng lên Tường AI');location.reload();}else alert(j.error||'Lỗi URL')}).catch(e=>alert(e.message||'Lỗi URL'));};
+function galleryHtml(p){let imgs=(p.images||[]).filter(Boolean);if(!imgs.length&&p.img)imgs=[p.img];if(!imgs.length)return '';return '<div class="ai-wall-gallery">'+imgs.slice(0,12).map(u=>`<img src="${esc(u)}" loading="lazy">`).join('')+'</div>';}
+async function getWall(){try{return (await (await fetch('/api/ai_wall')).json()).posts||[]}catch(e){return []}}
+window.aiReadWallPatched=window.aiReadWall=async function(i){let arr=window.patchedWall||window.aiWall||[];if(!arr.length)arr=await getWall();let p=arr[i];if(!p)return;showView('view-article');let sources='';if(p.sources&&p.sources.length){sources='<div class="article-summary"><b>Nguồn tham khảo:</b><br>'+p.sources.slice(0,5).map(s=>`• ${esc(s.title||s.url||'Nguồn')} ${s.url?`(${esc(new URL(s.url).hostname.replace('www.',''))})`:''}`).join('<br>')+'</div>'}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>${galleryHtml(p)}${sources}<p class="article-p" style="white-space:pre-wrap">${esc(p.text)}</p>${p.video?`<video class="article-img" src="${p.video}" controls playsinline></video>`:''}<div class="article-actions">${p.url?`<button onclick="window.open('${p.url}','_blank')">🔗 Nguồn</button>`:''}<button onclick="aiMakeShortPatched?aiMakeShortPatched(${i}):aiMakeShort(${i})">🎬 Tạo video shorts</button></div></div>`;document.getElementById('view-article').innerHTML=h;window.scrollTo(0,0);};
+setTimeout(hideTopicControls,200);setInterval(hideTopicControls,1000);
+})();
+</script>
+'''
+
+@app.get('/')
+async def index_final():
+    with open('/app/static/index.html','r',encoding='utf-8') as f:
+        html=f.read()
+    old_inject=getattr(rt.old,'PATCH_INJECT','')
+    body=old_inject + getattr(rt,'FINAL_PATCH','') + FINAL_INJECT
+    return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)