add skill library

app.py

@@ -1,57 +1,89 @@
 import gradio as gr
-from scene_recognition import recognize_scene
+import json
 from dialogue_management import manage_dialogue
 from function_trigger import trigger_function
 from config import load_config
-from skill_repository import load_skills
+from skill_repository import load_skills, save_skills, add_skill, delete_skill
 
 config = load_config()
 skills = load_skills()
 
-# 场景识别功能
-def recognize(system_time, app_name, system_load):
-    system_data = {
-        'time': system_time,
-        'app_name': app_name,
-        'system_load': system_load
-    }
-    scene = recognize_scene(config, system_data)
-    return scene['type']
-
 # 对话功能
 def respond(user_input):
     response, action = manage_dialogue(user_input, skills)
     result = trigger_function(action, config)
-    return response,result
-
-# 创建两个独立的接口
-recognize_interface = gr.Interface(
-    fn=recognize,
-    inputs=[
-        gr.Textbox(label="System Time"),
-        gr.Textbox(label="Application Name"),
-        gr.Number(label="System Load")
-    ],
-    outputs=gr.Textbox(label="Scene Type"),
-    title="Scene Recognition",
-    description="Enter system data to recognize the scene."
-)
-
-respond_interface = gr.Interface(
-    fn=respond,
-    inputs=[
-        gr.Textbox(label="User Input"),
-    ],
-    outputs=[
-        gr.Markdown(label="AI Response"),
-        gr.Textbox(label="Action Result"),
-    ],
-    title="AI Assistant Response",
-    description="Enter your command and the recognized scene type. The AI Assistant will respond accordingly."
-)
-
-# 合并两个接口
-interface = gr.TabbedInterface([recognize_interface, respond_interface], ["Scene Recognition", "AI Assistant Response"])
+    return response, result
+
+# 显示指令库文件
+def show_skills():
+    with open('skills.json', 'r', encoding='utf-8') as file:
+        skills_data = json.load(file)
+    return json.dumps(skills_data, ensure_ascii=False, indent=4)
+
+# 添加新的技能
+def add_new_skill(triggers, action):
+    trigger_list = [trigger.strip() for trigger in triggers.split(',')]
+    new_skill = {
+        "triggers": trigger_list,
+        "action": action
+    }
+    add_skill(new_skill)
+    return show_skills()
+
+# 删除技能
+def remove_skill(action):
+    delete_skill(action)
+    return show_skills()
+
+with gr.Blocks() as demo:
+    with gr.Tab("AI Assistant Response"):
+        skills = load_skills()
+        gr.Markdown("## AI Assistant Response")
+        user_input = gr.Textbox(label="User Input")
+        ai_response = gr.Markdown()
+        action_result = gr.Textbox(label="Action Result")
+        respond_button = gr.Button("Respond")
+
+        respond_button.click(
+            respond, 
+            inputs=user_input, 
+            outputs=[ai_response, action_result]
+        )
+
+    with gr.Tab("View Skills Library"):
+        gr.Markdown("## Skills Library")
+        skills_output = gr.Code(label="Skills Library")
+        load_skills_button = gr.Button("Load Skills")
+
+        load_skills_button.click(
+            show_skills, 
+            outputs=skills_output
+        )
+
+    with gr.Tab("Add Skill"):
+        gr.Markdown("## Add Skill")
+        new_triggers = gr.Textbox(label="Triggers (comma separated)")
+        new_action = gr.Textbox(label="Action")
+        add_skill_output = gr.Code(label="Updated Skills Library")
+        add_skill_button = gr.Button("Add Skill")
+
+        add_skill_button.click(
+            add_new_skill, 
+            inputs=[new_triggers, new_action], 
+            outputs=add_skill_output
+        )
+
+    with gr.Tab("Delete Skill"):
+        gr.Markdown("## Delete Skill")
+        delete_action = gr.Textbox(label="Action to Delete")
+        delete_skill_output = gr.Code(label="Updated Skills Library")
+        delete_skill_button = gr.Button("Delete Skill")
+
+        delete_skill_button.click(
+            remove_skill, 
+            inputs=delete_action, 
+            outputs=delete_skill_output
+        )
 
 if __name__ == "__main__":
-    respond_interface.launch()
+    demo.launch()

dialogue_management.py

@@ -4,7 +4,6 @@ import json
 
 def manage_dialogue(user_input, skills):
     client = OpenAI(api_key="sk-6d543a8d079d4cee83f2bda58a6469d1", base_url="https://api.deepseek.com")
-    print(skills)
     prompt = f"你是一个具有意图识别能力的助手。你需要仔细判断用户的输入中是否包含命令，当你检测到用户的命令时并且这些命令存在于命令库中时，请严格按照以下格式返回相应的传输指令代码。否则，请进行正常对话。以下是命令库中的全部命令(triggers)及其对应的动作(action)指令代码：{skills} \n严格要求：1. 只有当用户输入匹配命令库中的命令时，才返回相应的指令代码。2. 返回格式必须严格遵守：S:{{指令代码}}。不能有任何拼写或格式错误。3. 当用户输入未包含任何命令时，进行正常对话。\n示例：用户输入：能帮我重启系统吗？ \n输出：S:CMD_RESTART \n用户输入：今天天气怎么样？ \n 输出：今天天气很好，可能会有阵雨。\n 现在，请根据上述规则严格响应以下用户输入："
     response = client.chat.completions.create(
         model="deepseek-chat",

flagged/log.csv

@@ -0,0 +1,2 @@
+User Input,AI Response,Action Result,flag,username,timestamp
+,,,,,2024-05-31 16:51:57.320514

skill_repository.py

@@ -1,12 +1,26 @@
 import json
 
 def load_skills():
-    with open('skills.json', 'r') as file:
+    with open('skills.json', 'r', encoding='utf-8') as file:
         skills = json.load(file)
     return skills
 
-def get_skill(user_input, skills):
+def save_skills(skills_data):
+    with open('skills.json', 'w', encoding='utf-8') as file:
+        json.dump(skills_data, file, ensure_ascii=False, indent=4)
+
+def get_skill(user_input, scene, skills):
     for skill in skills:
         if any(trigger in user_input for trigger in skill['triggers']):
             return skill
     return None
+
+def add_skill(new_skill):
+    skills = load_skills()
+    skills.append(new_skill)
+    save_skills(skills)
+
+def delete_skill(action):
+    skills = load_skills()
+    skills = [skill for skill in skills if skill['triggers'] != action]
+    save_skills(skills)

skills.json

@@ -1,38 +1,62 @@
 [
     {
-        "triggers": ["调高亮度"],
+        "triggers": [
+            "调高亮度"
+        ],
         "action": "CMD_LIGHT_UP"
     },
     {
-        "triggers": ["调低亮度"],
+        "triggers": [
+            "调低亮度"
+        ],
         "action": "CMD_LIGHT_DOWN"
     },
     {
-        "triggers": ["调高音量"],
+        "triggers": [
+            "调高音量"
+        ],
         "action": "CMD_SOUND_UP"
     },
     {
-        "triggers": ["调低音量"],
+        "triggers": [
+            "调低音量"
+        ],
         "action": "CMD_SOUND_DOWN"
     },
     {
-        "triggers": ["重启系统"],
+        "triggers": [
+            "重启系统"
+        ],
         "action": "CMD_RESTART"
     },
     {
-        "triggers": ["锁定屏幕"],
+        "triggers": [
+            "锁定屏幕"
+        ],
         "action": "CMD_LOCK_SCREEN"
     },
     {
-        "triggers": ["深色模式"],
+        "triggers": [
+            "深色模式"
+        ],
         "action": "CMD_DARKMODE"
     },
     {
-        "triggers": ["调节亮度至{level}%"],
+        "triggers": [
+            "调节亮度至{level}%"
+        ],
         "action": "CMD_SET_BRIGHTNESS_{{level}}"
     },
     {
-        "triggers": ["设置{time}分钟的计时器"],
+        "triggers": [
+            "设置{time}分钟的计时器"
+        ],
         "action": "CMD_SET_TIMER_{{time}}M"
+    },
+    {
+        "triggers": [
+            "播放{name}音乐"
+        ],
+        "action": "CMD_MUSIC_{name}"
     }
-]
+]