完善模板编辑与模型管理体验

backend/routers/export.py

@@ -49,7 +49,9 @@ async def export_docx(document_id: int, db: AsyncSession = Depends(get_db)):
     for log, paragraph in result.all():
         logs.append(
             {
+                "anchor_title": paragraph.anchor_title or paragraph.title,
                 "title": paragraph.title,
+                "write_mode": paragraph.write_mode,
                 "content": json.loads(log.content) if log.content else {"content": []},
             }
         )

backend/routers/models.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
+import httpx
 
 from database import get_db
 from models.ai_model import AiModel
@@ -11,6 +12,11 @@ from services.security import decrypt_text, encrypt_text, mask_secret
 router = APIRouter()
 
 
+def _is_deepseek_model(model: AiModel) -> bool:
+    provider = (model.provider or "").strip().lower()
+    return provider == "deepseek"
+
+
 def _serialize_model(model: AiModel) -> dict:
     api_key = decrypt_text(model.api_key_encrypted)
     return {
@@ -121,3 +127,36 @@ async def test_model(model_id: int, db: AsyncSession = Depends(get_db)):
             message=str(error),
             data={"id": model.id, "success": False},
         )
+
+
+@router.get("/{model_id}/balance")
+async def get_model_balance(model_id: int, db: AsyncSession = Depends(get_db)):
+    model = await db.get(AiModel, model_id)
+    if model is None:
+        raise HTTPException(status_code=404, detail="模型不存在")
+    if not _is_deepseek_model(model):
+        raise HTTPException(status_code=400, detail="仅 DeepSeek 模型支持余额查询")
+
+    api_key = decrypt_text(model.api_key_encrypted)
+    if not api_key:
+        raise HTTPException(status_code=400, detail="模型 API Key 不可用")
+
+    try:
+        async with httpx.AsyncClient(timeout=20, trust_env=False) as client:
+            response = await client.get(
+                "https://api.deepseek.com/user/balance",
+                headers={"Authorization": f"Bearer {api_key}", "Accept": "application/json"},
+            )
+            response.raise_for_status()
+            payload = response.json()
+    except Exception as error:
+        raise HTTPException(status_code=400, detail=f"查询余额失败：{error}")
+
+    return Response(
+        data={
+            "id": model.id,
+            "provider": model.provider,
+            "is_available": payload.get("is_available", False),
+            "balance_infos": payload.get("balance_infos", []),
+        }
+    )

backend/routers/templates.py

@@ -32,12 +32,14 @@ def _serialize_paragraph(paragraph: Paragraph) -> dict:
         "id": paragraph.id,
         "template_id": paragraph.template_id,
         "sort_index": paragraph.sort_index,
+        "anchor_title": paragraph.anchor_title,
         "title": paragraph.title,
         "content": paragraph.content,
         "style_json": paragraph.style_json,
         "is_table": paragraph.is_table,
         "table_json": paragraph.table_json,
         "edit_mode": paragraph.edit_mode,
+        "write_mode": paragraph.write_mode,
         "model_id": paragraph.model_id,
         "need_prompt": paragraph.need_prompt,
         "prompt_text": paragraph.prompt_text,
@@ -151,11 +153,14 @@ async def upload_template(file: UploadFile = File(...), db: AsyncSession = Depen
         paragraph = Paragraph(
             template_id=template.id,
             sort_index=item.sort_index,
+            anchor_title=item.anchor_title,
             title=item.title,
             content=item.content,
             style_json=item.style_json,
             is_table=item.is_table,
             table_json=item.table_json,
+            edit_mode="manual",
+            write_mode=item.write_mode,
         )
         db.add(paragraph)
         paragraph_rows.append(paragraph)
@@ -180,16 +185,30 @@ async def save_template_paragraphs(
     if template is None:
         raise HTTPException(status_code=404, detail="模板不存在")
 
-    result = await db.execute(select(Paragraph).where(Paragraph.template_id == template_id))
-    paragraph_map = {item.id: item for item in result.scalars().all()}
+    result = await db.execute(
+        select(Paragraph)
+        .where(Paragraph.template_id == template_id)
+        .order_by(Paragraph.sort_index.asc(), Paragraph.id.asc())
+    )
+    existing_paragraphs = result.scalars().all()
+    paragraph_map = {item.id: item for item in existing_paragraphs}
+    incoming_ids = {config.id for config in body.paragraphs if config.id}
 
-    for config in body.paragraphs:
-        paragraph = paragraph_map.get(config.id)
+    for paragraph in existing_paragraphs:
+        if paragraph.id not in incoming_ids:
+            await db.delete(paragraph)
+
+    for index, config in enumerate(body.paragraphs, start=1):
+        paragraph = paragraph_map.get(config.id) if config.id else None
         if paragraph is None:
-            continue
-        paragraph.sort_index = config.sort_index
+            paragraph = Paragraph(template_id=template_id)
+            db.add(paragraph)
+        paragraph.sort_index = index
+        paragraph.anchor_title = config.anchor_title or config.title or paragraph.anchor_title
         paragraph.title = config.title
+        paragraph.content = config.content
         paragraph.edit_mode = config.edit_mode
+        paragraph.write_mode = config.write_mode
         paragraph.model_id = config.model_id
         paragraph.need_prompt = config.need_prompt
         paragraph.prompt_text = config.prompt_text
@@ -197,6 +216,8 @@ async def save_template_paragraphs(
         paragraph.file_note = config.file_note
         paragraph.output_format = config.output_format
 
+    await db.commit()
+    template.paragraph_count = len(body.paragraphs)
     await db.commit()
     return Response(data={"template_id": template_id, "saved": len(body.paragraphs)})