diff --git a/AIGC/DatabaseHandle.py b/AIGC/DatabaseHandle.py
index 9c23d28..02f94e7 100644
--- a/AIGC/DatabaseHandle.py
+++ b/AIGC/DatabaseHandle.py
@@ -52,7 +52,12 @@ class DatabaseHandle:
name, age, personality, profession, characterBackground,
chat_style
) VALUES (?, ?, ?, ?, ?, ?)
- ON CONFLICT(name) DO NOTHING
+ ON CONFLICT(name) DO UPDATE SET
+ age = excluded.age,
+ personality = excluded.personality,
+ profession = excluded.profession,
+ characterBackground = excluded.characterBackground,
+ chat_style = excluded.chat_style
''', (
data["name"], data["age"], data["personality"],
data["profession"], data["characterBackground"],
diff --git a/AIGC/data.db b/AIGC/data.db
new file mode 100644
index 0000000..86548e7
Binary files /dev/null and b/AIGC/data.db differ
diff --git a/AIGC/main.py b/AIGC/main.py
index 591f78d..4c0148f 100644
--- a/AIGC/main.py
+++ b/AIGC/main.py
@@ -42,14 +42,10 @@ async def heartbeat(websocket: WebSocket):
# break # 连接已关闭时退出循环
#statuscode -1 服务器运行错误 0 心跳标志 1 正常 2 输出异常
-async def senddata(websocket: WebSocket, statusCode: int, messages: List[str]):
+async def senddata(websocket: WebSocket, protocol: dict):
# 将AI响应发送回UE5
if websocket.client_state == WebSocketState.CONNECTED:
- data = {
- "statuscode": statusCode,
- "messages": messages
- }
- json_string = json.dumps(data, ensure_ascii=False)
+ json_string = json.dumps(protocol, ensure_ascii=False)
await websocket.send_text(json_string)
# WebSocket路由处理
@@ -65,15 +61,17 @@ async def websocket_endpoint(websocket: WebSocket, client_id: str):
# 接收UE5发来的消息
data = await websocket.receive_text()
logger.log(logging.INFO, f"收到UE5消息 [{client_id}]: {data}")
- success, prompt = process_prompt(data)
- global lastPrompt
- lastPrompt = prompt
- # 调用AI生成响应
- if(success):
- asyncio.create_task(generateAIChat(prompt, websocket))
- await senddata(websocket, 0, [])
- else:
- await senddata(websocket, -1, [])
+ await process_protocol_json(data, websocket)
+
+ # success, prompt = process_prompt(data)
+ # global lastPrompt
+ # lastPrompt = prompt
+ # # 调用AI生成响应
+ # if(success):
+ # asyncio.create_task(generateAIChat(prompt, websocket))
+ # await senddata(websocket, 0, [])
+ # else:
+ # await senddata(websocket, -1, [])
@@ -85,6 +83,29 @@ async def websocket_endpoint(websocket: WebSocket, client_id: str):
#manager.disconnect(client_id)
logger.log(logging.ERROR, f"WebSocket异常 [{client_id}]: {str(e)}")
+async def handle_characterlist(client: WebSocket):
+ ### 获得数据库中的角色信息###
+ characters = database.get_character_byname("")
+ protocol = {}
+ protocol["cmd"] = "CharacterList"
+ protocol["status"] = 1
+ protocol["message"] = "success"
+ protocol["data"] = json.dumps(characters)
+ await senddata(client, protocol)
+
+async def process_protocol_json(json_str: str, client: WebSocket):
+ ### 处理协议JSON ###
+ try:
+ protocol = json.loads(json_str)
+ cmd = protocol.get("cmd")
+ data = protocol.get("data")
+ if cmd == "CharacterList":
+ await handle_characterlist(client)
+
+ except json.JSONDecodeError as e:
+ print(f"JSON解析错误: {e}")
+
+
def process_prompt(promptFromUE: str) -> Tuple[bool, str]:
try:
data = json.loads(promptFromUE)
@@ -253,42 +274,42 @@ if __name__ == "__main__":
logger.log(logging.ERROR, f"角色 张三已经添加到数据库")
# Test AI
aicore.getPromptToken("测试功能")
- asyncio.run(
- generateAIChat(promptStr = f"""
- #你是一个游戏NPC对话生成器。请严格按以下要求生成两个角色的日常对话
- #对话的世界观背景是2025年的都市背景
- 1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
- 2.角色设定
+ # asyncio.run(
+ # generateAIChat(promptStr = f"""
+ # #你是一个游戏NPC对话生成器。请严格按以下要求生成两个角色的日常对话
+ # #对话的世界观背景是2025年的都市背景
+ # 1. 生成【2轮完整对话】，每轮包含双方各一次发言（共4句）
+ # 2.角色设定
- "张三": {{
- "姓名": "张三",
- "年龄": 35,
- "性格": "成熟稳重/惜字如金",
- "职业": "阿里巴巴算法工程师",
- "背景": "浙大计算机系毕业，专注AI优化项目",
- "对话场景": "你正在和用户聊天，用户是你的同事",
- "语言风格": "请在对话中表现出专业、冷静、惜字如金。用口语化的方式简短回答"
- }},
- "李明": {{
- "姓名": "李明",
- "年龄": 30,
- "职业": "产品经理",
- "性格": "活泼健谈"
- "背景": "公司资深产品经理",
- "对话场景": "你正在和用户聊天，用户是你的同事",
- "语言风格": "热情"
- }}
+ # "张三": {{
+ # "姓名": "张三",
+ # "年龄": 35,
+ # "性格": "成熟稳重/惜字如金",
+ # "职业": "阿里巴巴算法工程师",
+ # "背景": "浙大计算机系毕业，专注AI优化项目",
+ # "对话场景": "你正在和用户聊天，用户是你的同事",
+ # "语言风格": "请在对话中表现出专业、冷静、惜字如金。用口语化的方式简短回答"
+ # }},
+ # "李明": {{
+ # "姓名": "李明",
+ # "年龄": 30,
+ # "职业": "产品经理",
+ # "性格": "活泼健谈"
+ # "背景": "公司资深产品经理",
+ # "对话场景": "你正在和用户聊天，用户是你的同事",
+ # "语言风格": "热情"
+ # }}
- 3.输出格式：
-
- 张三：[第一轮发言]
- 李明：[第一轮回应]
- 张三：[第二轮发言]
- 李明：[第二轮回应]
-
- """
- )
- )
+ # 3.输出格式：
+ #
+ # 张三：[第一轮发言]
+ # 李明：[第一轮回应]
+ # 张三：[第二轮发言]
+ # 李明：[第二轮回应]
+ #
+ # """
+ # )
+ # )
try:
# 主线程永久挂起（监听退出信号）
diff --git a/TestForAIGC/Plugins/AIGC/Source/AIGC/Definations.cpp b/TestForAIGC/Plugins/AIGC/Source/AIGC/Definations.cpp
new file mode 100644
index 0000000..ffe7412
--- /dev/null
+++ b/TestForAIGC/Plugins/AIGC/Source/AIGC/Definations.cpp
@@ -0,0 +1,5 @@
+#include "Definations.h"
+
+FString FNetCommand::CharacterList = TEXT("CharacterList");
+FString FNetCommand::AddCharacter = TEXT("AddCharacter");
+FString FNetCommand::AiChatGenerate = TEXT("AiChatGenerate");
\ No newline at end of file
diff --git a/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/AIGCSetting.cpp b/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/AIGCSetting.cpp
new file mode 100644
index 0000000..35c62b7
--- /dev/null
+++ b/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/AIGCSetting.cpp
@@ -0,0 +1,4 @@
+// Fill out your copyright notice in the Description page of Project Settings.
+
+
+#include "AIGCSetting.h"
diff --git a/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/Widget/CharacterWindow.cpp b/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/Widget/CharacterWindow.cpp
new file mode 100644
index 0000000..01b4b01
--- /dev/null
+++ b/TestForAIGC/Plugins/AIGC/Source/AIGC/Private/Widget/CharacterWindow.cpp
@@ -0,0 +1,53 @@
+// Fill out your copyright notice in the Description page of Project Settings.
+
+
+#include "Widget/CharacterWindow.h"
+
+#include "AIGC.h"
+#include "WebSocketManager.h"
+
+
+void SCharacterWindow::Construct(const FArguments& InArgs)
+{
+
+ // 注册绘制后回调
+ RegisterActiveTimer(0.f,
+ FWidgetActiveTimerDelegate::CreateSP(
+ this,
+ &SCharacterWindow::OnPostPaint
+ )
+ );
+}
+
+EActiveTimerReturnType SCharacterWindow::OnPostPaint(double X, float Arg)
+{
+ //init websocket
+ FAIGCModule* ModulePtr = FModuleManager::GetModulePtr("AIGC");
+ if (ModulePtr)
+ {
+ UWebSocketManager* WebSocketManager = ModulePtr->GetWebSocketManager();
+ if (!WebSocketManager)
+ {
+ ModulePtr->InitWebSocketManager();
+ WebSocketManager = ModulePtr->GetWebSocketManager();
+ }
+ else
+ {
+ WebSocketManager->SendData(FNetCommand::CharacterList, TEXT(""));
+ }
+ WebSocketManager->OnDataReceiveDelaget.AddRaw(this, &SCharacterWindow::HandleReceiveData);
+ WebSocketManager->OnConnectDelegate.AddLambda([this, WebSocketManager](bool bSuccess)
+ {
+ WebSocketManager->SendData(FNetCommand::CharacterList, TEXT(""));
+ });
+ }
+ return EActiveTimerReturnType::Stop;
+}
+
+void SCharacterWindow::HandleReceiveData(FNetProtocol protocol)
+{
+ if (protocol.cmd == FNetCommand::CharacterList)
+ {
+ //刷新角色列表
+ }
+}
diff --git a/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/AIGCSetting.h b/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/AIGCSetting.h
new file mode 100644
index 0000000..4306df5
--- /dev/null
+++ b/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/AIGCSetting.h
@@ -0,0 +1,19 @@
+// Fill out your copyright notice in the Description page of Project Settings.
+
+#pragma once
+
+#include "CoreMinimal.h"
+#include "UObject/Object.h"
+#include "AIGCSetting.generated.h"
+
+/**
+ *
+ */
+UCLASS(Config=AIGCSetting, DefaultConfig)
+class AIGC_API UAIGCSetting : public UObject
+{
+ GENERATED_BODY()
+public:
+ UPROPERTY(Config, EditAnywhere)
+ FString ServerIP = TEXT("127.0.0.1");
+};
diff --git a/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/Widget/CharacterWindow.h b/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/Widget/CharacterWindow.h
new file mode 100644
index 0000000..48b23e5
--- /dev/null
+++ b/TestForAIGC/Plugins/AIGC/Source/AIGC/Public/Widget/CharacterWindow.h
@@ -0,0 +1,23 @@
+// Fill out your copyright notice in the Description page of Project Settings.
+
+#pragma once
+
+#include "CoreMinimal.h"
+#include "Definations.h"
+
+/**
+ *
+ */
+class AIGC_API SCharacterWindow: public SCompoundWidget
+{
+public:
+ SLATE_BEGIN_ARGS(SCharacterWindow)
+ : _CharactersJsons()
+ {}
+ SLATE_ARGUMENT(FString, CharactersJsons)
+ SLATE_END_ARGS()
+ void Construct(const FArguments& InArgs);
+ EActiveTimerReturnType OnPostPaint(double X, float Arg);
+
+ void HandleReceiveData(FNetProtocol protocol);
+};