Upload folder using huggingface_hub

app.py

@@ -1,4 +1,3 @@
-import random
 import nltk
 import ssl
 
@@ -10,18 +9,17 @@ else:
     ssl._create_default_https_context = _create_unverified_https_context
 nltk.download("cmudict")
 
+import os
+import json
+import random
 import gradio as gr
 import numpy as np
 import torch
-import os
 import re_matching
-from tools.sentence import split_by_language, sentence_split
 import utils
 from infer import infer, latest_version, get_net_g
 import gradio as gr
-import webbrowser
 from config import config
-from tools.translate import translate
 from tools.webui import reload_javascript, get_character_html
 
 device = config.webui_config.device
@@ -112,30 +110,28 @@ def init_fn():
     gr.Info("Only support Chinese now. Trying to train a mutilingual model. 欢迎在 Community 中提建议~")
     
     index = random.randint(1,7)
-    welcome_dict = {
-        1: "哇！你来找我玩啦！",
-        2: "你好呀！",
-        3: "哇！你来啦~",
-        4: "真高兴见到你！",
-        5: "我一朵花好无聊啊！",
-        6: "欢迎你！",
-        7: "来找我玩啦！",
-    }
+    welcome_text = get_sentence("Welcome", index)
     
-    return gr.update(value=f"./assets/audios/Welcome{index}.wav"), get_character_html(welcome_dict[index])
+    return gr.update(value=f"./assets/audios/Welcome{index}.wav"), get_character_html(welcome_text)
 
+def get_sentence(category, index=-1):
+    if index == -1:
+        index = random.randint(1, len(full_lines[category]))
+    return full_lines[category][f"{index}"]
 
 with open("./css/style.css", "r", encoding="utf-8") as f:
     customCSS = f.read()
+with open("./assets/lines.json", "r", encoding="utf-8") as f:
+    full_lines = json.load(f)
 
 with gr.Blocks(css=customCSS) as demo:
     exceed_flag = gr.State(value=False)
-    bubble_text = gr.State(value="")
+    tmp_string = gr.Textbox(value="", visible=False)
     character_area = gr.HTML(get_character_html("你好呀！"), elem_id="character_area")
     with gr.Tab("Speak", elem_id="tab-speak"):
         speak_input = gr.Textbox(lines=1, label="Talking Flower will say:", elem_classes="wonder-card", elem_id="input_text")
         speak_button = gr.Button("Speak!", elem_id="speak_button", elem_classes="main-button wonder-card")
-        gr.Examples(["你今天好棒！", "雄蕊羊痒的~", "我一朵花好害怕", "再来找我玩哦", "冲呀冲呀！", "塔塔开！一字摸塔塔开！", "好了是闺蜜，不好嘞是敌咪"], label=None, inputs=[speak_input], elem_id="examples")
+        example_category = gr.Examples(["夸夸你 | Praise", "游戏台词 | Scripts", "玩梗 | Meme"], fn=get_sentence, inputs=[tmp_string], outputs=[speak_input], run_on_click=True, elem_id="examples")
     with gr.Tab("Chat", elem_id="tab-chat"):
         chat_input = gr.Textbox(lines=1, placeholder="Coming Soon...", label="Chat to Talking Flower:", elem_classes="wonder-card", elem_id="input_text", interactive=False)
         chat_button = gr.Button("Chat!", elem_id="chat_button", elem_classes="main-button wonder-card")

assets/lines.json

@@ -0,0 +1,38 @@
+{
+    "Welcome": {
+        "1": "哇！你来找我玩啦！",
+        "2": "你好呀！",
+        "3": "哇！你来啦~",
+        "4": "真高兴见到你！",
+        "5": "我一朵花好无聊啊！",
+        "6": "欢迎你！",
+        "7": "来找我玩啦！"
+    },
+    "夸夸你 | Praise": {
+        "1": "你今天好棒！",
+        "2": "冲呀冲呀！",
+        "3": "我好喜欢你呀！",
+        "4": "嫁给我好吗！",
+        "5": "你今天已经很优秀了呢！",
+        "6": "祝你做个好梦！",
+        "7": "你真的好厉害啊！"
+    },
+    "游戏台词 | Scripts": {
+        "1": "雄蕊羊痒的~",
+        "2": "我一朵花好害怕",
+        "3": "冲呀冲呀！",
+        "4": "欸？？我很吵吗？",
+        "5": "舔一口！",
+        "6": "所以我就说了嘛！！！",
+        "7": "我觉得你别管我，快点走比较好哦！"
+    },
+    "玩梗 | Meme": {
+        "1": "塔塔开！一字摸塔塔开！",
+        "2": "好了是闺蜜，不好嘞是敌咪",
+        "3": "读书的拟人最美丽！",
+        "4": "形成两面包夹芝士！",
+        "5": "反正他都不难受！他只要自由！",
+        "6": "嗯？哪里出现了什么问题？",
+        "7": "原氨神！启动！"
+    }
+}

gradio_cached_examples/7/log.csv

@@ -0,0 +1,4 @@
+Talking Flower will say:,flag,username,timestamp
+冲呀冲呀！,,,2023-11-25 16:24:34.033852
+我一朵花好害怕,,,2023-11-25 16:24:34.059410
+好了是闺蜜，不好嘞是敌咪,,,2023-11-25 16:24:34.079006