Update utils.py

utils.py

@@ -44,10 +44,29 @@ def analyze_company_data(company_name: str) -> Dict[str, Any]:
             summary = text_summarizer.summarize(article['content'])
             article['summary'] = summary
             
-            # Analyze sentiment
+            # Analyze overall sentiment
             sentiment = sentiment_analyzer.analyze(article['content'])
             article['sentiment'] = sentiment
             
+            # Analyze fine-grained sentiment
+            try:
+                # Financial sentiment
+                if 'financial' in sentiment_analyzer.fine_grained_models:
+                    financial_sentiment = sentiment_analyzer.fine_grained_models['financial'](article['content'])[0]
+                    article['financial_sentiment'] = financial_sentiment['label']
+                
+                # Emotional sentiment
+                if 'emotion' in sentiment_analyzer.fine_grained_models:
+                    emotional_sentiment = sentiment_analyzer.fine_grained_models['emotion'](article['content'])[0]
+                    article['emotional_sentiment'] = emotional_sentiment['label']
+                
+                # ESG sentiment
+                if 'esg' in sentiment_analyzer.fine_grained_models:
+                    esg_sentiment = sentiment_analyzer.fine_grained_models['esg'](article['content'])[0]
+                    article['esg_sentiment'] = esg_sentiment['label']
+            except Exception as e:
+                print(f"Error in fine-grained sentiment analysis: {str(e)}")
+            
             # Track sentiment by source
             source = article['source']
             if source not in sentiment_scores: