feat chat robot 接口迁移

app/api/api_chat_robot.py

@@ -0,0 +1,27 @@
+import logging
+import time
+from fastapi import APIRouter
+
+from app.schemas.chat_robot import ChatRobotModel
+from app.service.chat_robot.script.main import chat
+
+router = APIRouter()
+logger = logging.getLogger()
+
+
+@router.post("/chat_robot")
+def chat_robot(request_data: ChatRobotModel):
+    try:
+        logger.info(f"chat_robot request item is : @@@@@@:{request_data}")
+        code = 200
+        message = "access"
+        start_time = time.time()
+        data = chat(post_data=request_data)
+        logger.info(f"chat_robot Run time is @@@@@@:{time.time() - start_time}")
+    except Exception as e:
+        code = 400
+        message = str(e)
+        data = str(e)
+        logger.warning(f"chat_robot Run Exception @@@@@@:{e}")
+    logger.info({"code": code, "message": message, "data": data})
+    return {"code": code, "message": message, "data": data}

app/api/api_prompt_generation.py

@@ -0,0 +1,28 @@
+import logging
+import time
+
+from fastapi import APIRouter
+
+from app.schemas.prompt_generation import PromptGenerationImageModel
+from app.service.prompt_generation.chatgpt_for_translation import translate_to_en
+
+router = APIRouter()
+logger = logging.getLogger()
+
+
+@router.post("/translateToEN")
+def prompt_generation(request_data: PromptGenerationImageModel):
+    try:
+        logger.info(f"prompt_translate to English request data :  @@@@@@:{request_data}")
+        code = 200
+        message = "access"
+        start_time = time.time()
+        data = translate_to_en(request_data.text)
+        logger.info(f"prompt_generation Run time is @@@@@@:{time.time() - start_time}")
+    except Exception as e:
+        code = 400
+        message = str(e)
+        data = str(e)
+        logger.warning(f"prompt_generation Run Exception @@@@@@:{e}")
+    logger.info({"code": code, "message": message, "data": data})
+    return {"code": code, "message": message, "data": data}

app/api/api_route.py

@@ -5,6 +5,9 @@ from app.api import api_super_resolution
 from app.api import api_generate_image
 from app.api import api_attribute_retrieve
 from app.api import api_design
+from app.api import api_chat_robot
+from app.api import api_prompt_generation
+
 
 router = APIRouter()
 
@@ -13,3 +16,5 @@ router.include_router(api_super_resolution.router, tags=["super_resolution"], pr
 router.include_router(api_generate_image.router, tags=["generate_image"], prefix="/api")
 router.include_router(api_attribute_retrieve.router, tags=["attribute_retrieve"], prefix="/api")
 router.include_router(api_design.router, tags=['design'], prefix="/api")
+router.include_router(api_chat_robot.router, tags=['chat_robot'], prefix="/api")
+router.include_router(api_prompt_generation.router, tags=['prompt_generation'], prefix="/api")