Upload ai_runtime_final4.py

ai_runtime_final4.py

@@ -0,0 +1,172 @@
+"""Final4 runtime: fix topic button visibility, shorts home feed, AI asking for videos/articles."""
+import re, time, json, os, requests
+from urllib.parse import urlparse
+import ai_runtime_final3 as f3
+from ai_runtime_final3 import app, base, rt, HTMLResponse, JSONResponse, Request, Query
+try:
+    import main as main_mod
+except Exception:
+    main_mod=None
+
+AI_INTERACTIONS_FILE=f3.AI_INTERACTIONS_FILE
+_SHORTS_CACHE={"t":0,"d":[]}
+SHORT_CHANNELS=f3.SHORT_CHANNELS
+
+def clean(s):
+    import html as html_lib
+    return re.sub(r"\s+"," ",html_lib.unescape(s or "")).strip()
+
+def _domain(u):
+    try:return urlparse(u or '').netloc.replace('www.','')
+    except Exception:return ''
+
+def _load_json(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_json(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 _fallback_shorts():
+    out=[];seen=set()
+    candidates=[]
+    try:candidates+=(getattr(main_mod,'SHORTS_FALLBACK',[]) or [])
+    except Exception:pass
+    try:candidates+=(getattr(rt,'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ô | 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'),('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')]
+    for vid,title,ch in hard:
+        candidates.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'})
+    for v in candidates:
+        vid=v.get('id') or ''
+        if vid and vid not in seen:
+            seen.add(vid)
+            if not v.get('link'):v['link']='https://www.youtube.com/watch?v='+vid
+            if not v.get('img'):v['img']='https://i.ytimg.com/vi/'+vid+'/hqdefault.jpg'
+            v['source']='yt';out.append(v)
+    return out
+
+def _fresh_shorts():
+    items=[];seen=set()
+    for ch in SHORT_CHANNELS:
+        got=f3._youtube_shorts_ytdlp(ch,24) or f3._youtube_shorts_html(ch,24)
+        for v in got:
+            vid=v.get('id')
+            if vid and vid not in seen:
+                seen.add(vid);items.append(v)
+    for v in _fallback_shorts():
+        vid=v.get('id')
+        if vid and vid not in seen:
+            seen.add(vid);items.append(v)
+    return items[:60]
+
+_PATCH={('/api/shorts','GET'),('/api/ai/interact','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_final4(refresh:int=Query(default=0)):
+    now=time.time()
+    if not refresh and _SHORTS_CACHE['d'] and now-_SHORTS_CACHE['t']<900:return JSONResponse(_SHORTS_CACHE['d'])
+    data=_fresh_shorts()
+    _SHORTS_CACHE.update({'t':now,'d':data})
+    return JSONResponse(data)
+
+@app.post('/api/ai/interact')
+async def ai_interact_final4(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',''));context=clean(body.get('context',''));title=clean(body.get('title',''))
+    if not pid:return JSONResponse({'error':'missing id'},status_code=400)
+    db=_load_json(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[:240],'ts':int(time.time())});st['comments']=st['comments'][:80]
+    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') or '')
+        if not context:context=title or pid
+        prompt=f"""Bạn là trợ lý VNEWS. Trả lời chi tiết bằng tiếng Việt dựa trên thông tin có sẵn về video/bài viết.
+
+Tiêu đề/ngữ cảnh: {title}
+Nội dung mô tả: {context[:5000]}
+
+Câu hỏi người dùng: {text}
+
+Yêu cầu:
+- Nếu là video YouTube/Shorts và chỉ có tiêu đề, hãy nói rõ rằng bạn suy luận từ tiêu đề/mô tả, không khẳng định đã xem video.
+- Trả lời cụ thể, có giải thích, không quá ngắn.
+"""
+        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 lại cụ thể hơn.'
+        st.setdefault('asks',[]).insert(0,{'q':text[:240],'a':ans[:1500],'ts':int(time.time())});st['asks']=st['asks'][:50]
+    db[key]=st;_save_json(AI_INTERACTIONS_FILE,db)
+    return JSONResponse({'stats':st})
+
+@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',''))
+    if not question:return JSONResponse({'error':'missing question'},status_code=400)
+    title='';raw=''
+    try:
+        data=None
+        if url and hasattr(f3.f2.f1,'_scrape_url_article_only'):
+            data=f3.f2.f1._scrape_url_article_only(url)
+        if not data and url:data=base.scrape_any_url(url)
+        if data:
+            title=data.get('title','');raw=(data.get('summary','')+'\n'+data.get('text','')).strip()
+    except Exception:pass
+    context=raw[:12000] if raw else clean(body.get('context',''))[:12000]
+    prompt=f"""Bạn là trợ lý đọc hiểu bài viết của VNEWS. Hãy 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 đề bài: {title}
+Nội dung bài:
+{context}
+
+Câu hỏi: {question}
+
+Yêu cầu:
+- Trả lời bằng tiếng Việt.
+- Dựa sát nội dung bài, nếu bài không có thông tin thì nói rõ.
+- Giải thích chi tiết, có gạch đầu dòng khi hữu ích.
+"""
+    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})
+
+FINAL4_INJECT = r'''
+<style>
+.topic-final4{display:flex!important;flex-direction:column!important;gap:8px!important;width:100%!important;margin-top:6px}.topic-final4 input,.topic-final4 button{display:block!important;width:100%!important;box-sizing:border-box!important}.topic-final4 button{background:#2d8659!important;color:#fff!important;border:0!important;border-radius:18px!important;padding:9px 12px!important;font-size:11px!important;font-weight:700!important}.article-ai-ask{margin-top:12px;background:#141414;border:1px solid #2a2a2a;border-radius:10px;padding:10px}.article-ai-ask textarea{width:100%;min-height:70px;background:#222;border:1px solid #444;color:#eee;border-radius:10px;padding:9px}.article-ai-ask button{background:#2d8659;border:0;color:#fff;border-radius:10px;padding:8px 12px;margin-top:6px}.article-ai-answer{white-space:pre-wrap;color:#ccc;font-size:13px;line-height:1.55;margin-top:8px}.ai-compose-row:has(#ai-url-input){display:flex!important;flex-direction:column!important}.ai-compose-row:has(#ai-url-input) input,.ai-compose-row:has(#ai-url-input) button{width:100%!important}
+</style>
+<script>
+(function(){
+function esc(s){return String(s||'').replace(/[&<>"']/g,m=>({'&':'&amp;','<':'&lt;','>':'&gt;','"':'&quot;',"'":'&#39;'}[m]));}
+let shortsMap={};
+function ensureTopicButtonFinal4(){let comp=document.querySelector('.ai-compose');if(!comp)return;if(!document.getElementById('ai-topic-input-final4')){let row=document.createElement('div');row.className='topic-final4';row.innerHTML='<input id="ai-topic-input-final4" placeholder="Nhập chủ đề để Qwen2.5VL tạo bài lên Tường AI..."><button id="ai-topic-btn-final4" onclick="createTopicPostFinal4()">✨ Tạo bài bằng Qwen</button>';comp.insertBefore(row,comp.firstChild.nextSibling);}let b=document.getElementById('ai-topic-btn-final4');if(b){b.style.display='block';b.textContent='✨ Tạo bài bằng Qwen';}}
+window.createTopicPostFinal4=async function(){let inp=document.getElementById('ai-topic-input-final4');let topic=(inp&&inp.value||'').trim();if(!topic)return alert('Nhập chủ đề trước');let btn=document.getElementById('ai-topic-btn-final4');if(btn){btn.disabled=true;btn.textContent='Đang tạo...'}try{let r=await fetch('/api/topic_post',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({topic})});let j=await r.json();if(!r.ok||j.error)throw new Error(j.error||'Lỗi');if(window.finalWall)window.finalWall.unshift(j.post);if(window.finalWall3)window.finalWall3.unshift(j.post);if(inp)inp.value='';if(window.renderWall)window.renderWall();alert('Đã tạo bài bằng Qwen và đăng lên Tường AI.');}catch(e){alert(e.message)}finally{if(btn){btn.disabled=false;btn.textContent='✨ Tạo bài bằng Qwen'}}};
+async function ensureShortsHome(){let home=document.getElementById('view-home');if(!home||document.getElementById('shorts-final4'))return;let sh=await fetch('/api/shorts?refresh=1').then(r=>r.json()).catch(()=>[]);if(!sh.length)return;let wrap=document.createElement('div');wrap.id='shorts-final4';wrap.className='slider-wrap';let h='<div class="slider-header"><span class="slider-label">📱 Shorts Dân trí & SKĐS</span><span class="slider-note">Cập nhật YouTube</span></div><div class="slider-track">';sh.slice(0,30).forEach((a,i)=>{shortsMap[a.id]=a;h+=`<div class="slider-item shorts-item" onclick="openTikTok('shorts',${i})"><div class="slider-thumb shorts-thumb">${a.img?`<img src="${a.img}">`:''}<div class="card-play">▶</div></div><div class="slider-title">${esc(a.title)}</div></div>`});h+='</div>';wrap.innerHTML=h;let after=document.querySelector('.ai-compose')||home.firstChild;if(after)after.after(wrap);else home.prepend(wrap);}
+let oldShortAct=window.shortAct;
+window.shortAct=async function(kind,id,action,text=''){let meta=shortsMap[id]||{};let url='/api/ai/interact';let body={id,kind:kind==='yt'?'yt':kind,action,text,title:meta.title||'',context:meta.title?('Video Shorts YouTube từ kênh '+(meta.channel||'')+'. Tiêu đề: '+meta.title):''};let r=await fetch(url,{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.submitShortAsk=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>';};
+let oldOpenTikTok=window.openTikTok;
+window.openTikTok=async function(type,startIdx){if(type==='shorts'){let arts=await fetch('/api/shorts?refresh=1').then(r=>r.json()).catch(()=>[]);arts.forEach(a=>{if(a.id)shortsMap[a.id]=a});}return oldOpenTikTok?oldOpenTikTok(type,startIdx):null;};
+function addArticleAskBox(){let view=document.getElementById('view-article');if(!view||document.getElementById('article-ai-ask'))return;let art=view.querySelector('.article-view');if(!art)return;let box=document.createElement('div');box.id='article-ai-ask';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="Nhập câu hỏi cần AI trả lời chi tiết về bài viết..."></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 trước');let ans=document.getElementById('article-ai-answer');ans.textContent='Đang hỏi AI...';let url=(window._currentArticle&&window._currentArticle.url)||((typeof _currentArticle!=='undefined'&&_currentArticle.url)||'');let context=document.querySelector('.article-view')?.innerText||'';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||j.error||'Không có trả lời';}catch(e){ans.textContent='Lỗi hỏi AI: '+e.message}}
+let oldReadArticle=window.readArticle;if(oldReadArticle){window.readArticle=async function(){let ret=await oldReadArticle.apply(this,arguments);setTimeout(addArticleAskBox,700);return ret;}}
+setTimeout(()=>{ensureTopicButtonFinal4();ensureShortsHome();},1000);setInterval(()=>{ensureTopicButtonFinal4();if(document.getElementById('view-home')?.classList.contains('active'))ensureShortsHome();addArticleAskBox();},2000);
+})();
+</script>
+'''
+
+@app.get('/')
+async def index_final4():
+    html=f3.f2.f1._load_index_html();body=getattr(rt.old,'PATCH_INJECT','')+f3.f2.f1.FINAL_INJECT+f3.FINAL3_INJECT+FINAL4_INJECT
+    return HTMLResponse(html.replace('</body>',body+'\n</body>') if '</body>' in html else html+body)