Update app.py

app.py

@@ -31,14 +31,14 @@ with st.sidebar:
     st.header("🧠 Settings")
     selected_model = st.selectbox("Choose classification model", list(model_options.keys()))
     selected_model_id = model_options[selected_model]
-    classifier = pipeline("text-classification", model=selected_model_id, device=0 if torch.cuda.is_available() else -1)
+    classifier = pipeline("text-classification", model=selected_model_id,
+                          device=0 if torch.cuda.is_available() else -1)
 
 # 初始化历史记录
 if "history" not in st.session_state:
     st.session_state.history = []
 
 # 核心函数: 翻译并分类
-
 def classify_emoji_text(text: str):
     prompt = f"输入：{text}\n输出："
     input_ids = emoji_tokenizer(prompt, return_tensors="pt").to(emoji_model.device)
@@ -50,9 +50,18 @@ def classify_emoji_text(text: str):
     result = classifier(translated_text)[0]
     label = result["label"]
     score = result["score"]
-    reasoning = f"The sentence was flagged as '{label}' due to potentially offensive phrases. Consider replacing emotionally charged, ambiguous, or abusive terms."
+    reasoning = (
+        f"The sentence was flagged as '{label}' due to potentially offensive phrases. "
+        "Consider replacing emotionally charged, ambiguous, or abusive terms."
+    )
 
-    st.session_state.history.append({"text": text, "translated": translated_text, "label": label, "score": score, "reason": reasoning})
+    st.session_state.history.append({
+        "text": text,
+        "translated": translated_text,
+        "label": label,
+        "score": score,
+        "reason": reasoning
+    })
     return translated_text, label, score, reasoning
 
 # 页面主体
@@ -73,7 +82,7 @@ with col1:
 
                 st.markdown(f"#### 🎯 Prediction: {label}")
                 st.markdown(f"#### 📊 Confidence Score: {score:.2%}")
-                st.markdown(f"#### 🧠 Model Explanation:")
+                st.markdown("#### 🧠 Model Explanation:")
                 st.info(reason)
             except Exception as e:
                 st.error(f"❌ An error occurred during processing:\n\n{e}")
@@ -93,13 +102,14 @@ with col2:
 # 分析仪表盘
 st.markdown("---")
 st.title("📊 Violation Analysis Dashboard")
+
 if st.session_state.history:
     st.markdown("### 🧾 Offensive Terms & Suggestions")
     for item in st.session_state.history:
         st.markdown(f"- 🔹 **Input:** {item['text']}")
         st.markdown(f"   - ✨ **Translated:** {item['translated']}")
         st.markdown(f"   - ❗ **Label:** {item['label']} with **{item['score']:.2%}** confidence")
-        st.markdown(f"   - 🔧 **Suggestion:** {item['reason']}  ")
+        st.markdown(f"   - 🔧 **Suggestion:** {item['reason']}")
 
     # 雷达图演示示例（可替换为动态数据）
     radar_df = pd.DataFrame({