Update utils.py

utils.py

@@ -59,6 +59,12 @@ def analyze_company_data(company_name: str) -> Dict[str, Any]:
         # Calculate overall sentiment
         overall_sentiment = sentiment_analyzer.get_overall_sentiment(processed_articles)
         
+        # Ensure consistent array lengths in sentiment_scores
+        max_length = max(len(scores) for scores in sentiment_scores.values())
+        for source in sentiment_scores:
+            # Pad shorter arrays with 'neutral' to match the longest array
+            sentiment_scores[source].extend(['neutral'] * (max_length - len(sentiment_scores[source])))
+        
         return {
             "articles": processed_articles,
             "comparative_sentiment_score": sentiment_scores,