Upload note services

.gitignore

@@ -0,0 +1,4 @@
+NOTE_SERVICE_SETUP.md
+.myvenv
+__pycache__/
+*.pyc

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.11-slim
+WORKDIR /app
+
+# install dependencies
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 8080
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8080"]

app/api/notes.py

@@ -0,0 +1,35 @@
+from fastapi import APIRouter, BackgroundTasks, HTTPException
+from pydantic import BaseModel
+from typing import Optional, List
+from app.services.storage import create_note, get_note
+from app.jobs.enrichment_job import run_enrichment
+
+router = APIRouter(prefix="/notes")
+
+class CreateNoteRequest(BaseModel):
+    note_id: str
+    raw_text: str
+    normalized_text: Optional[str] = None
+    keywords: List[str] = []
+    chunks: list = []
+    duration: Optional[float] = None
+    sample_rate: Optional[int] = None
+    asr_model: Optional[str] = None
+    normalization_model: Optional[str] = None
+    generate: List[str] = []
+
+@router.post("")
+async def create_note(req: CreateNoteRequest, bg: BackgroundTasks):
+    create_note(req.note_id, req.dict())
+
+    if req.generate:
+        bg.add_task(run_enrichment, req.note_id, req.generate)
+
+    return {"note_id": req.note_id, "status": "stored"}
+
+@router.get("/{note_id}")
+def fetch_note(note_id: str):
+    note = get_note(note_id)
+    if not note:
+        raise HTTPException(404, "Note not found")
+    return note

app/config.py

@@ -0,0 +1,4 @@
+import os
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
+FIREBASE_SERVICE_ACCOUNT = os.getenv("FIREBASE_SERVICE_ACCOUNT", "")

app/jobs/enrichment_job.py

@@ -0,0 +1,21 @@
+from app.services.storage import get_note, update_note
+from app.services.summary_service import generate_summary
+from app.services.mindmap_service import generate_mindmap
+
+async def run_enrichment(note_id: str, tasks: list):
+    note = get_note(note_id)
+    if not note:
+        return
+
+    text = note.get("normalized_text") or note["raw_text"]
+
+    update_note(note_id, status="processing")
+    updates = {}
+
+    if "summary" in tasks:
+        updates["summary"] = await generate_summary(text)
+
+    if "mindmap" in tasks:
+        updates["mindmap"] = await generate_mindmap(text)
+
+    update_note(note_id, data=updates, status="ready")

app/main.py

@@ -0,0 +1,9 @@
+from fastapi import FastAPI
+from app.api.notes import router as notes_router
+
+app = FastAPI(title="Note Services API")
+app.include_router(notes_router)
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}

app/services/firebase.py

@@ -0,0 +1,13 @@
+import firebase_admin
+from firebase_admin import credentials, firestore
+import json
+from app.config import FIREBASE_SERVICE_ACCOUNT
+
+if not firebase_admin._apps:
+    if not FIREBASE_SERVICE_ACCOUNT:
+        raise RuntimeError("Missing FIREBASE_SERVICE_ACCOUNT")
+
+    cred = credentials.Certificate(json.loads(FIREBASE_SERVICE_ACCOUNT))
+    firebase_admin.initialize_app(cred)
+
+db = firestore.client()

app/services/mindmap_service.py

@@ -0,0 +1,56 @@
+import asyncio, json
+from app.config.settings import GEMINI_API_KEY
+import google.generativeai as genai
+
+if GEMINI_API_KEY:
+    genai.configure(api_key=GEMINI_API_KEY)
+    _model = genai.GenerativeModel("gemini-pro")
+else:
+    _model = None
+
+
+async def generate_mindmap(text: str) -> dict:
+    if not _model:
+        return {}
+
+    prompt = f"""
+Bạn là chuyên gia tạo Sơ đồ tư duy. Hãy phân tích văn bản sau và tạo cấu trúc JSON Mindmap.
+Yêu cầu:
+1. Xác định Ý chính làm Root.
+2. Phân tách ý phụ thành nhánh con (tối đa 3 cấp).
+3. Nhãn (label) ngắn gọn (< 7 từ).
+4. Màu sắc (colorHex): Root="#6200EE", Con="#F59E2B", "#2ECF9A", "#2F9BFF".
+
+Cấu trúc JSON bắt buộc (Chỉ trả về JSON):
+{{
+    "root": {{
+        "label": "Chủ đề",
+        "colorHex": "#6200EE",
+        "children": [
+            {{
+                "label": "Ý 1",
+                "colorHex": "#F59E2B",
+                "children": []
+            }}
+        ]
+    }}
+}}
+
+Văn bản:
+{text}
+"""
+
+    loop = asyncio.get_event_loop()
+
+    def call():
+        r = _model.generate_content(prompt)
+        return r.text
+
+    raw = await loop.run_in_executor(None, call)
+
+    start = raw.find("{")
+    end = raw.rfind("}")
+    if start != -1 and end != -1:
+        return json.loads(raw[start:end+1])
+
+    return {}

app/services/storage.py

@@ -0,0 +1,26 @@
+from datetime import datetime
+from app.services.firebase import db
+
+COLLECTION = "notes"
+
+def create_note(note_id: str, payload: dict):
+    now = datetime.utcnow()
+    payload.update({
+        "status": "created",
+        "created_at": now,
+        "updated_at": now
+    })
+    db.collection(COLLECTION).document(note_id).set(payload)
+
+def update_note(note_id: str, data: dict = None, status: str = None):
+    updates = {"updated_at": datetime.utcnow()}
+    if data:
+        updates.update(data)
+    if status:
+        updates["status"] = status
+
+    db.collection(COLLECTION).document(note_id).update(updates)
+
+def get_note(note_id: str):
+    doc = db.collection(COLLECTION).document(note_id).get()
+    return doc.to_dict() if doc.exists else None

app/services/summary_service.py

@@ -0,0 +1,35 @@
+import asyncio
+from app.config.settings import GEMINI_API_KEY
+import google.generativeai as genai
+
+if GEMINI_API_KEY:
+    genai.configure(api_key=GEMINI_API_KEY)
+    _model = genai.GenerativeModel("gemini-pro")
+else:
+    _model = None
+
+
+async def generate_summary(text: str) -> str:
+    if not _model:
+        return ""
+
+    prompt = f"""
+    Bạn là chuyên gia tóm tắt. Hãy tóm tắt văn bản sau thành **một đoạn văn duy nhất**.
+    Yêu cầu:
+    1. Viết khoảng 3-5 câu, tổng hợp đầy đủ chủ đề và các ý chính.
+    2. Viết liền mạch, KHÔNG xuống dòng, KHÔNG dùng gạch đầu dòng hay đánh số.
+    3. Chỉ dựa trên thông tin được cung cấp, tuyệt đối KHÔNG tự thêm thông tin bên ngoài.
+    4. Trả về văn bản thuần (plain text).
+
+    Văn bản:
+    \"\"\"{text}\"\"\"
+    """
+
+    loop = asyncio.get_event_loop()
+
+    def call():
+        r = _model.generate_content(prompt)
+        return r.text.strip()
+
+    result = await loop.run_in_executor(None, call)
+    return result.replace("```", "").strip()

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+google-generativeai
+firebase-admin