initial commit

app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import json
+import requests
+import config as cfg
+
+
+DEFAULT_SKILL = ["A", "B"]
+ENVS = ["canary", "production"]
+
+
+MARKDOWN = """## Skill Repository
+How to Use:
+1. Click 'Get All skills' to get list of skill.
+2. Choose specific skill that need to be updated
+3. Click Get Skill Data to get skill information related prompt and config
+4. Edit prompt and config, then click update skill to save and update the skill
+"""
+
+
+def edit_persona(skill_name, config_json, skprompt):
+    # todo update using API
+    return "Success"
+
+
+def get_all_skills():
+    response = requests.get(cfg.URL_SKILLS)
+    if response.status_code == 200:
+        result = response.json()["data"]["skills"]
+    else:
+        result = []
+    return gr.Dropdown(choices=result)
+
+
+def get_skill_data(skill_name, env):
+    URL = cfg.URL_SKILL + "/{}?env={}".format(skill_name, env)
+    prompt, config_json = "", ""
+    response = requests.get(URL)
+    if response.status_code == 200:
+        data = response.json()["data"]
+        prompt = data["prompt"]
+        config_json = json.dumps(data["config"], indent=4)
+
+    return prompt, config_json
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(MARKDOWN)
+    skills = gr.Button("1. Get All skills")
+    with gr.Row():
+        with gr.Column(scale=5):
+            env = gr.Dropdown(ENVS, label="Environment")
+        with gr.Column(scale=5):
+            skill_name = gr.Dropdown(DEFAULT_SKILL, label="Skill")
+
+    skill_data = gr.Button("2. Get Skill Data")
+    prompt = gr.Textbox(placeholder="You are helpful assistence", lines=10, label="skprompt")
+    config_json = gr.Textbox(placeholder="Config Json", lines=5, label="config json")
+
+    with gr.Row():
+        with gr.Column(scale=5):
+            update_skill = gr.Button("3. Save and Update Skill")
+        with gr.Column(scale=5):
+            status_box = gr.Textbox(label="Status code from server", )
+
+    skills.click(get_all_skills, [], [skill_name], concurrency_limit=cfg.QUEUE_CONCURENCY_COUNT)
+    skill_data.click(get_skill_data, [skill_name, env], [prompt, config_json], concurrency_limit=cfg.QUEUE_CONCURENCY_COUNT)
+    update_skill.click(edit_persona, [skill_name, prompt, config_json], [status_box], concurrency_limit=cfg.QUEUE_CONCURENCY_COUNT)
+
+    (
+        demo
+        .queue(max_size=cfg.QUEUE_MAX_SIZE)
+        .launch(auth=(cfg.USERNAME, cfg.PASSWORD), debug=True)
+    )

config.py

@@ -0,0 +1,9 @@
+import os
+
+
+USERNAME = os.getenv("USERNAME")
+PASSWORD = os.getenv("PASSWORD")
+URL_SKILLS = os.getenv("URL_ALL_SKILLS")
+URL_SKILL = os.getenv("URL_SKILL")
+QUEUE_MAX_SIZE = int(os.getenv("QUEUE_MAX_SIZE", 20))
+QUEUE_CONCURENCY_COUNT = int(os.getenv("QUEUE_CONCURENCY_COUNT", 10))