Upload app.py

app.py

@@ -12,13 +12,14 @@ from utils import *
 def set_interface_language(lang):
     if lang == "Chinese":
         return {
-            "title": "# LLM角色扮演竞技场：在角色扮演场景中评估LLMs的表现",
+            "title": "# LLM角色扮演竞技场：在角色扮演场景中评估LLMs的表现\n\n## 点赞❤️这个Space，下次还能找到！",
             "intro": """
             ## 📜 规则
             #### ·  与两个匿名的模型（例如 Claude, Gemini, Llama ）同时进行角色扮演（他们会成为一个相同的角色），投票选出更好的那个
             #### ·  你可以一直对话直到选出赢家(单轮对话上限是5轮)
-            #### ·  [角色来自Rubii.ai, 想要和角色进行更长久的角色扮演？来Rubii.ai吧。](https://rubii.ai/)
             #### ·  想要开始对话，您需要先在"选择角色"中选择一个角色，之后您需要在"选择时刻"中选择一个时刻，时刻是给角色选一个场景和开场白，这样可以与角色在想要的场景中进行对话。
+            ### [角色来自Rubii.ai, 想要和角色进行更长久的角色扮演？来Rubii.ai吧。](https://rubii.ai/)
+            ### [在线免费语音合成以及声音克隆](https://huggingface.co/spaces/YoMioAI/Free-Zero-shot-TTS-by-GSV-Infer-Engine)
             """,
             "avatar_label": "角色图片",
             "char_choice_label": "选择角色",
@@ -48,9 +49,9 @@ def set_interface_language(lang):
             "selected_char_label": "当前选择的角色",
             "contant": """
                 ## 联系我们
+                ### [技术讨论，或有任何问题，或是对LLM角色扮演感兴趣，欢迎加入我们的Discord](https://discord.gg/DyECTyqhkC)
                 ### 我们正在积极寻求更多模型愿意参与我们排行榜，
                 ### 我们也在寻求合作，如果您有意的话，欢迎请联系我们。**Email:** [contact@yomio.ai](mailto:contact@yomio.ai)
-                ### 您可以在我们的[Discord](https://discord.gg/jwcTfTpYj5)反馈遇到的BUG和问题
                 ## 服务条款
                 ### 用户在使用服务之前需同意以下条款：
                 ### 该服务为研究预览版。它仅提供有限的安全措施，可能会生成冒犯性内容。不得将该服务用于任何非法、有害、暴力、种族主义或性相关的目的。请勿上传任何私人信息。该服务会收集用户的对话数据，包括文字和图像，并保留在创作共用署名（CC-BY）或类似许可证下分发这些数据的权利。
@@ -58,13 +59,14 @@ def set_interface_language(lang):
         }
     elif lang == "English":
         return {
-            "title": "# Roleplay LLM Arena: Benchmarking LLMs in the Roleplay Scenario",
+            "title": "# Roleplay LLM Arena: Benchmarking LLMs in the Roleplay Scenario\n\n## Like❤️ this Space, you can find it next time!",
             "intro": """
             ## 📜 Rules
             #### ·  Engage in roleplay with two anonymous models (e.g., Claude, Gemini, Llama) simultaneously, as they take on the same character. Vote for the one that performs better.
             #### ·  You can continue the conversation until you select a winner (the maximum number of dialogue rounds per session is 5).
-            #### ·  [The character comes from Rubii.ai. Want to engage in longer roleplay sessions with the character? Come to Rubii.ai.](https://rubii.ai/)
             #### ·  To start the comparison, you need to first select a character in "Choose Character." Then, you need to select a "Moment" in "Choose Moment." A moment is used to set a scene and opening line for the character, allowing you to converse with the character in the desired scenario.
+            ### [Characters are from Rubii.ai. Want to roleplay with characters for a longer time? Come to Rubii.ai.](https://rubii.ai/)
+            ### [Online free voice synthesis and voice cloning](https://huggingface.co/spaces/YoMioAI/Free-Zero-shot-TTS-by-GSV-Infer-Engine)
             """,
             "avatar_label": "Character Image",
             "char_choice_label": "Select Character",
@@ -94,9 +96,9 @@ def set_interface_language(lang):
             "selected_char_label": "Currently Selected Character",
             "contant": """
                 ## Contact Us
+                ### [Technical discussion, or any questions, or if you are interested in LLM roleplay, welcome to join our Discord](https://discord.gg/DyECTyqhkC)
                 ### We are actively seeking more models willing to participate in our leaderboard.
                 ### We are also looking for collaboration opportunities. If you are interested, please contact us. **Email:** [contact@yomio.ai](mailto:contact@yomio.ai).
-                ### You can report any bugs and issues on our [Discord](https://discord.gg/jwcTfTpYj5).
                 ## Terms of Service
                 ### Users must agree to the following terms before using the service:
                 ### This service is a research preview. It provides limited safety measures and may generate offensive content. The service should not be used for any illegal, harmful, violent, racist, or sexually related purposes. Please do not upload any personal information. The service collects user conversation data, including text and images, and reserves the right to distribute this data under Creative Commons Attribution (CC-BY) or similar licenses.
@@ -104,13 +106,14 @@ def set_interface_language(lang):
         }
     elif lang == "Japanese":
         return {
-            "title": "# LLMロールプレイアリーナ：ロールプレイシナリオでのLLMのパフォーマンスを評価",
+            "title": "# LLMロールプレイアリーナ：ロールプレイシナリオでのLLMのパフォーマンスを評価\n\n## Like❤️this Space！",
             "intro": """
             ## 📜 ルール
             #### ·  2つの匿名モデル（例: Claude, Gemini, Llama）と同時にロールプレイを行い（彼らは同じキャラクターになります）、より良い方に投票してください。
             #### ·  勝者が決まるまで会話を続けることができます（1ターンあたりの会話の上限は5ターンです）。
-            #### ·  [キャラクターはRubii.aiから来ました。キャラクターともっと長いロールプレイをしたいですか？Rubii.aiに来てください。](https://rubii.ai/)]
             #### ·  会話を始めるには、まず「キャラクターを選択」でキャラクターを選択し、「時刻を選択」でシーンとオープニングを選択してください。これにより、キャラクターと望むシーンで会話ができます。
+            ### [キャラクターはRubii.aiから来ています。キャラクターとのロールプレイをより長く続けたい場合は、Rubii.aiに来てください。](https://rubii.ai/)
+            ### [オンライン無料音声合成と声のクローン](https://huggingface.co/spaces/YoMioAI/Free-Zero-shot-TTS-by-GSV-Infer-Engine)
             """,
             "avatar_label": "キャラクター画像",
             "char_choice_label": "キャラクターを選択",
@@ -140,9 +143,9 @@ def set_interface_language(lang):
             "selected_char_label": "現在選択されているキャラクター",
             "contant": """
                 ## お問い合わせ
+                ### [技術的な議論、またはご質問、またはLLMロールプレイに興味がある場合は、私たちのDiscordにご参加ください。](https://discord.gg/DyECTyqhkC)
                 ### 私たちは、リーダーボードに参加したいモデルを積極的に探しています。
                 ### 私たちはコラボレーションの機会も探しています。興味がある方は、ぜひご連絡ください。**メール:** [contact@yomio.ai](mailto:contact@yomio.ai)。
-                ### バグや問題が発生した場合は、[Discord](https://discord.gg/jwcTfTpYj5)で報告できます。
                 ## 利用規約
                 ### サービスを利用する前に、ユーザーは以下の規約に同意する必要があります：
                 ### 本サービスは研究プレビュー版です。限られた安全対策を提供しており、攻撃的な内容を生成する可能性があります。本サービスを違法、有害、暴力的、人種差別的、または性的な目的で使用しないでください。個人情報のアップロードは避けてください。本サービスはユーザーの会話データ（テキストおよび画像）を収集し、クリエイティブ・コモンズ・ライセンス（CC-BY）または同様のライセンスの下でこれらのデータを配布する権利を保有します。
@@ -151,13 +154,14 @@ def set_interface_language(lang):
 
     elif lang == "Korean":
         return {
-            "title": "# LLM 역할 수행 경기장: 역할 수행 시나리오에서 LLM의 성능 평가",
+            "title": "# LLM 역할 수행 경기장: 역할 수행 시나리오에서 LLM의 성능 평가\n\n## Like❤️this Space！",
             "intro": """
             ## 📜 규칙
             #### ·  두 개의 익명의 모델(예: Claude, Gemini, Llama)과 동시에 역할 수행을 진행하고, 더 나은 모델을 선택하세요.
             #### ·  우승자를 선택할 때까지 계속 대화를 진행할 수 있습니다(최대 5라운드).
-            #### ·  [캐릭터는 Rubii.ai에서 왔습니다. 캐릭터와 더 긴 롤플레이를 하고 싶으신가요? Rubii.ai로 오세요.](https://rubii.ai/)
             #### ·  대화를 시작하려면 먼저 "캐릭터 선택"에서 캐릭터를 선택해야 하며, 그 다음 "시나리오 선택"에서 시나리오를 선택해야 합니다. 시나리오는 캐릭터에게 장면과 오프닝을 제공하여 원하는 시나리오에서 대화를 진행할 수 있게 합니다.
+            ### [캐릭터는 Rubii.ai에서 오고 있습니다. 캐릭터와 더 긴 역할 수행을 하고 싶으신가요? Rubii.ai로 오세요.](https://rubii.ai/)
+            ### [온라인 무료 음성 합성 및 음성 클론](https://huggingface.co/spaces/YoMioAI/Free-Zero-shot-TTS-by-GSV-Infer-Engine)
             """,
             "avatar_label": "캐릭터 이미지",
             "char_choice_label": "캐릭터 선택",
@@ -187,9 +191,9 @@ def set_interface_language(lang):
             "selected_char_label": "현재 선택된 캐릭터",
             "contant": """
                 ## 문의하기
+                ### [기술적인 논의, 또는 문의사항, 또는 LLM 역할 수행에 관심이 있으실 경우, 우리의 Discord에 참여해 주세요.](https://discord.gg/DyECTyqhkC)
                 ### 우리는 리더보드에 참여할 의향이 있는 모델을 적극적으로 찾고 있습니다.
                 ### 우리는 또한 협력 기회를 모색하고 있습니다. 관심이 있으시면 연락해 주세요. **이메일:** [contact@yomio.ai](mailto:contact@yomio.ai)
-                ### 버그 및 문제는 [Discord](https://discord.gg/jwcTfTpYj5)에서 보고할 수 있습니다.
                 ## 이용 약관
                 ### 사용자는 서비스를 사용하기 전에 다음 약관에 동의해야 합니다:
                 ### 이 서비스는 연구 미리보기 버전입니다. 제한된 안전 조치를 제공하며, 불쾌한 콘텐츠를 생성할 수 있습니다. 이 서비스를 불법적, 해롭거나, 폭력적이거나, 인종차별적이거나, 성적으로 관련된 목적으로 사용하지 마십시오. 개인 정보를 업로드하지 마십시오. 이 서비스는 사용자 대화 데이터(텍스트 및 이미지)를 수집하며, 크리에이티브 커먼즈 저작자 표시(CC-BY) 또는 유사한 라이선스 하에 이 데이터를 배포할 권리를 보유합니다.
@@ -421,13 +425,12 @@ def refresh_chat(moment):
 def update_language(lang):
     print("update_language", lang)
     text = set_interface_language(lang)
-    characters = recommand_character(lang)
-    characters = [[id_to_avatar(item["_id"]), item["name"]] for item in characters]  # 只保留头像和名称
+    gallery_data, characters = refresh_character_gallery(lang)
     return (
         text["title"],
         text["intro"],
         None,
-        gr.update(label=text["char_choice_label"], value=characters),
+        gr.update(label=text["char_choice_label"], value=gallery_data),
         gr.update(label=text["preset_prompt_label"]),
         gr.update(value=text["refresh_button"]),
         gr.update(placeholder=text["bio_placeholder"], label=text["bio_label"]),
@@ -446,7 +449,8 @@ def update_language(lang):
         gr.update(label=text["select_language"], value=lang),
         text["score_instruction"],
         text["contant"],
-        gr.update(value=lang,label=text["select_language2"])
+        gr.update(value=lang,label=text["select_language2"]),
+        characters  # 新增的输出，用于更新 characters_state
     )
 
 
@@ -467,7 +471,36 @@ def auto_i18n(request: gr.Request):
 
 def init_and_update(request: gr.Request):
     detected_lang = auto_i18n(request)
-    return [detected_lang] + list(update_language(detected_lang))
+    text = set_interface_language(detected_lang)
+    gallery_data, characters = refresh_character_gallery(detected_lang)
+    return [
+        detected_lang,
+        text["title"],
+        text["intro"],
+        None,
+        gr.update(label=text["char_choice_label"], value=gallery_data),
+        gr.update(label=text["preset_prompt_label"]),
+        gr.update(value=text["refresh_button"]),
+        gr.update(placeholder=text["bio_placeholder"], label=text["bio_label"]),
+        gr.update(label=text["chatbox1_label"]),
+        gr.update(label=text["chatbox2_label"]),
+        gr.update(placeholder=text["user_input_placeholder"]),
+        gr.update(value=text["battle_button"]),
+        gr.update(placeholder=text["result_placeholder"]),
+        gr.update(value=text["refresh_chat_button"]),
+        gr.update(value=text["model1win_button"]),
+        gr.update(value=text["model2win_button"]),
+        gr.update(value=text["tie_button"]),
+        gr.update(value=text["bothbad_button"]),
+        gr.update(label=text["random_model"]),
+        gr.update(label=text["ranking_tab"]),
+        gr.update(label=text["select_language"], value=detected_lang),
+        text["score_instruction"],
+        text["contant"],
+        characters  # 新增的输出
+    ]
+
+
 def passive_language_change(lang):
     return gr.update(value=lang)
 
@@ -479,34 +512,28 @@ def update_character_gallery(language):
     characters = get_character_gallery(language)
     return gr.update(value=characters)
 
-def select_character(evt: gr.SelectData, gallery, language):
-    print(f"Event triggered. Index: {evt.index}")
-    print(f"Selected character from gallery: {gallery[evt.index]}")
-    if evt.index >= len(gallery):
-        return None, None
-    
-    selected_char = gallery[evt.index]
-    selected_char_id = os.path.basename(selected_char[0]).split('_')[1].split('.')[0]
-    print(f"Extracted character ID: {selected_char_id}")
-    
-    characters = recommand_character(language)
-    print(f"Number of characters from recommand_character: {len(characters)}")
-    
-    for char in characters:
-        if char['_id'].split('_')[1] == selected_char_id:
-            print(f"Match found: {char['name']}, {char['_id']}")
-            return char['name'], char['_id']
-    
-    print("No match found")
-    return None, None
-
 def refresh_character_gallery(language):
     characters = recommand_character(language)
     gallery_data = [[id_to_avatar(char['_id']), char['name']] for char in characters]
-    return gallery_data
+    return gallery_data, characters
+
+def select_character(evt: gr.SelectData, characters):
+    index = evt.index
+    print(f"Event triggered. Index: {index}")
+    if index >= len(characters):
+        return None, None
+
+    selected_char = characters[index]
+    selected_char_name = selected_char['name']
+    selected_char_id = selected_char['_id']
+    print(f"Selected character: {selected_char_name}, ID: {selected_char_id}")
+
+    return selected_char_name, selected_char_id
+
 
 
 with gr.Blocks() as demo:
+    characters_state = gr.State(value=[])
     # load 的时候就会刷新掉default_language
     default_language = gr.State("Chinese")
     language = "Chinese"
@@ -620,13 +647,14 @@ with gr.Blocks() as demo:
     refresh_button.click(
         fn=refresh_character_gallery,
         inputs=[language],
-        outputs=[char_gallery]
+        outputs=[char_gallery, characters_state]
     )
     char_gallery.select(
         fn=select_character,
-        inputs=[char_gallery, language],
+        inputs=[characters_state],
         outputs=[selected_char_name, selected_char_id]
     )
+
     selected_char_id.change(
         fn=update_preset_prompt,
         inputs=[selected_char_id, language],
@@ -678,7 +706,8 @@ with gr.Blocks() as demo:
             language,
             score_instruction,
             contant,
-            language2
+            language2,
+            characters_state  # 新增的输出
         ],
     )
     language2.change(
@@ -686,11 +715,12 @@ with gr.Blocks() as demo:
         inputs=language2,
         outputs=language
     )
-    char_gallery.change(
-        fn=update_preset_prompt,
-        inputs=[char_gallery, language],
-        outputs=[preset_prompt, avatar_image],
+    char_gallery.select(
+        fn=select_character,
+        inputs=[characters_state],
+        outputs=[selected_char_name, selected_char_id]
     )
+
     preset_prompt.change(
         fn=update_chat_and_avatar,
         inputs=[preset_prompt],
@@ -834,9 +864,11 @@ with gr.Blocks() as demo:
             language,
             score_instruction,
             contant,
+            characters_state  # 新增的输出
         ],
     )
 
+
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=8).launch(
         server_name="0.0.0.0",