Upload 3 files

app.py

@@ -44,6 +44,8 @@ def set_interface_language(lang):
             "random_model": "⚔️ 随机模型",
             "select_language": "选择语言（语言很重要，选择语言决定了角色扮演中AI用的语言）",
             "select_language2": "选择语言",
+            "char_choice_label": "选择角色",
+            "selected_char_label": "当前选择的角色",
             "contant": """
                 ## 联系我们
                 ### 我们正在积极寻求更多模型愿意参与我们排行榜，
@@ -88,6 +90,8 @@ def set_interface_language(lang):
             "random_model": "⚔️ Random Model",
             "select_language": "Select a language (Language is important; the choice of language determines the language the AI will use in the roleplay)",
             "select_language2": "Select a language",
+            "char_choice_label": "Select Character",
+            "selected_char_label": "Currently Selected Character",
             "contant": """
                 ## Contact Us
                 ### We are actively seeking more models willing to participate in our leaderboard.
@@ -132,6 +136,8 @@ def set_interface_language(lang):
             "random_model": "⚔️ ランダムモデル",
             "select_language": "言語を選択してください（言語は非常に重要です。選択した言語はロールプレイでAIが使用する言語を決定します）",
             "select_language2": "言語を選択してください",
+            "char_choice_label": "キャラクターを選択",
+            "selected_char_label": "現在選択されているキャラクター",
             "contant": """
                 ## お問い合わせ
                 ### 私たちは、リーダーボードに参加したいモデルを積極的に探しています。
@@ -177,6 +183,8 @@ def set_interface_language(lang):
             "random_model": "⚔️ 랜덤 모델",
             "select_language": "언어를 선택하세요 (언어는 매우 중요합니다. 선택한 언어는 역할 놀이에서 AI가 사용할 언어를 결정합니다)",
             "select_language2": "언어를 선택하세요",
+            "char_choice_label": "캐릭터 선택",
+            "selected_char_label": "현재 선택된 캐릭터",
             "contant": """
                 ## 문의하기
                 ### 우리는 리더보드에 참여할 의향이 있는 모델을 적극적으로 찾고 있습니다.
@@ -348,7 +356,7 @@ async def update_preset_prompt(char_id, language):
     avatar_image_url = id_to_avatar(char_id)
     response = requests.get(avatar_image_url)
     image = Image.open(io.BytesIO(response.content))
-    resized_image = image.resize((224, 224))
+    resized_image = image.resize((256, 256))
     return gr.update(choices=preset_prompts), resized_image
 
 
@@ -358,7 +366,20 @@ def update_chat_and_avatar(moment):
     while selected_models[0] == selected_models[1]:
         selected_models = random.sample(models, 2)
     print(selected_models)
-    return opening, opening, moment["image_url"], "", "", selected_models
+    
+    # Resize the image
+    response = requests.get(moment["image_url"])
+    img = Image.open(io.BytesIO(response.content))
+    
+    # Calculate new width to maintain aspect ratio
+    aspect_ratio = img.width / img.height
+    new_height = 768
+    new_width = int(new_height * aspect_ratio)
+    
+    # Resize image
+    resized_img = img.resize((new_width, new_height))
+    
+    return opening, opening, resized_img, "", "", selected_models
 
 
 def refresh_data(language):
@@ -392,12 +413,12 @@ def update_language(lang):
     print("update_language", lang)
     text = set_interface_language(lang)
     characters = recommand_character(lang)
-    characters = [(item["name"], item["_id"]) for item in characters]
+    characters = [[id_to_avatar(item["_id"]), item["name"]] for item in characters]  # 只保留头像和名称
     return (
         text["title"],
         text["intro"],
         None,
-        gr.update(label=text["char_choice_label"], choices=characters),
+        gr.update(label=text["char_choice_label"], value=characters),
         gr.update(label=text["preset_prompt_label"]),
         gr.update(value=text["refresh_button"]),
         gr.update(placeholder=text["bio_placeholder"], label=text["bio_label"]),
@@ -441,6 +462,38 @@ def init_and_update(request: gr.Request):
 def passive_language_change(lang):
     return gr.update(value=lang)
 
+async def get_character_gallery(language):
+    characters = recommand_character(language)
+    return [[id_to_avatar(char['_id']), char['name']] for char in characters]
+
+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]}")
+    
+    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
+
 
 with gr.Blocks() as demo:
     # load 的时候就会刷新掉default_language
@@ -462,27 +515,40 @@ with gr.Blocks() as demo:
                     value=default_language.value,
                 )
         with gr.Row():
-            with gr.Column(scale=1):
+            with gr.Column(scale=4):
+                char_gallery = gr.Gallery(
+                    label=text["char_choice_label"],
+                    show_label=True,
+                    elem_id="character_gallery",
+                    columns=[5],
+                    object_fit="contain",
+                    height="auto",
+                    interactive=True,
+                )
+                refresh_button = gr.Button(
+                    text["refresh_button"], scale=1, variant="primary"
+            )
+            with gr.Column(scale=6):
+                preset_prompt = gr.Dropdown(
+                    label=text["preset_prompt_label"], scale=3
+                )
                 avatar_image = gr.Image(scale=1, label=text["avatar_label"])
-            with gr.Column(scale=7):
-                with gr.Row():
-                    char_choice = gr.Dropdown(
-                        choices=characters,
-                        label=text["char_choice_label"],
-                        scale=3,
-                    )
-                    preset_prompt = gr.Dropdown(
-                        label=text["preset_prompt_label"], scale=3
-                    )
-                    refresh_button = gr.Button(
-                        text["refresh_button"], scale=1, variant="primary"
-                    )
-                with gr.Row():
-                    bio = gr.Textbox(
-                        show_label=True,
-                        label=text["bio_label"],
-                        placeholder=text["bio_placeholder"],
-                    )
+            
+        with gr.Row():
+            with gr.Column(scale=1):
+                selected_char_name = gr.Textbox(
+                    label=text["selected_char_label"],
+                    interactive=False
+                )
+                selected_char_id = gr.Textbox(
+                    visible=False
+                )
+            with gr.Column(scale=3):
+                bio = gr.Textbox(
+                    show_label=True,
+                    label=text["bio_label"],
+                    placeholder=text["bio_placeholder"]
+                )
         with gr.Row():
             chatbox1 = gr.Chatbot(label=text["chatbox1_label"])
             chatbox2 = gr.Chatbot(label=text["chatbox2_label"])
@@ -540,7 +606,23 @@ with gr.Blocks() as demo:
     session_id1 = gr.State("")
     session_id2 = gr.State("")
 
-    refresh_button.click(fn=refresh_data, inputs=language, outputs=char_choice)
+    refresh_button.click(
+        fn=refresh_character_gallery,
+        inputs=[language],
+        outputs=[char_gallery]
+    )
+    char_gallery.select(
+        fn=select_character,
+        inputs=[char_gallery, language],
+        outputs=[selected_char_name, selected_char_id]
+    )
+
+    selected_char_id.change(
+        fn=update_preset_prompt,
+        inputs=[selected_char_id, language],
+        outputs=[preset_prompt, avatar_image],
+    )
+
     refresh_chat_button.click(
         fn=refresh_chat,
         inputs=[preset_prompt],
@@ -567,7 +649,7 @@ with gr.Blocks() as demo:
             title,
             intro,
             avatar_image,
-            char_choice,
+            char_gallery,
             preset_prompt,
             refresh_button,
             bio,
@@ -594,9 +676,9 @@ with gr.Blocks() as demo:
         inputs=language2,
         outputs=language
     )
-    char_choice.change(
+    char_gallery.change(
         fn=update_preset_prompt,
-        inputs=[char_choice, language],
+        inputs=[char_gallery, language],
         outputs=[preset_prompt, avatar_image],
     )
     preset_prompt.change(
@@ -723,7 +805,7 @@ with gr.Blocks() as demo:
             title,
             intro,
             avatar_image,
-            char_choice,
+            char_gallery,
             preset_prompt,
             refresh_button,
             bio,

utils.py

@@ -64,8 +64,8 @@ def load_dataframe():
             elo_dict[model1] = calculate_elo(elo_dict[model1], elo_dict[model2], outcome=0.8)
             elo_dict[model2] = calculate_elo(elo_dict[model2], elo_dict[model1], outcome=0.8)
         elif winner == 'bothbad':
-            elo_dict[model1] = calculate_elo(elo_dict[model1], elo_dict[model2], outcome=0.1)
-            elo_dict[model2] = calculate_elo(elo_dict[model2], elo_dict[model1], outcome=0.1)
+            elo_dict[model1] = calculate_elo(elo_dict[model1], elo_dict[model2], outcome=0.05)
+            elo_dict[model2] = calculate_elo(elo_dict[model2], elo_dict[model1], outcome=0.05)
 
     for model in models:
         count = data['Model 1'].value_counts().get(model, 0) + data['Model 2'].value_counts().get(model, 0)