Upload app.py

app.py

@@ -816,7 +816,9 @@ def analyze_news(news_text):
         fake_confidence = 1 - combined_confidence
     
         # Build the detailed report with better formatting
-        prediction_emoji = "✅" if distilbert_prediction == "REAL" else "❌" if distilbert_prediction == "FAKE" else "❓"
+        # Use combined_confidence to determine the final classification (not just DistilBERT)
+        final_prediction = "REAL" if combined_confidence > 0.5 else "FAKE" if combined_confidence < 0.5 else "UNCERTAIN"
+        prediction_emoji = "✅" if final_prediction == "REAL" else "❌" if final_prediction == "FAKE" else "❓"
         confidence_level = "Cao" if combined_confidence > 0.7 else "Trung bình" if combined_confidence > 0.4 else "Thấp"
         confidence_emoji = "🟢" if combined_confidence > 0.7 else "🟡" if combined_confidence > 0.4 else "🔴"
         
@@ -860,13 +862,13 @@ def analyze_news(news_text):
 ## 🔍 **KẾT QUẢ PHÂN TÍCH TIN TỨC**
 
 <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 20px; border-radius: 10px; margin: 20px 0; text-align: center;">
-<h2 style="margin: 0; font-size: 24px;">{prediction_emoji} {'TIN THẬT' if distilbert_prediction == 'REAL' else 'TIN GIẢ' if distilbert_prediction == 'FAKE' else 'KHÔNG XÁC ĐỊNH'}</h2>
+<h2 style="margin: 0; font-size: 24px;">{prediction_emoji} {'TIN THẬT' if final_prediction == 'REAL' else 'TIN GIẢ' if final_prediction == 'FAKE' else 'KHÔNG XÁC ĐỊNH'}</h2>
 <p style="margin: 10px 0 0 0; font-size: 18px; opacity: 0.9;">{confidence_emoji} Độ tin cậy: {confidence_level} ({combined_confidence:.0%})</p>
 </div>
 
 ### 🤖 **Phân tích bằng AI**
 <div style="background: #f8f9fa; padding: 15px; border-radius: 8px; border-left: 4px solid #007bff; margin: 10px 0;">
-<p><strong>Kết quả:</strong> {prediction_emoji} {'Tin tức này có vẻ THẬT' if distilbert_prediction == 'REAL' else 'Tin tức này có vẻ GIẢ' if distilbert_prediction == 'FAKE' else 'Không thể xác định'}</p>
+<p><strong>Kết quả:</strong> {prediction_emoji} {'Tin tức này có vẻ THẬT' if final_prediction == 'REAL' else 'Tin tức này có vẻ GIẢ' if final_prediction == 'FAKE' else 'Không thể xác định'}</p>
 <p><strong>Độ chắc chắn:</strong> {f"{distilbert_confidence:.0%}" if distilbert_confidence else 'Không có'} - {'Rất cao' if distilbert_confidence and distilbert_confidence > 0.8 else 'Cao' if distilbert_confidence and distilbert_confidence > 0.6 else 'Trung bình' if distilbert_confidence and distilbert_confidence > 0.4 else 'Thấp'}</p>
 </div>
 
@@ -889,7 +891,7 @@ def analyze_news(news_text):
 
 ### 📊 **KẾT LUẬN CUỐI CÙNG**
 <div style="background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%); color: white; padding: 15px; border-radius: 8px; margin: 20px 0;">
-<p style="margin: 0; font-size: 16px;"><strong>Tin tức này có khả năng {'THẬT' if real_confidence > fake_confidence else 'GIẢ'} với độ tin cậy {max(real_confidence, fake_confidence):.0%}</strong></p>
+<p style="margin: 0; font-size: 16px;"><strong>Tin tức này có khả năng {'THẬT' if final_prediction == 'REAL' else 'GIẢ' if final_prediction == 'FAKE' else 'KHÔNG XÁC ĐỊNH'} với độ tin cậy {max(real_confidence, fake_confidence):.0%}</strong></p>
 <p style="margin: 5px 0 0 0; font-size: 14px; opacity: 0.9;">Dựa trên phân tích AI, kiểm tra nguồn tin và đánh giá thông minh</p>
 </div>