Update app.py

app.py

@@ -12,6 +12,7 @@ st.title("Spam Detection App")
 
 # สร้างกล่องข้อความให้ผู้ใช้ป้อนข้อความ
 user_input = st.text_input("ใส่ข้อความที่ต้องการตรวจสอบ:")
+
 # ตรวจสอบข้อความเมื่อผู้ใช้กดปุ่ม "ตรวจสอบ"
 if st.button("ตรวจสอบ"):
     if user_input:
@@ -22,10 +23,14 @@ if st.button("ตรวจสอบ"):
         st.write("ผลลัพธ์:")
         st.write(f"ข้อความ: {user_input}")
         
-        # แก้ไขข้อความ "Staff" เป็น "Spam"
-        label = "Spam" if result[0]['label'] == 'Staff' else result[0]['label']
-        
-        st.write(f"ผลลัพธ์: {label}")
-        st.write(f"ความมั่นใจ: {result[0]['score']}")
+        # แสดงค่าความมั่นใจ
+        confidence = result[0]['score'] * 100  # แปลงค่าความมั่นใจเป็นเปอร์เซ็นต์
+        st.write(f"สปัม: {result[0]['label']} ({confidence:.2f}% ความมั่นใจ)")
+
+        # สร้างเงื่อนไขในการตัดสินใจ
+        if confidence >= 70:
+            st.warning("โมเดลมั่นใจว่าข้อความนี้เป็นสแปม")
+        else:
+            st.success("โมเดลมั่นใจว่าข้อความนี้ไม่เป็นสแปม")
     else:
         st.warning("โปรดป้อนข้อความก่อนตรวจสอบ")