From ddd01f813bb03c81ed792b2f2769536bdb5e1275 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Thu, 21 Aug 2025 09:25:50 +0800
Subject: [PATCH] 'commit'

---
 dsLightRag/Routes/JiMeng.py      | 33 ----------------------
 dsLightRag/Routes/JiMengRoute.py | 47 ++++++++++++++++++++++++++++++++
 dsLightRag/Start.py              |  2 +-
 3 files changed, 48 insertions(+), 34 deletions(-)
 delete mode 100644 dsLightRag/Routes/JiMeng.py
 create mode 100644 dsLightRag/Routes/JiMengRoute.py

diff --git a/dsLightRag/Routes/JiMeng.py b/dsLightRag/Routes/JiMeng.py
deleted file mode 100644
index 344a4257..00000000
--- a/dsLightRag/Routes/JiMeng.py
+++ /dev/null
@@ -1,33 +0,0 @@
-import json
-import logging
-import time
-import uuid
-
-import fastapi
-import jieba
-from fastapi import APIRouter
-from fastapi import HTTPException
-from openai import AsyncOpenAI
-from sse_starlette.sse import EventSourceResponse
-
-from Config import Config
-from ElasticSearch.Utils.EsSearchUtil import EsSearchUtil
-
-# 创建路由路由器
-router = APIRouter(prefix="/api/jimeng", tags=["即梦"])
-
-# 配置日志
-logger = logging.getLogger(__name__)
-
-# 初始化异步 OpenAI 客户端
-client = AsyncOpenAI(
-    api_key=Config.ALY_LLM_API_KEY,
-    base_url=Config.ALY_LLM_BASE_URL
-)
-
-
-@router.post("/prompt_input")
-async def prompt_input(request: fastapi.Request):
-    data = await request.json()
-    pass
-
diff --git a/dsLightRag/Routes/JiMengRoute.py b/dsLightRag/Routes/JiMengRoute.py
new file mode 100644
index 00000000..00beceed
--- /dev/null
+++ b/dsLightRag/Routes/JiMengRoute.py
@@ -0,0 +1,47 @@
+import logging
+
+import fastapi
+from fastapi import APIRouter
+from fastapi import HTTPException
+from openai import AsyncOpenAI
+
+from Config import Config
+from JiMeng.Kit.JmTxt2ImgUtil import JmTxt2Img
+
+# 创建路由路由器
+router = APIRouter(prefix="/api/jimeng", tags=["即梦"])
+
+# 配置日志
+logger = logging.getLogger(__name__)
+
+# 初始化异步 OpenAI 客户端
+client = AsyncOpenAI(
+    api_key=Config.ALY_LLM_API_KEY,
+    base_url=Config.ALY_LLM_BASE_URL
+)
+
+
+@router.post("/prompt_input")
+async def prompt_input(request: fastapi.Request):
+    try:
+        data = await request.json()
+        prompt = data.get("prompt")
+        if not prompt:
+            raise HTTPException(status_code=400, detail="缺少提示词参数")
+
+        logger.info(f"收到图片生成请求，提示词: {prompt}")
+        # 调用 JmTxt2Img 生成图片
+        image_url = JmTxt2Img.generate_image(prompt)
+        logger.info(f"图片生成成功，URL: {image_url}")
+
+        return {
+            "code": 200,
+            "message": "成功",
+            "data": {
+                "image_url": image_url
+            }
+        }
+    except Exception as e:
+        logger.error(f"图片生成失败: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"图片生成失败: {str(e)}")
+
diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py
index 0f795051..06a4f8ce 100644
--- a/dsLightRag/Start.py
+++ b/dsLightRag/Start.py
@@ -19,7 +19,7 @@ from Routes.TeachingModel.api.ThemeController import router as theme_router
 from Routes.TeachingModel.api.DocumentController import router as document_router
 from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
 from Routes.QA import router as qa_router
-from Routes.JiMeng import router as jimeng_router
+from Routes.JiMengRoute import router as jimeng_router
 
 from Util.LightRagUtil import *
 from contextlib import asynccontextmanager