compress upload data

api.py

@@ -11,6 +11,7 @@ import os
 from dotenv import load_dotenv
 from pymongo import MongoClient
 from bson import Binary, ObjectId
+import zlib
 
 # 尝试直接获取环境变量
 BASE_URL = os.environ.get("BASE_URL")
@@ -106,9 +107,9 @@ def load_characters_csv(lang):
 
 async def generate_voice(avatar, name, emotion, tags, gender, audio_data, language):
     # 将图像数据转换为二进制
-    avatar_binary = pickle.dumps(avatar)
+    avatar_binary = zlib.compress(pickle.dumps(avatar))
     # 将音频数据转换为二进制
-    audio_binary = pickle.dumps(audio_data)
+    audio_binary = zlib.compress(pickle.dumps(audio_data))
 
     # 创建声音对象
     voice = {
@@ -120,7 +121,8 @@ async def generate_voice(avatar, name, emotion, tags, gender, audio_data, langua
         "audio_data": Binary(audio_binary),
         "language": language,
         "create_at": datetime.now().isoformat(),
-        "is_public": False
+        "is_public": False,
+        "is_reviewed": False
     }
     result = create_collection.insert_one(voice)
     return result.inserted_id

i18n/emotion.json

@@ -1,122 +0,0 @@
-{
-    "zh": {
-      "正常": "正常",
-      "生气": "生气",
-      "悲伤": "悲伤",
-      "惊讶": "惊讶",
-      "害怕": "害怕",
-      "厌恶": "厌恶",
-      "开心": "开心",
-      "失望": "失望",
-      "紧张": "紧张",
-      "不屑": "不屑",
-      "不甘": "不甘",
-      "愉悦": "愉悦",
-      "戏谑": "戏谑",
-      "疑惑": "疑惑",
-      "尴尬": "尴尬",
-      "伤心": "伤心",
-      "困倦": "困倦",
-      "无奈": "无奈",
-      "警惕": "警惕",
-      "坚决": "坚决",
-      "委屈": "委屈",
-      "害羞": "害羞",
-      "傲娇": "傲娇",
-      "女主": "女主",
-      "平淡": "平淡",
-      "疑问": "疑问",
-      "男主": "男主",
-      "警惕 ": "警惕 "
-    },
-    "en": {
-      "正常": "normal",
-      "生气": "angry",
-      "悲伤": "sad",
-      "惊讶": "surprise",
-      "害怕": "fear",
-      "厌恶": "disgust",
-      "开心": "happy",
-      "失望": "disappointment",
-      "紧张": "nervous",
-      "女主": "female",
-      "平淡": "indifferent",
-      "疑问": "question",
-      "戏谑": "playful",
-      "困倦": "sleepy",
-      "男主": "male",
-      "疑惑": "puzzled",
-      "无奈": "helpless",
-      "不甘": "unwilling",
-      "傲娇": "tsundere",
-      "警惕": "vigilant",
-      "害羞": "shy",
-      "警惕 ": "vigilant ",
-      "尴尬": "embarrassed",
-      "坚决": "determined",
-      "不屑": "disdainful",
-      "愉悦": "joyful",
-      "委屈": "aggrieved",
-      "伤心": "heartbroken"
-    },
-    "ja": {
-      "正常": "正常",
-      "生气": "怒り",
-      "悲伤": "悲しい",
-      "惊讶": "驚き",
-      "害怕": "恐れ",
-      "厌恶": "嫌悪",
-      "开心": "嬉しい",
-      "失望": "失望",
-      "紧张": "緊張",
-      "女主": "ヒロイン",
-      "平淡": "平淡",
-      "疑问": "疑問",
-      "戏谑": "からかい",
-      "困倦": "眠い",
-      "男主": "主人公",
-      "疑惑": "困惑",
-      "无奈": "仕方ない",
-      "不甘": "悔しい",
-      "傲娇": "ツンデレ",
-      "警惕": "警戒",
-      "害羞": "恥ずかしい",
-      "警惕 ": "警戒 ",
-      "尴尬": "気まずい",
-      "坚决": "断固とした",
-      "不屑": "軽蔑",
-      "愉悦": "喜び",
-      "委屈": "悔しい",
-      "伤心": "悲しい"
-    },
-    "ko": {
-      "正常": "보통",
-      "生气": "화남",
-      "悲伤": "슬픔",
-      "惊讶": "놀람",
-      "害怕": "두려움",
-      "厌恶": "혐오",
-      "开心": "행복",
-      "失望": "실망",
-      "紧张": "긴장",
-      "女主": "여주인공",
-      "平淡": "평범함",
-      "疑问": "의문",
-      "戏谑": "농담",
-      "困倦": "피곤함",
-      "男主": "남주인공",
-      "疑惑": "의혹",
-      "无奈": "어쩔 수 없음",
-      "不甘": "불만",
-      "傲娇": "츤데레",
-      "警惕": "경계",
-      "害羞": "부끄러움",
-      "警惕 ": "경계 ",
-      "尴尬": "당황",
-      "坚决": "단호함",
-      "不屑": "경멸",
-      "愉悦": "즐거움",
-      "委屈": "억울함",
-      "伤心": "상심"
-    }
-  }

i18n/translate.py

@@ -1,63 +0,0 @@
-import json
-import pandas as pd
-from typing import Literal
-import os
-import shutil
-
-os.chdir(os.path.dirname(os.path.abspath(__file__)))
-
-class LanguageConvertor:
-    def __init__(self, lang: Literal["en", "ja", "ko"]):
-        self.lang = lang
-        self.emotion = json.load(open("emotion.json", "r", encoding="utf-8"))[lang]
-        self.data = pd.read_csv(f"characters_{self.lang}.csv")
-
-    def get_name(self):
-        names = self.data["名称"].tolist()
-        return [name for i, name in enumerate(names) if name not in names[:i]]
-
-    def replace_emotion(self):
-        emotion = self.emotion
-        for i, row in self.data.iterrows():
-            try:
-                self.data.loc[i, "情绪"] = emotion[row["情绪"]]
-            except KeyError:
-                pass
-            
-    def generate_name_template(self, names: set[str]):
-        res = "{\n"
-        for i, name in enumerate(names):
-            if i == len(names) - 1:
-                res += f"  \"{name}\": \"{name}\"\n"
-            else:
-                res += f"  \"{name}\": \"{name}\",\n"
-        with open(os.path.join("temp", f"translate_name_{self.lang}.json"), "w", encoding="utf-8") as f:
-            f.write(res + "}")
-
-    def replace_name(self):
-        with open(os.path.join("temp", f"translate_name_{self.lang}.json"), "r", encoding="utf-8") as f:
-            translate_name = json.load(f)
-        for i, row in self.data.iterrows():
-            self.data.loc[i, "名称"] = translate_name[row["名称"]]
-
-    def to_file(self):
-        self.data.to_csv(f"characters_{self.lang}.csv", index=False)
-
-if __name__ == "__main__":
-    # step = 1为生成翻译模板
-    # step = 2为生成翻译后的角色文件
-    # 请确保第一步已完成再进行第二步
-    
-    step = 2
-    convertors = [LanguageConvertor(lang) for lang in ["en", "ja", "ko"]]
-    os.makedirs("temp", exist_ok=True)
-    for convertor in convertors:
-        convertor.replace_emotion()
-        print(convertor.data)
-        if step == 1:
-            convertor.generate_name_template(convertor.get_name())
-    if step == 2:
-        for convertor in convertors:
-            convertor.replace_name()
-            convertor.to_file()
-        # shutil.rmtree("temp")