Upload 2 files

app.py

@@ -339,7 +339,7 @@ def select_winner(model1_name, model2_name, state, turn, anony, Language):
         elif state == "bothbad":
             result = f"투표해 주셔서 감사합니다. {model1_name} 과(와) {model2_name} 둘 다 나쁘다고 선택하셨습니다. 다음 테스트를 위해 새로 고침하세요."
     update_model_stats(model1_name, model2_name, state, turn, anony, Language)
-    # 返回结果并让打分按钮置灰
+    # 返回结果并让打分按钮置灰，同时重置chat_count
     return (
         result,
         gr.update(interactive=False),
@@ -347,6 +347,7 @@ def select_winner(model1_name, model2_name, state, turn, anony, Language):
         gr.update(interactive=False),
         gr.update(interactive=False),
         gr.update(interactive=False),
+        0  # 重置chat_count为0
     )
 
 
@@ -386,7 +387,7 @@ def update_chat_and_avatar(moment):
     
     # Calculate new width to maintain aspect ratio
     aspect_ratio = img.width / img.height
-    new_height = 768
+    new_height = 720
     new_width = int(new_height * aspect_ratio)
     
     # Resize image
@@ -431,7 +432,7 @@ def update_language(lang):
         text["intro"],
         None,
         gr.update(label=text["char_choice_label"], value=gallery_data),
-        gr.update(label=text["preset_prompt_label"]),
+        gr.update(label=text["preset_prompt_label"], choices=[], value=None),
         gr.update(value=text["refresh_button"]),
         gr.update(placeholder=text["bio_placeholder"], label=text["bio_label"]),
         gr.update(label=text["chatbox1_label"]),
@@ -449,11 +450,14 @@ 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"]),
-        characters  # 新增的输出，用于更新 characters_state
+        gr.update(value=lang, label=text["select_language2"]),
+        characters,  # 更新 characters_state
+        gr.update(value=None),  # 重置 selected_char_name
+        gr.update(value=None),  # 重置 selected_char_id
     )
 
 
+
 def auto_i18n(request: gr.Request):
     print(request.headers["Accept-Language"])
     if request.headers["Accept-Language"].split(",")[0].lower().startswith("zh"):
@@ -562,10 +566,18 @@ with gr.Blocks() as demo:
                     object_fit="contain",
                     height="auto",
                     interactive=True,
+                    allow_preview=False
+                )
+                selected_char_name = gr.Textbox(
+                    label=text["selected_char_label"],
+                    interactive=False
+                )
+                selected_char_id = gr.Textbox(
+                    visible=False
                 )
                 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
@@ -573,20 +585,11 @@ with gr.Blocks() as demo:
                 avatar_image = gr.Image(scale=1, label=text["avatar_label"])
             
         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"]
-                )
+            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"])
@@ -707,7 +710,9 @@ with gr.Blocks() as demo:
             score_instruction,
             contant,
             language2,
-            characters_state  # 新增的输出
+            characters_state,  # 新增的输出
+            selected_char_name,  # 新增的输出
+            selected_char_id,    # 新增的输出
         ],
     )
     language2.change(
@@ -750,6 +755,7 @@ with gr.Blocks() as demo:
             tie_button,
             bothbad_button,
             battle_button,
+            chat_count
         ],
     )
     model2win_button.click(
@@ -769,6 +775,7 @@ with gr.Blocks() as demo:
             tie_button,
             bothbad_button,
             battle_button,
+            chat_count
         ],
     )
     tie_button.click(
@@ -788,6 +795,7 @@ with gr.Blocks() as demo:
             tie_button,
             bothbad_button,
             battle_button,
+            chat_count
         ],
     )
     bothbad_button.click(
@@ -807,6 +815,7 @@ with gr.Blocks() as demo:
             tie_button,
             bothbad_button,
             battle_button,
+            chat_count
         ],
     )
     battle_button.click(

utils.py

@@ -69,14 +69,15 @@ def load_dataframe():
 
     for model in models:
         count = data['Model 1'].value_counts().get(model, 0) + data['Model 2'].value_counts().get(model, 0)
-        win_count = 0
-        win_count += len(data[(data['Winner'] == 'Model 1') & (data['Model 1'] == model)])
-        win_count += len(data[(data['Winner'] == 'Model 2') & (data['Model 2'] == model)])
-        win_count += len(data[(data['Winner'] == 'tie') & ((data['Model 1'] == model) | (data['Model 2'] == model))])
-        results['模型名称'].append(model)
-        results['参赛次数'].append(count)
-        results['胜利次数'].append(win_count)
-        results['ELO'].append(round(elo_dict[model]))
+        if count > 5:  # Only include models with more than 5 participations
+            win_count = 0
+            win_count += len(data[(data['Winner'] == 'Model 1') & (data['Model 1'] == model)])
+            win_count += len(data[(data['Winner'] == 'Model 2') & (data['Model 2'] == model)])
+            win_count += len(data[(data['Winner'] == 'tie') & ((data['Model 1'] == model) | (data['Model 2'] == model))])
+            results['模型名称'].append(model)
+            results['参赛次数'].append(count)
+            results['胜利次数'].append(win_count)
+            results['ELO'].append(round(elo_dict[model]))
 
     # 将结果字典转换为DataFrame
     result_df = pd.DataFrame(results)