Upload opinion_v3_patch.py

opinion_v3_patch.py

@@ -0,0 +1,146 @@
+"""
+PERSONAL OPINION POST v3 - AI synthesis from opinion + hot sources
+Patched as standalone module imported by app_v2_entry.py
+"""
+import os, re, json, time, uuid, threading
+from fastapi import Request, Query
+from fastapi.responses import JSONResponse
+from urllib.parse import quote_plus, quote
+
+try:
+    from main import app
+except:
+    from fastapi import FastAPI
+    app = FastAPI()
+
+# --- reuse functions from app_v2_entry ---
+try:
+    from app_v2_entry import _clean, _has_kw, _search_all, _get_hot_topics, _load_wall_posts, _save_wall_posts, qwen_generate
+except:
+    # fallback imports
+    from ai_ext import qwen_generate, _clean
+    from app_v2_entry import _has_kw, _search_all, _get_hot_topics, _load_wall_posts, _save_wall_posts
+
+
+async def _ai_synthesize(topic, opinion_text, hot_sources, max_tokens=1200):
+    """Generate an article using AI by combining user opinion + news context."""
+    sources_str = "\n".join([f"- {s.get('title','')} ({s.get('via','')})" for s in hot_sources[:8]])
+    
+    prompt = f'''Dưới đây là quan điểm của người dùng và các tin tức liên quan.
+
+QUAN ĐIỂM CÁ NHÂN:
+{opinion_text}
+
+TIN TỨC LIÊN QUAN (tham khảo):
+{sources_str}
+
+YÊU CẦU: Hãy viết một bài viết tổng hợp dựa trên quan điểm trên và các nguồn tin liên quan. 
+Bài viết cần:
+1. Có tiêu đề hấp dẫn (bắt đầu bằng "## ")
+2. Trình bày quan điểm của người dùng làm nòng cốt
+3. Lồng ghép thông tin từ các nguồn tin để hỗ trợ/đối chiếu
+4. Kết luận ở cuối
+5. Viết bằng tiếng Việt tự nhiên, dài 300-500 từ
+6. Định dạng Markdown rõ ràng với heading, bullet points nếu cần'''
+
+    try:
+        result = await qwen_generate(prompt, max_tokens=max_tokens)
+        if result:
+            return result
+    except Exception as e:
+        pass
+    return f"**{topic}**\n\n{opinion_text}\n\n*Bài viết đang được cập nhật...*"
+
+
+@app.post('/api/opinion/post')
+async def api_opinion_post(request: Request):
+    """POST /api/opinion/post
+    Body: {"topic", "opinion", "sources": [{"title","url","via"}]}
+    Returns: {"article": "markdown", "title": "...", "post": {...}, "ok": true}
+    """
+    try:
+        body = await request.json()
+    except Exception:
+        return JSONResponse({"error": "Invalid JSON"}, status_code=400)
+    
+    topic = (body.get('topic') or '').strip()
+    opinion = (body.get('opinion') or '').strip()
+    sources = body.get('sources', [])
+    
+    if not opinion:
+        return JSONResponse({"error": "Vui lòng nhập quan điểm cá nhân"}, status_code=400)
+    if not topic:
+        words = [w for w in re.findall(r'[A-Za-zÀ-ỹ0-9]+', opinion) if len(w) > 3]
+        topic = ' '.join(words[:5]) if words else 'Bài viết quan điểm'
+    
+    # Step 1: Get hot topics as additional context if sources empty
+    if not sources or len(sources) < 2:
+        hot = _get_hot_topics() if '_get_hot_topics' in dir() else []
+        if isinstance(hot, dict):
+            hot = hot.get('topics', hot) if isinstance(hot, dict) else list(hot)
+        related = [t for t in (hot or []) if _has_kw(topic, t.get('topic',''))]
+        if related:
+            for rtopic in related[:3]:
+                more = _search_all(rtopic.get('topic',''), 6)
+                seen_urls = set(s.get('url') for s in sources)
+                for s in more:
+                    if s.get('url') not in seen_urls:
+                        seen_urls.add(s.get('url'))
+                        sources.append(s)
+                        if len(sources) >= 10:
+                            break
+    
+    # Step 2: Generate article via AI
+    article = await _ai_synthesize(topic, opinion, sources)
+    
+    # Step 3: Extract title from article
+    title_match = re.search(r'^##\s+(.+)$', article, re.MULTILINE)
+    title = title_match.group(1).strip() if title_match else f'Quan điểm: {topic}'
+    
+    # Step 4: Save to wall
+    post_id = str(uuid.uuid4())[:12]
+    post = {
+        "id": post_id,
+        "title": title[:200],
+        "text": article[:2000],
+        "source": "opinion_v3",
+        "opinion": opinion,
+        "topic": topic,
+        "sources": sources[:10],
+        "img": None,
+        "video": None,
+        "created": int(time.time()),
+        "created_str": time.strftime('%H:%M %d/%m/%Y', time.localtime()),
+    }
+    posts = _load_wall_posts()
+    if not isinstance(posts, list):
+        posts = []
+    posts.insert(0, post)
+    posts = posts[:200]
+    _save_wall_posts(posts)
+    
+    return JSONResponse({
+        "post": post,
+        "article": article,
+        "title": title,
+        "ok": True
+    })
+
+
+@app.get('/api/opinion/hot_context')
+async def api_opinion_hot_context(topic: str = Query(...)):
+    """GET /api/opinion/hot_context?topic=X
+    Returns: {"sources": [...], "hot_topics": [...]}
+    """
+    sources = _search_all(topic, 12)
+    hot = _get_hot_topics() if '_get_hot_topics' in dir() else []
+    if isinstance(hot, dict):
+        hot = hot.get('topics', hot) if isinstance(hot, dict) else list(hot)
+    related_hot = [t for t in (hot or []) if _has_kw(topic, t.get('topic',''))]
+    return JSONResponse({
+        "sources": sources,
+        "hot_topics": related_hot[:5],
+    })
+
+
+print("[opinion_v3_patch] PERSONAL OPINION POST v3 endpoints registered: POST /api/opinion/post, GET /api/opinion/hot_context")