Update StockSentimentAnalyser.py

StockSentimentAnalyser.py

@@ -456,8 +456,45 @@ class StockSentimentAnalyzer:
         """Get comprehensive sentiment analysis for a stock"""
         # If company name not provided, try to extract from symbol
         self.symbol = symbol
-        if not company_name:
-            company_name = symbol.replace('.NS', '').replace('.BO', '')
+            # If company name not provided, try to extract from symbol
+    if not company_name:
+        try:
+            # Add .NS for NSE stocks if not present
+            if not symbol.endswith('.NS') and not symbol.endswith('.BO'):
+                symbol_with_suffix = symbol + '.NS'
+            else:
+                symbol_with_suffix = symbol
+            
+            # Get company info from yfinance
+            ticker = yf.Ticker(symbol_with_suffix)
+            info = ticker.info
+            
+            # Extract company name with multiple fallbacks
+            company_name = (
+                info.get('longName') or 
+                info.get('shortName') or 
+                info.get('name') or 
+                symbol  # Final fallback to symbol
+            )
+            
+            # Validate the extracted name
+            if company_name:
+                # Remove special characters and check if meaningful
+                cleaned_name = ''.join(c for c in company_name if c.isalnum() or c in (' ', '-', '&'))
+                if (len(cleaned_name.strip()) < 2 or  # Too short
+                    cleaned_name.strip() == symbol or  # Same as symbol
+                    any(x in cleaned_name for x in ['-', ' - ']) or  # Contains dashes (likely placeholder)
+                    cleaned_name.isnumeric()):  # Just numbers
+                    company_name = symbol  # Fallback to symbol if name is invalid
+            else:
+                company_name = symbol
+                
+        except Exception as e:
+            print(f"⚠️ Could not fetch company name for {symbol}: {str(e)}")
+            company_name = symbol
+    
+    # Ensure we have at least the symbol as name
+    company_name = company_name or symbol
 
         print(f"\n{'='*80}")
         print(f"🔍 ANALYZING: {company_name.upper()} ({symbol})")