update

visualization/roberta_visualizer.py

@@ -64,8 +64,23 @@ def create_sentiment_visualization(analysis_results):
                     for model_name in models:
                         if model_name in sa_data:
                             model_result = sa_data[model_name]
-                            score = model_result.get("sentiment_score", 0)
-                            label = model_result.get("label", "neutral").capitalize()
+                            if model_result is not None:
+                                score = model_result.get("sentiment_score", 0)
+                                label = model_result.get("label", "neutral").capitalize()
+                            else:
+                                score = 0
+                                label = "Neutral"
+                            for model_name in models:
+                                if model_name in sa_data:
+                                    model_result = sa_data[model_name]
+                                    if model_result is not None:
+                                        # Now safely access model_result
+                                        score = model_result.get("sentiment_score", 0)
+                                        label = model_result.get("label", "neutral").capitalize()
+                                    else:
+                                        # Provide defaults when model_result is None
+                                        score = 0
+                                        label = "Neutral"
                             
                             # Set color based on sentiment
                             if label.lower() == "positive":