Update app.py

app.py

@@ -38,15 +38,36 @@ sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=token
 def fetch_news(ticker):
     try:
         url = f"https://finviz.com/quote.ashx?t={ticker}"
-        headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'}
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+            'Accept-Language': 'en-US,en;q=0.5',
+            'Referer': 'https://finviz.com/',
+            'Connection': 'keep-alive',
+        }
         response = requests.get(url, headers=headers)
+        if response.status_code != 200:
+            st.error(f"Failed to fetch page for {ticker}: Status code {response.status_code}")
+            return []
+        
         soup = BeautifulSoup(response.text, 'html.parser')
+        title = soup.title.text if soup.title else ""
+        if ticker not in title:
+            st.error(f"Page for {ticker} not found or access denied.")
+            return []
+        
         news_table = soup.find(id='news-table')
+        if news_table is None:
+            st.error(f"News table not found for {ticker}. The website structure might have changed.")
+            return []
+        
         news = []
         for row in news_table.findAll('tr')[:50]:  # Fetch up to 50 articles
-            title = row.a.get_text()
-            link = row.a['href']
-            news.append({'title': title, 'link': link})
+            a_tag = row.find('a')
+            if a_tag:
+                title = a_tag.get_text()
+                link = a_tag['href']
+                news.append({'title': title, 'link': link})
         return news
     except Exception as e:
         st.error(f"Failed to fetch news for {ticker}: {e}")
@@ -66,6 +87,8 @@ st.markdown("""
 This tool parses stock tickers and analyzes the sentiment of related news articles.
 
 💡 *Example input:* `META, NVDA, AAPL, NTES, NCTY`
+
+**Note:** If news fetching fails, it might be due to changes in the Finviz website structure or access restrictions. Please verify the website manually or try again later.
 """)
 
 # Input field for stock tickers