Restore Space to revision d7f41b2

restore_runner.py

@@ -1,10 +1,10 @@
 import os
 import sys
-from textwrap import dedent
 from huggingface_hub import snapshot_download
 
-REVISION = os.environ.get("VNEWS_RESTORE_REVISION", "b9b6ba4")
+REVISION = os.environ.get("VNEWS_RESTORE_REVISION", "d7f41b2")
 REPO_ID = os.environ.get("VNEWS_REPO_ID", "bep40/vnews")
+APP_MODULE = os.environ.get("VNEWS_RESTORE_APP", "restore_b9b6ba4_patch:app")
 
 snapshot_dir = snapshot_download(
     repo_id=REPO_ID,
@@ -17,228 +17,6 @@ snapshot_dir = snapshot_download(
 os.chdir(snapshot_dir)
 sys.path.insert(0, snapshot_dir)
 
-PATCHED = r'''
-import os, re, json, time, requests
-from urllib.parse import quote, urlparse
-import ai_runtime_final6 as base_runtime
-from ai_runtime_final6 import app, base, rt, HTMLResponse, JSONResponse, Request, Query
-try:
-    import main as main_mod
-except Exception:
-    main_mod = None
-
-DATA_DIR = "/data" if os.path.isdir('/data') else "/app/data"
-AI_INTERACTIONS_FILE = os.path.join(DATA_DIR, 'ai_interactions.json')
-_SHORTS_CACHE = {'t':0,'d':[]}
-CHANNELS = ['baodantri7941','baosuckhoedoisongboyte']
-
-
-def clean(s):
-    import html as html_lib
-    return re.sub(r'\s+', ' ', html_lib.unescape(s or '')).strip()
-
-def _load(path, default):
-    try:
-        if os.path.exists(path):
-            with open(path,'r',encoding='utf-8') as f:return json.load(f)
-    except Exception:pass
-    return default
-
-def _save(path, data):
-    try:
-        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 _domain(u):
-    try:return urlparse(u or '').netloc.replace('www.','')
-    except Exception:return ''
-
-def _fallback_shorts():
-    out=[];seen=set();c=[]
-    try:c += (getattr(main_mod,'SHORTS_FALLBACK',[]) or [])
-    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ô ở Hà Nội | 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'),
-        ('b1Nxzv9ixlU','Y án 3 năm tù với nữ tài xế uống 8 lon bia lái xe tông chủ tịch xã tử vong | Dân trí','baodantri7941'),
-        ('7Pd6vZ2Lz1M','Hành động ấm lòng trong tìm kiếm 5 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'),
-        ('YY8ojFNE-AU','Quái xế tự quay clip nẹt pô, đánh võng đăng TikTok bị xử lý | SKĐS','baosuckhoedoisongboyte')]
-    for vid,title,ch in hard:c.append({'id':vid,'title':title,'channel':ch})
-    for v in c:
-        vid=v.get('id')
-        if vid and vid not in seen:
-            seen.add(vid)
-            out.append({'id':vid,'title':v.get('title') or 'YouTube Short','channel':v.get('channel',''),'link':'https://www.youtube.com/watch?v='+vid,'img':'https://i.ytimg.com/vi/'+vid+'/hqdefault.jpg','source':'yt'})
-    return out
-
-def _yt_html(handle, count=24):
-    try:
-        html=requests.get(f'https://www.youtube.com/@{handle}/shorts',headers=getattr(base,'HEADERS',{}),timeout=8).text
-        ids=[];out=[]
-        for m in re.finditer(r'"videoId":"([A-Za-z0-9_-]{11})"', html):
-            vid=m.group(1)
-            if vid in ids:continue
-            ids.append(vid)
-            snip=html[max(0,m.start()-900):m.start()+1600]
-            mt=re.search(r'"title":\{"runs":\[\{"text":"([^"]+)"',snip) or re.search(r'"accessibilityText":"([^"]+)"',snip)
-            title=clean((mt.group(1) if mt else 'YouTube Short').replace('\\n',' '))
-            out.append({'id':vid,'title':title,'channel':handle,'link':'https://www.youtube.com/watch?v='+vid,'img':'https://i.ytimg.com/vi/'+vid+'/hqdefault.jpg','source':'yt'})
-            if len(out)>=count:break
-        return out
-    except Exception:return []
-
-def _fresh_shorts_fast():
-    # Fast: HTML first; yt-dlp caused slow page. Always append fallback so slide appears instantly.
-    seen=set();out=[]
-    for ch in CHANNELS:
-        for v in _yt_html(ch,20):
-            if v['id'] not in seen:
-                seen.add(v['id']);out.append(v)
-    for v in _fallback_shorts():
-        if v['id'] not in seen:
-            seen.add(v['id']);out.append(v)
-    return out[:60]
-
-# Remove routes to override.
-_PATCH={('/api/shorts','GET'),('/api/ai/interact','POST'),('/api/topic_post','POST'),('/api/article/ask','POST'),('/','GET')}
-app.router.routes=[r for r in app.router.routes if not any(getattr(r,'path',None)==p and m in getattr(r,'methods',set()) for p,m in _PATCH)]
-
-@app.get('/api/shorts')
-def api_shorts_fast(refresh:int=Query(default=0)):
-    now=time.time()
-    if not refresh and _SHORTS_CACHE['d'] and now-_SHORTS_CACHE['t']<300:return JSONResponse(_SHORTS_CACHE['d'])
-    data=_fresh_shorts_fast();_SHORTS_CACHE.update({'t':now,'d':data});return JSONResponse(data)
-
-@app.post('/api/ai/interact')
-async def ai_interact_fixed(request:Request):
-    body=await request.json();pid=str(body.get('id','')).strip();kind=str(body.get('kind','wall')).strip();action=str(body.get('action','')).strip();text=clean(body.get('text',''));title=clean(body.get('title',''));context=clean(body.get('context',''))
-    if not pid:return JSONResponse({'error':'missing id'},status_code=400)
-    db=_load(AI_INTERACTIONS_FILE,{})
-    key=kind+':'+pid
-    st=db.get(key) or {'views':0,'likes':0,'comments':[],'asks':[]}
-    if action=='view':st['views']=int(st.get('views',0))+1
-    elif action=='like':st['likes']=int(st.get('likes',0))+1
-    elif action=='comment' and text:
-        st.setdefault('comments',[]).insert(0,{'text':text[:300],'ts':int(time.time())});st['comments']=st['comments'][:100]
-    elif action=='ask' and text:
-        if kind in ('ai','short','wall'):
-            posts=base._load_ai_wall();p=next((x for x in posts if str(x.get('id'))==pid),{})
-            title=title or p.get('title','');context=context or p.get('text','')
-        if not context:context=title or pid
-        prompt=f"""Bạn là trợ lý VNEWS. Hãy trả lời đúng trọng tâm câu hỏi dựa trên nội dung/mô tả short hoặc bài viết.
-
-Tiêu đề: {title}
-Ngữ cảnh/mô tả short: {context[:6000]}
-Câu hỏi: {text}
-
-Yêu cầu:
-- Trả lời bằng tiếng Việt, cụ thể, có giải thích.
-- Nếu là short YouTube và chỉ có tiêu đề, hãy phân tích đúng trọng tâm tiêu đề; nói rõ bạn đang suy luận từ tiêu đề/mô tả, không khẳng định đã xem toàn bộ video.
-- Không trả lời chung chung.
-"""
-        ans=await base.qwen_generate(prompt,max_tokens=900)
-        if not ans:ans='AI chưa trả lời được lúc này. Bạn thử hỏi cụ thể hơn.'
-        st.setdefault('asks',[]).insert(0,{'q':text[:300],'a':ans[:1800],'ts':int(time.time())});st['asks']=st['asks'][:60]
-    db[key]=st;_save(AI_INTERACTIONS_FILE,db);return JSONResponse({'stats':st})
-
-@app.post('/api/topic_post')
-async def topic_post_clean(request:Request):
-    body=await request.json();topic=clean(body.get('topic',''))
-    if not topic:return JSONResponse({'error':'missing topic'},status_code=400)
-    try:img=base.pollinations_image_url(topic)
-    except Exception:img='https://image.pollinations.ai/prompt/'+quote('Vietnamese editorial illustration '+topic)+'?width=1024&height=576&nologo=true'
-    prompt=f"""Viết một bài báo tiếng Việt hoàn chỉnh, chất lượng cao về chủ đề: {topic}
-
-Chỉ xuất bản nội dung cuối cùng, KHÔNG nhắc lại yêu cầu, KHÔNG liệt kê chỉ dẫn, KHÔNG mở đầu bằng các câu như 'Nhiệm vụ', 'Dưới đây', 'Bài viết này sẽ'.
-
-Yêu cầu nội dung:
-- Tiêu đề hấp dẫn, cụ thể.
-- Sapo 2-3 câu đi thẳng vào chủ đề.
-- Các đoạn phân tích có kiến thức thực chất: bối cảnh, nguyên nhân, tác động, ví dụ, nhận định.
-- Nếu chủ đề là World Cup/thể thao: nói về lịch sử, ý nghĩa giải đấu, các yếu tố chuyên môn, văn hóa người hâm mộ, tác động kinh tế - xã hội.
-- Nếu không chắc thông tin thời sự mới, diễn đạt thận trọng.
-- Không sao chép nguyên văn, hãy tổng hợp và diễn đạt lại.
-- Cuối bài có mục 'Nguồn tham khảo' ngắn: Qwen2.5-VL / kiến thức tổng hợp.
-"""
-    text=await base.qwen_generate(prompt,image_url=img,max_tokens=1600)
-    if not text:text=f"{topic}\n\n{topic} là một chủ đề có nhiều lớp ý nghĩa, từ bối cảnh hình thành đến tác động thực tế đối với xã hội, văn hóa và người quan tâm. Nội dung này tổng hợp các khía cạnh quan trọng để người đọc có cái nhìn rõ hơn về vấn đề.\n\nNguồn tham khảo: Qwen2.5-VL / kiến thức tổng hợp."
-    # Clean leaked instructions if model echoes them.
-    bad_patterns=[r'Chỉ xuất bản bài viết cuối cùng.*?Nguồn tham khảo ngắn\.?',r'Nhiệm vụ:\s*viết.*?nguồn bên dưới\.?',r'Không sao chép nguyên văn.*?diễn đạt lại\.?']
-    for pat in bad_patterns:text=re.sub(pat,'',text,flags=re.I|re.S).strip()
-    post=base.make_post(topic,text,img,'','topic_qwen',sources=[{'title':'Qwen2.5-VL / kiến thức tổng hợp','url':'','via':'Qwen2.5-VL'}]);post['images']=[img]
-    posts=base._load_ai_wall();posts.insert(0,post);base._save_ai_wall(posts)
-    return JSONResponse({'post':post})
-
-@app.post('/api/article/ask')
-async def article_ask_fixed(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=''
-    try:
-        if url:
-            data=base.scrape_any_url(url)
-            title=data.get('title','');raw=(data.get('summary','')+'\n'+data.get('text','')).strip()
-    except Exception:pass
-    ctx=(raw or context)[:12000]
-    prompt=f"""Trả lời chi tiết câu hỏi của người dùng dựa trên bài viết.
-
-Tiêu đề: {title}
-Nội dung bài:
-{ctx}
-
-Câu hỏi: {question}
-
-Trả lời bằng tiếng Việt, bám sát nội dung. Nếu bài không có thông tin thì nói rõ.
-"""
-    ans=await 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 hoặc rút gọn câu hỏi.'
-    return JSONResponse({'answer':ans,'title':title})
-
-PATCH_JS=r'''
-<style>.short-modal{position:fixed;inset:auto 0 0 0;max-height:60vh;background:#181818;border-radius:16px 16px 0 0;z-index:99999;padding:14px;display:none;overflow:auto}.short-modal.active{display:block}.comment-item{background:#222;border-radius:8px;padding:7px;margin:5px 0;color:#ccc;font-size:12px}</style>
-<div id="short-modal" class="short-modal"></div>
-<script>
-(function(){
-function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
-let shorts=[];
-async function loadShorts(){shorts=await fetch('/api/shorts?refresh=1').then(r=>r.json()).catch(()=>[]);return shorts;}
-function actionPanel(kind,id){return `<div class="short-action-panel"><button class="short-action-btn" onclick="shortAct('${kind}','${id}','view')"><div class="ico">👁</div><span id="v-${kind}-${id}">0</span></button><button class="short-action-btn" onclick="shortAct('${kind}','${id}','like')"><div class="ico">❤️</div><span id="l-${kind}-${id}">0</span></button><button class="short-action-btn" onclick="openComments('${kind}','${id}')"><div class="ico">💬</div><span>BL</span></button><button class="short-action-btn" onclick="openAsk('${kind}','${id}')"><div class="ico">🤖</div><span>Hỏi</span></button><button class="short-action-btn" onclick="shareShortCtx('${kind}','${id}')"><div class="ico">📤</div><span>Share</span></button></div>`}
-window.openShortsFixed=async function(start){let arr=shorts.length?shorts:await loadShorts();if(!arr.length)return alert('Không tải được Shorts');let ordered=start>0?arr.slice(start).concat(arr.slice(0,start)):arr;showView('view-tiktok');let h='<button class="back-btn" onclick="switchCat(\'home\')">← Shorts</button><div class="tiktok-container"><div class="tiktok-feed" id="tiktok-feed">';ordered.forEach((v,i)=>{let id=v.id;let src='https://www.youtube.com/embed/'+id+'?autoplay=1&rel=0&playsinline=1';h+=`<div class="tiktok-slide" data-kind="yt" data-id="${id}" data-title="${esc(v.title)}" data-context="${esc('Video Shorts YouTube từ kênh '+(v.channel||'')+'. Tiêu đề: '+v.title)}"><iframe data-yt-src="${src}" allowfullscreen allow="accelerometer;autoplay;clipboard-write;encrypted-media;gyroscope;picture-in-picture"></iframe><div class="tiktok-bottom"><span class="badge badge-fpt">YT</span><p class="tiktok-title">${esc(v.title)}</p></div>${actionPanel('yt',id)}<span class="tiktok-counter">${i+1}/${ordered.length}</span></div>`});h+='</div></div>';document.getElementById('view-tiktok').innerHTML=h;initFeed();}
-function initFeed(){let feed=document.getElementById('tiktok-feed');if(!feed)return;let slides=feed.querySelectorAll('.tiktok-slide');let cur=-1;function act(i){if(i===cur)return;slides.forEach((sl,idx)=>{let fr=sl.querySelector('iframe'),v=sl.querySelector('video');if(idx===i){if(fr&&!fr.src&&fr.dataset.ytSrc)fr.src=fr.dataset.ytSrc;if(v)v.play().catch(()=>{});shortAct(sl.dataset.kind,sl.dataset.id,'view').catch(()=>{})}else{if(fr&&fr.src)fr.src='';if(v)v.pause();}});cur=i}let t;feed.addEventListener('scroll',()=>{clearTimeout(t);t=setTimeout(()=>{let rect=feed.getBoundingClientRect(),ctr=rect.top+rect.height/2,b=-1,d=1e9;slides.forEach((sl,i)=>{let dd=Math.abs(sl.getBoundingClientRect().top+sl.getBoundingClientRect().height/2-ctr);if(dd<d){d=dd;b=i}});if(b>=0)act(b)},120)});setTimeout(()=>act(0),250)}
-window.shortAct=async function(kind,id,action,text=''){let sl=document.querySelector(`.tiktok-slide[data-id="${id}"]`);let body={id,kind,action,text,title:sl?.dataset.title||'',context:sl?.dataset.context||sl?.dataset.title||''};let r=await fetch('/api/ai/interact',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify(body)});let j=await r.json();let st=j.stats||j;let v=document.getElementById(`v-${kind}-${id}`),l=document.getElementById(`l-${kind}-${id}`);if(v&&st.views!=null)v.textContent=st.views;if(l&&st.likes!=null)l.textContent=st.likes;return st;}
-window.openComments=function(kind,id){let m=document.getElementById('short-modal');m.innerHTML=`<h3>💬 Bình luận</h3><div id="comment-list">Đang tải...</div><textarea id="short-comment-text" style="width:100%;background:#222;color:#eee;border:1px solid #444;border-radius:10px;padding:8px" placeholder="Nhập bình luận..."></textarea><button onclick="submitComment('${kind}','${id}')">Gửi</button><button onclick="closeShortModal()">Đóng</button>`;m.classList.add('active');shortAct(kind,id,'noop').then(st=>{document.getElementById('comment-list').innerHTML=(st.comments||[]).map(c=>`<div class="comment-item">${esc(c.text)}</div>`).join('')||'<div class="comment-item">Chưa có bình luận</div>'})}
-window.submitComment=async function(kind,id){let t=document.getElementById('short-comment-text').value.trim();if(!t)return;let st=await shortAct(kind,id,'comment',t);document.getElementById('comment-list').innerHTML=(st.comments||[]).map(c=>`<div class="comment-item">${esc(c.text)}</div>`).join('')}
-window.openAsk=function(kind,id){let m=document.getElementById('short-modal');m.innerHTML=`<h3>🤖 Hỏi AI</h3><input id="short-ask-text" style="width:100%;background:#222;color:#eee;border:1px solid #444;border-radius:10px;padding:8px" placeholder="Bạn muốn hỏi gì?"><div id="short-answer"></div><button onclick="submitAsk('${kind}','${id}')">Hỏi</button><button onclick="closeShortModal()">Đóng</button>`;m.classList.add('active')}
-window.submitAsk=async function(kind,id){let t=document.getElementById('short-ask-text').value.trim();if(!t)return;let st=await shortAct(kind,id,'ask',t);let a=(st.asks&&st.asks[0]&&st.asks[0].a)||'Chưa có trả lời';document.getElementById('short-answer').innerHTML='<p style="white-space:pre-wrap;color:#ccc">'+esc(a)+'</p>'}
-window.closeShortModal=function(){document.getElementById('short-modal').classList.remove('active')}
-let oldOpen=window.openTikTok;window.openTikTok=function(type,start){if(type==='shorts')return openShortsFixed(start||0);return oldOpen?oldOpen(type,start):null;}
-setTimeout(async()=>{await loadShorts();document.querySelectorAll('.slider-label').forEach(label=>{if((label.textContent||'').includes('Shorts'))label.closest('.slider-wrap')?.querySelectorAll('.slider-item').forEach((el,i)=>el.setAttribute('onclick',`openShortsFixed(${i})`));});},800);
-})();
-</script>
-'''
-
-@app.get('/')
-async def index_patched_restore():
-    html=base_runtime._load_index_html() if hasattr(base_runtime,'_load_index_html') else open('/app/static/index.html','r',encoding='utf-8').read()
-    body=getattr(base_runtime.old,'PATCH_INJECT','')+getattr(base_runtime,'FINAL_INJECT','')+PATCH_JS
-    return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)
-'''
-
-with open(os.path.join(snapshot_dir, 'patched_restore.py'), 'w', encoding='utf-8') as f:
-    f.write(PATCHED)
-
-cmd = [
-    "uvicorn",
-    "patched_restore:app",
-    "--host",
-    "0.0.0.0",
-    "--port",
-    "7860",
-]
-os.execvp(cmd[0], cmd)
+import uvicorn
+module, app_name = APP_MODULE.split(":", 1)
+uvicorn.run(f"{module}:{app_name}", host="0.0.0.0", port=7860)