commit

frontend/src/components/ClusterCard.jsx

@@ -62,34 +62,20 @@ export default function ClusterCard({ cluster }) {
                                                 {article.title}
                                             </h4>
                                             <div className="flex flex-wrap gap-2">
-                                                {/* Clickbait Label */}
-                                                {article.is_clickbait === true && (
+                                                {article.is_clickbait && (
                                                     <span className="text-[10px] uppercase font-bold text-red-400 bg-red-900/30 border border-red-900/50 px-1.5 py-0.5 rounded">
                                                         Clickbait
                                                     </span>
                                                 )}
-                                                {article.is_clickbait === false && (
-                                                    <span className="text-[10px] uppercase font-bold text-emerald-400 bg-emerald-900/30 border border-emerald-900/50 px-1.5 py-0.5 rounded">
-                                                        Objective
-                                                    </span>
-                                                )}
-
-                                                {/* Sensationalism Label */}
-                                                {article.is_sensationalist === true && (
+                                                {article.is_sensationalist && (
                                                     <span className="text-[10px] uppercase font-bold text-amber-400 bg-amber-900/30 border border-amber-900/50 px-1.5 py-0.5 rounded">
                                                         Sensationalist
                                                     </span>
                                                 )}
-                                                {article.is_sensationalist === false && (
-                                                    <span className="text-[10px] uppercase font-bold text-slate-400 bg-slate-800 border border-slate-700 px-1.5 py-0.5 rounded">
-                                                        Neutral
-                                                    </span>
-                                                )}
                                             </div>
                                         </div>
                                         <ExternalLink className="w-4 h-4 text-slate-500 shrink-0 group-hover:text-blue-400 mt-1" />
                                     </div>
-                                    {/* Info footer */}
                                     <div className="mt-2 flex items-center gap-2 text-xs text-slate-500">
                                         <span className="font-semibold text-slate-400">{article.newspaper}</span>
                                         <span>•</span>

main.py

@@ -172,9 +172,8 @@ def get_clusters():
                     "newspaper": row.get('newspaper', 'Unknown'),
                     "url": row.get('article_url', '#'),
                     "newspaper_url": row.get('newspaper_url', '#'),
-                    # Robust boolean conversion handling strings like 'True', 'False'
-                    "is_clickbait": (str(row.get('is_clickbait')).lower() == 'true' or row.get('is_clickbait') is True) if pd.notna(row.get('is_clickbait')) else None,
-                    "is_sensationalist": (str(row.get('is_sensationalist')).lower() == 'true' or row.get('is_sensationalist') is True) if pd.notna(row.get('is_sensationalist')) else None
+                    "is_clickbait": bool(row.get('is_clickbait', False)) if pd.notna(row.get('is_clickbait')) else None,
+                    "is_sensationalist": bool(row.get('is_sensationalist', False)) if pd.notna(row.get('is_sensationalist')) else None
                 })
             
             # Size calc
@@ -247,22 +246,13 @@ def analyze_article(request: AnalyzeRequest):
         prob_sensationalist = sens_probs[0][1].item()
         is_sensationalist = prob_sensationalist >= sens_threshold
         
-        # Calculate confidences of the PREDICTED class
-        # If is_clickbait is True, we want prob of Clickbait (prob_clickbait)
-        # If is_clickbait is False, we want prob of NOT Clickbait (1 - prob_clickbait or prob_not_clickbait)
-        # Clickbait model: Label 0 = Clickbait
-        conf_clickbait = prob_clickbait if is_clickbait else cb_probs[0][1].item()
-        
-        # Sensationalism model: Label 1 = Sensationalist
-        conf_sensationalist = prob_sensationalist if is_sensationalist else sens_probs[0][0].item()
-
         return {
             "title": title,
             "text_snippet": text[:200] + "..." if len(text) > 200 else text,
             "is_clickbait": is_clickbait,
             "is_sensationalist": is_sensationalist,
-            "clickbait_conf": float(conf_clickbait),
-            "sensationalist_conf": float(conf_sensationalist)
+            "clickbait_conf": float(prob_clickbait), # Sending confidence of being clickbait
+            "sensationalist_conf": float(prob_sensationalist)
         }
         
     except Exception as e: