From 0da6f210a6db74848b1317e189a04b835b0ace9b Mon Sep 17 00:00:00 2001 From: liangguodong Date: Fri, 13 Mar 2026 15:19:47 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E4=BA=91=E5=AD=98?= =?UTF-8?q?=E5=82=A8=E4=B8=8A=E4=BC=A0API=E8=B7=AF=E5=BE=84=E5=8F=8A?= =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=B5=8B=E8=AF=95=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/app/routers/drafts.py | 208 +++++++++++++++- server/app/routers/story.py | 466 +++++++++++++++++++++++++++++++++++ 2 files changed, 669 insertions(+), 5 deletions(-) diff --git a/server/app/routers/drafts.py b/server/app/routers/drafts.py index 64ef17b..de97978 100644 --- a/server/app/routers/drafts.py +++ b/server/app/routers/drafts.py @@ -8,9 +8,12 @@ from sqlalchemy.sql import func from pydantic import BaseModel from typing import List, Optional from datetime import datetime +import os +import base64 from app.database import get_db from app.models.story import Story, StoryDraft, DraftStatus, StoryCharacter +from app.config import get_settings router = APIRouter(prefix="/drafts", tags=["草稿箱"]) @@ -36,6 +39,148 @@ async def get_story_characters(db: AsyncSession, story_id: int) -> List[dict]: ] +async def upload_to_cloud_storage(image_bytes: bytes, cloud_path: str) -> str: + """ + 上传图片到微信云存储（云托管容器内调用） + cloud_path: 云存储路径，如 stories/1/drafts/10/branch_1/background.jpg + 返回: 文件访问路径 + """ + import httpx + + env_id = os.environ.get('TCB_ENV') or os.environ.get('CBR_ENV_ID') + if not env_id: + # 尝试从配置获取 + settings = get_settings() + env_id = getattr(settings, 'wx_cloud_env', None) + + if not env_id: + raise Exception("未检测到云环境ID") + + try: + async with httpx.AsyncClient(timeout=60.0) as client: + # 云托管内网调用云开发 API（不需要 access_token） + # 参考: https://developers.weixin.qq.com/miniprogram/dev/wxcloudrun/src/development/storage/service/upload.html + + # 1. 获取上传链接 + resp = await client.post( + "http://api.weixin.qq.com/tcb/uploadfile", + json={ + "env": env_id, + "path": cloud_path + }, + headers={"Content-Type": "application/json"} + ) + + if resp.status_code != 200: + raise Exception(f"获取上传链接失败: {resp.status_code} - {resp.text[:200]}") + + data = resp.json() + if data.get("errcode", 0) != 0: + raise Exception(f"获取上传链接失败: {data.get('errmsg')}") + + upload_url = data.get("url") + authorization = data.get("authorization") + token = data.get("token") + cos_file_id = data.get("cos_file_id") + file_id = data.get("file_id") + + # 2. 上传文件到 COS + form_data = { + "key": cloud_path, + "Signature": authorization, + "x-cos-security-token": token, + "x-cos-meta-fileid": cos_file_id, + } + + files = {"file": ("background.jpg", image_bytes, "image/jpeg")} + + upload_resp = await client.post(upload_url, data=form_data, files=files) + + if upload_resp.status_code not in [200, 204]: + raise Exception(f"上传文件失败: {upload_resp.status_code} - {upload_resp.text[:200]}") + + print(f" [CloudStorage] 文件上传成功: {file_id}") + return file_id + + except Exception as e: + print(f"[upload_to_cloud_storage] 上传失败: {e}") + raise + + +async def generate_draft_images(story_id: int, draft_id: int, ai_nodes: dict, story_category: str): + """ + 为草稿的 AI 生成节点生成背景图 + 本地环境：保存到文件系统 /uploads/stories/{story_id}/drafts/{draft_id}/{node_key}/background.jpg + 云端环境：上传到云存储 + """ + from app.services.image_gen import ImageGenService + + if not ai_nodes: + return + + settings = get_settings() + + # 检测是否是云端环境（TCB_ENV 或 CBR_ENV_ID 是云托管容器自动注入的） + is_cloud = os.environ.get('TCB_ENV') or os.environ.get('CBR_ENV_ID') + + # 本地环境使用文件系统 + base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..', settings.upload_path)) + draft_dir = os.path.join(base_dir, "stories", str(story_id), "drafts", str(draft_id)) + + service = ImageGenService() + + for node_key, node_data in ai_nodes.items(): + if not isinstance(node_data, dict): + continue + + content = node_data.get('content', '')[:150] + if not content: + continue + + try: + # 生成背景图 - 强调情绪表达 + bg_prompt = f"Background scene for {story_category} story. Scene: {content}. Wide shot, atmospheric, no characters, anime style. Strong emotional expression, dramatic mood, vivid colors reflecting the scene's emotion." + result = await service.generate_image(bg_prompt, "background", "anime") + + if result and result.get("success"): + image_bytes = base64.b64decode(result["image_data"]) + # 路径格式和本地一致：uploads/stories/{story_id}/drafts/{draft_id}/{node_key}/background.jpg + cloud_path = f"uploads/stories/{story_id}/drafts/{draft_id}/{node_key}/background.jpg" + + # 云端环境：上传到云存储 + if is_cloud: + try: + file_id = await upload_to_cloud_storage(image_bytes, cloud_path) + # 云存储返回的 file_id 格式: cloud://env-id.xxx/path + # 前端通过 CDN 地址访问: https://7072-prod-xxx.tcb.qcloud.la/uploads/... + node_data['background_url'] = f"/{cloud_path}" + print(f" ✓ 云端草稿节点 {node_key} 背景图上传成功") + except Exception as cloud_e: + print(f" ✗ 云端上传失败: {cloud_e}") + continue + + # 本地环境：保存到文件系统 + node_dir = os.path.join(draft_dir, node_key) + os.makedirs(node_dir, exist_ok=True) + bg_path = os.path.join(node_dir, "background.jpg") + + with open(bg_path, "wb") as f: + f.write(image_bytes) + + # 更新节点数据，添加图片路径 + node_data['background_url'] = f"/uploads/stories/{story_id}/drafts/{draft_id}/{node_key}/background.jpg" + print(f" ✓ 草稿节点 {node_key} 背景图生成成功") + else: + print(f" ✗ 草稿节点 {node_key} 背景图生成失败: {result.get('error') if result else 'Unknown'}") + + except Exception as e: + print(f" ✗ 草稿节点 {node_key} 图片生成异常: {e}") + + # 避免请求过快 + import asyncio + await asyncio.sleep(1) + + # ============ 请求/响应模型 ============ class PathHistoryItem(BaseModel): @@ -121,11 +266,14 @@ async def process_ai_rewrite(draft_id: int): # 获取故事角色 characters = await get_story_characters(db, story.id) + print(f"[process_ai_rewrite] 获取到角色数: {len(characters)}") # 转换路径历史格式 path_history = draft.path_history or [] + print(f"[process_ai_rewrite] 路径历史长度: {len(path_history)}") # 调用AI服务 + print(f"[process_ai_rewrite] 开始调用 AI 服务...") ai_result = await ai_service.rewrite_branch( story_title=story.title, story_category=story.category or "未知", @@ -134,9 +282,21 @@ async def process_ai_rewrite(draft_id: int): user_prompt=draft.user_prompt, characters=characters ) + print(f"[process_ai_rewrite] AI 服务返回: {bool(ai_result)}") if ai_result and ai_result.get("nodes"): - # 成功 + # 成功 - 尝试生成配图（失败不影响改写结果） + try: + print(f"[process_ai_rewrite] AI生成成功，开始生成配图...") + await generate_draft_images( + story_id=draft.story_id, + draft_id=draft.id, + ai_nodes=ai_result["nodes"], + story_category=story.category or "都市言情" + ) + except Exception as img_e: + print(f"[process_ai_rewrite] 配图生成失败（不影响改写结果）: {img_e}") + draft.status = DraftStatus.completed draft.ai_nodes = ai_result["nodes"] draft.entry_node_key = ai_result.get("entryNodeKey", "branch_1") @@ -232,8 +392,7 @@ async def process_ai_rewrite_ending(draft_id: int): pass # 成功 - 存储为对象格式（与故事节点格式一致） - draft.status = DraftStatus.completed - draft.ai_nodes = { + ai_nodes = { "ending_rewrite": { "content": content, "speaker": "旁白", @@ -242,6 +401,21 @@ async def process_ai_rewrite_ending(draft_id: int): "ending_type": "rewrite" } } + + # 生成配图（失败不影响改写结果） + try: + print(f"[process_ai_rewrite_ending] AI生成成功，开始生成配图...") + await generate_draft_images( + story_id=draft.story_id, + draft_id=draft.id, + ai_nodes=ai_nodes, + story_category=story.category or "都市言情" + ) + except Exception as img_e: + print(f"[process_ai_rewrite_ending] 配图生成失败（不影响改写结果）: {img_e}") + + draft.status = DraftStatus.completed + draft.ai_nodes = ai_nodes draft.entry_node_key = "ending_rewrite" draft.tokens_used = ai_result.get("tokens_used", 0) draft.title = f"{story.title}-{new_ending_name}" @@ -313,7 +487,18 @@ async def process_ai_continue_ending(draft_id: int): ) if ai_result and ai_result.get("nodes"): - # 成功 - 存储多节点分支格式 + # 成功 - 尝试生成配图（失败不影响续写结果） + try: + print(f"[process_ai_continue_ending] AI生成成功，开始生成配图...") + await generate_draft_images( + story_id=draft.story_id, + draft_id=draft.id, + ai_nodes=ai_result["nodes"], + story_category=story.category or "都市言情" + ) + except Exception as img_e: + print(f"[process_ai_continue_ending] 配图生成失败（不影响续写结果）: {img_e}") + draft.status = DraftStatus.completed draft.ai_nodes = ai_result["nodes"] draft.entry_node_key = ai_result.get("entryNodeKey", "continue_1") @@ -648,7 +833,7 @@ async def delete_draft( userId: int, db: AsyncSession = Depends(get_db) ): - """删除草稿""" + """删除草稿（同时清理图片文件）""" result = await db.execute( select(StoryDraft).where( StoryDraft.id == draft_id, @@ -660,6 +845,19 @@ async def delete_draft( if not draft: raise HTTPException(status_code=404, detail="草稿不存在") + # 删除草稿对应的图片文件夹 + try: + import shutil + settings = get_settings() + base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..', settings.upload_path)) + draft_dir = os.path.join(base_dir, "stories", str(draft.story_id), "drafts", str(draft_id)) + if os.path.exists(draft_dir): + shutil.rmtree(draft_dir) + print(f"[delete_draft] 已清理图片目录: {draft_dir}") + except Exception as e: + print(f"[delete_draft] 清理图片失败: {e}") + # 图片清理失败不影响草稿删除 + await db.delete(draft) await db.commit() diff --git a/server/app/routers/story.py b/server/app/routers/story.py index 66b5e7c..430e372 100644 --- a/server/app/routers/story.py +++ b/server/app/routers/story.py @@ -39,6 +39,32 @@ class RewriteBranchRequest(BaseModel): prompt: str +class NodeImageUpdate(BaseModel): + nodeKey: str + backgroundImage: str = "" + characterImage: str = "" + + +class CharacterImageUpdate(BaseModel): + characterId: int + avatarUrl: str = "" + + +class ImageConfigRequest(BaseModel): + coverUrl: str = "" + nodes: List[NodeImageUpdate] = [] + characters: List[CharacterImageUpdate] = [] + + +class GenerateImageRequest(BaseModel): + prompt: str + style: str = "anime" # anime/realistic/illustration + category: str = "character" # character/background/cover + storyId: Optional[int] = None + targetField: Optional[str] = None # coverUrl/backgroundImage/characterImage/avatarUrl + targetKey: Optional[str] = None # nodeKey 或 characterId + + # ========== API接口 ========== @router.get("") @@ -110,6 +136,222 @@ async def get_categories(db: AsyncSession = Depends(get_db)): return {"code": 0, "data": categories} +@router.get("/test-image-gen") +async def test_image_generation(): + """测试图片生成服务是否正常""" + from app.config import get_settings + from app.services.image_gen import get_image_gen_service + import httpx + + settings = get_settings() + results = { + "api_key_configured": bool(settings.gemini_api_key), + "api_key_preview": settings.gemini_api_key[:8] + "..." if settings.gemini_api_key else "未配置", + "base_url": "https://work.poloapi.com/v1beta", + "network_test": None, + "generate_test": None + } + + # 测试网络连接 + try: + async with httpx.AsyncClient(timeout=10.0) as client: + response = await client.get("https://work.poloapi.com") + results["network_test"] = { + "success": response.status_code < 500, + "status_code": response.status_code, + "message": "网络连接正常" if response.status_code < 500 else "服务端错误" + } + except Exception as e: + results["network_test"] = { + "success": False, + "error": str(e), + "message": "网络连接失败" + } + + # 测试实际生图（简单测试） + if settings.gemini_api_key: + try: + service = get_image_gen_service() + gen_result = await service.generate_image( + prompt="a simple red circle on white background", + image_type="avatar", + style="illustration" + ) + results["generate_test"] = { + "success": gen_result.get("success", False), + "error": gen_result.get("error") if not gen_result.get("success") else None, + "has_image_data": bool(gen_result.get("image_data")) + } + except Exception as e: + results["generate_test"] = { + "success": False, + "error": str(e) + } + else: + results["generate_test"] = { + "success": False, + "error": "API Key 未配置" + } + + return { + "code": 0, + "message": "测试完成", + "data": results + } + + +@router.get("/test-deepseek") +async def test_deepseek(): + """测试 DeepSeek AI 服务是否正常""" + from app.config import get_settings + import httpx + + settings = get_settings() + results = { + "ai_service_enabled": settings.ai_service_enabled, + "provider": settings.ai_provider, + "api_key_configured": bool(settings.deepseek_api_key), + "api_key_preview": settings.deepseek_api_key[:8] + "..." + settings.deepseek_api_key[-4:] if settings.deepseek_api_key and len(settings.deepseek_api_key) > 12 else "未配置或太短", + "base_url": settings.deepseek_base_url, + "model": settings.deepseek_model, + "network_test": None, + "api_test": None + } + + # 测试网络连接 + try: + async with httpx.AsyncClient(timeout=10.0) as client: + response = await client.get("https://api.deepseek.com") + results["network_test"] = { + "success": response.status_code < 500, + "status_code": response.status_code, + "message": "网络连接正常" + } + except Exception as e: + results["network_test"] = { + "success": False, + "error": str(e), + "message": "网络连接失败" + } + + # 测试 API 调用 + if settings.deepseek_api_key: + try: + async with httpx.AsyncClient(timeout=30.0) as client: + response = await client.post( + f"{settings.deepseek_base_url}/chat/completions", + headers={ + "Content-Type": "application/json", + "Authorization": f"Bearer {settings.deepseek_api_key}" + }, + json={ + "model": settings.deepseek_model, + "messages": [{"role": "user", "content": "说'测试成功'两个字"}], + "max_tokens": 10 + } + ) + + if response.status_code == 200: + data = response.json() + content = data.get("choices", [{}])[0].get("message", {}).get("content", "") + results["api_test"] = { + "success": True, + "status_code": 200, + "response": content[:50] + } + else: + results["api_test"] = { + "success": False, + "status_code": response.status_code, + "error": response.text[:200] + } + except Exception as e: + results["api_test"] = { + "success": False, + "error": str(e) + } + else: + results["api_test"] = { + "success": False, + "error": "API Key 未配置" + } + + return { + "code": 0, + "message": "测试完成", + "data": results + } + + +@router.get("/test-cloud-upload") +async def test_cloud_upload(): + """测试云存储上传是否正常""" + import os + import httpx + + tcb_env = os.environ.get("TCB_ENV") or os.environ.get("CBR_ENV_ID") + + results = { + "env_id": tcb_env, + "is_cloud": bool(tcb_env) + } + + if not tcb_env: + return { + "code": 1, + "message": "非云托管环境，无法测试云存储上传", + "data": results + } + + try: + # 测试获取上传链接 + async with httpx.AsyncClient(timeout=30.0) as client: + resp = await client.post( + "http://api.weixin.qq.com/tcb/uploadfile", + json={ + "env": tcb_env, + "path": "test/cloud_upload_test.txt" + }, + headers={"Content-Type": "application/json"} + ) + + results["status_code"] = resp.status_code + results["response"] = resp.text[:500] if resp.text else "" + + if resp.status_code == 200: + data = resp.json() + if data.get("errcode", 0) == 0: + results["upload_url"] = data.get("url", "")[:100] + results["file_id"] = data.get("file_id", "") + return { + "code": 0, + "message": "云存储上传链接获取成功", + "data": results + } + else: + results["errcode"] = data.get("errcode") + results["errmsg"] = data.get("errmsg") + return { + "code": 1, + "message": f"获取上传链接失败: {data.get('errmsg')}", + "data": results + } + else: + return { + "code": 1, + "message": f"请求失败: HTTP {resp.status_code}", + "data": results + } + + except Exception as e: + results["error"] = str(e) + return { + "code": 1, + "message": f"测试异常: {str(e)}", + "data": results + } + + @router.get("/{story_id}") async def get_story_detail(story_id: int, db: AsyncSession = Depends(get_db)): """获取故事详情（含节点和选项）""" @@ -371,4 +613,228 @@ async def ai_rewrite_branch( "tokensUsed": 0, "error": "AI服务暂时不可用" } + } + + +@router.get("/{story_id}/images") +async def get_story_images(story_id: int, db: AsyncSession = Depends(get_db)): + """获取故事的所有图片配置""" + # 获取故事封面 + result = await db.execute(select(Story).where(Story.id == story_id)) + story = result.scalar_one_or_none() + if not story: + raise HTTPException(status_code=404, detail="故事不存在") + + # 获取所有节点的图片 + nodes_result = await db.execute( + select(StoryNode).where(StoryNode.story_id == story_id).order_by(StoryNode.sort_order) + ) + nodes = nodes_result.scalars().all() + + # 获取所有角色的头像 + chars_result = await db.execute( + select(StoryCharacter).where(StoryCharacter.story_id == story_id) + ) + characters = chars_result.scalars().all() + + return { + "code": 0, + "data": { + "storyId": story_id, + "title": story.title, + "coverUrl": story.cover_url or "", + "nodes": [ + { + "nodeKey": n.node_key, + "content": n.content[:50] + "..." if len(n.content) > 50 else n.content, + "backgroundImage": n.background_image or "", + "characterImage": n.character_image or "", + "isEnding": n.is_ending, + "endingName": n.ending_name or "" + } + for n in nodes + ], + "characters": [ + { + "characterId": c.id, + "name": c.name, + "roleType": c.role_type, + "avatarUrl": c.avatar_url or "", + "avatarPrompt": c.avatar_prompt or "" + } + for c in characters + ] + } + } + + +@router.put("/{story_id}/images") +async def update_story_images( + story_id: int, + request: ImageConfigRequest, + db: AsyncSession = Depends(get_db) +): + """批量更新故事的图片配置""" + # 验证故事存在 + result = await db.execute(select(Story).where(Story.id == story_id)) + story = result.scalar_one_or_none() + if not story: + raise HTTPException(status_code=404, detail="故事不存在") + + updated = {"cover": False, "nodes": 0, "characters": 0} + + # 更新封面 + if request.coverUrl: + await db.execute( + update(Story).where(Story.id == story_id).values(cover_url=request.coverUrl) + ) + updated["cover"] = True + + # 更新节点图片 + for node_img in request.nodes: + values = {} + if node_img.backgroundImage: + values["background_image"] = node_img.backgroundImage + if node_img.characterImage: + values["character_image"] = node_img.characterImage + if values: + await db.execute( + update(StoryNode) + .where(StoryNode.story_id == story_id, StoryNode.node_key == node_img.nodeKey) + .values(**values) + ) + updated["nodes"] += 1 + + # 更新角色头像 + for char_img in request.characters: + if char_img.avatarUrl: + await db.execute( + update(StoryCharacter) + .where(StoryCharacter.id == char_img.characterId, StoryCharacter.story_id == story_id) + .values(avatar_url=char_img.avatarUrl) + ) + updated["characters"] += 1 + + await db.commit() + + return { + "code": 0, + "message": "更新成功", + "data": updated + } + + +@router.post("/generate-image") +async def generate_story_image( + request: GenerateImageRequest, + db: AsyncSession = Depends(get_db) +): + """使用AI生成图片并可选保存到故事""" + from app.services.image_gen import get_image_gen_service + + # 生成图片 + result = await get_image_gen_service().generate_and_save( + prompt=request.prompt, + category=request.category, + style=request.style + ) + + if not result.get("success"): + return { + "code": 1, + "message": result.get("error", "生成失败"), + "data": None + } + + image_url = result["url"] + + # 如果指定了故事和目标字段，自动更新 + if request.storyId and request.targetField: + if request.targetField == "coverUrl": + await db.execute( + update(Story).where(Story.id == request.storyId).values(cover_url=image_url) + ) + elif request.targetField == "backgroundImage" and request.targetKey: + await db.execute( + update(StoryNode) + .where(StoryNode.story_id == request.storyId, StoryNode.node_key == request.targetKey) + .values(background_image=image_url) + ) + elif request.targetField == "characterImage" and request.targetKey: + await db.execute( + update(StoryNode) + .where(StoryNode.story_id == request.storyId, StoryNode.node_key == request.targetKey) + .values(character_image=image_url) + ) + elif request.targetField == "avatarUrl" and request.targetKey: + await db.execute( + update(StoryCharacter) + .where(StoryCharacter.story_id == request.storyId, StoryCharacter.id == int(request.targetKey)) + .values(avatar_url=image_url) + ) + await db.commit() + + return { + "code": 0, + "message": "生成成功", + "data": { + "url": image_url, + "filename": result.get("filename"), + "saved": bool(request.storyId and request.targetField) + } + } + + +@router.post("/{story_id}/generate-all-images") +async def generate_all_story_images( + story_id: int, + style: str = "anime", + db: AsyncSession = Depends(get_db) +): + """为故事批量生成所有角色头像""" + from app.services.image_gen import get_image_gen_service + image_service = get_image_gen_service() + + # 获取所有角色 + result = await db.execute( + select(StoryCharacter).where(StoryCharacter.story_id == story_id) + ) + characters = result.scalars().all() + + if not characters: + return {"code": 1, "message": "故事没有角色数据", "data": None} + + generated = [] + failed = [] + + for char in characters: + # 使用avatar_prompt或自动构建 + prompt = char.avatar_prompt or f"{char.name}, {char.gender}, {char.appearance or ''}" + + gen_result = await image_service.generate_and_save( + prompt=prompt, + category="character", + style=style + ) + + if gen_result.get("success"): + # 更新数据库 + await db.execute( + update(StoryCharacter) + .where(StoryCharacter.id == char.id) + .values(avatar_url=gen_result["url"]) + ) + generated.append({"id": char.id, "name": char.name, "url": gen_result["url"]}) + else: + failed.append({"id": char.id, "name": char.name, "error": gen_result.get("error")}) + + await db.commit() + + return { + "code": 0, + "message": f"生成完成: {len(generated)}成功, {len(failed)}失败", + "data": { + "generated": generated, + "failed": failed + } } \ No newline at end of file