u

app.py

@@ -485,10 +485,10 @@ def chatbot_response(message, history, window_size, threshold, score_threshold,u
         #筛选后的候选集进行投票，选出被投票最多的一条
         #投中第一个元素加双倍权重
         
-        group_scores = {}
-        if candidates:
+        group_scores={}
+        if(candidates):
             for candidate in candidates:
-                group_id = candidate['group_id']
+                group_id=candidate['group_id']
                 keyword = candidate['keyword']
                 keyword_list = candidate['keyword_list'].split(',')
 
@@ -505,8 +505,8 @@ def chatbot_response(message, history, window_size, threshold, score_threshold,u
                         score = 1
 
                 if keyword in triggered_keywords and current_turn - triggered_keywords[keyword] < window_size:
-                    if keyword == keyword_list[0]:
-                        score = triggered_weight * 3
+                    if(keyword == keyword_list[0]):
+                        score = triggered_weight*3
                     else:
                         keywords_dict[keyword] = triggered_weight
 
@@ -516,13 +516,6 @@ def chatbot_response(message, history, window_size, threshold, score_threshold,u
                 else:
                     group_scores[group_id] = score
 
-        total_score = sum(group_scores.values())
-
-        if total_score > 0:
-            for group_id in group_scores:
-                group_scores[group_id] /= total_score
-
-
         distance=1000
         if group_scores:
             max_group_id = max(group_scores, key=group_scores.get)
@@ -602,7 +595,7 @@ demo = gr.ChatInterface(
         # gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
         gr.Slider(minimum=1, maximum=5, value=2, step=1, label="Window size"),
         gr.Slider(minimum=0.01, maximum=0.3, value=0.25, step=0.01, label="Distance threshold"),
-        gr.Slider(minimum=0.05, maximum=1, value=0.4, step=0.01, label="Score threshold"),
+        gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Score threshold"),
         gr.Slider(minimum=1, maximum=5, value=2, step=1, label="Weight of keywords from users"),
         gr.Slider(minimum=0, maximum=2, value=0.5, step=0.5, label="Weight of triggered keywords"),
         gr.Slider(minimum=0, maximum=100, value=30, step=5, label="Number of candidates"),