Update app.py

app.py

@@ -7,17 +7,37 @@ from bs4 import BeautifulSoup
 generator = pipeline('text-generation', model='dbmdz/german-gpt2')
 
 # Define a function to fetch trending news related to a specific niche
+import streamlit as st
+from transformers import pipeline
+import requests
+from bs4 import BeautifulSoup
+
+# Initialize a text generation pipeline
+generator = pipeline('text-generation', model='dbmdz/german-gpt2')
+
 def fetch_trending_news(niche):
     url = f"https://www.google.com/search?q={niche}+news&tbs=qdr:d"
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"}
-    response = requests.get(url, headers=headers)
-    print("Response status code:", response.status_code)  # Debug print
-    soup = BeautifulSoup(response.content, "html.parser")
-    news_items = soup.find_all("div", class_="BNeawe vvjwJb AP7Wnd")
-    print("Number of news items found:", len(news_items))  # Debug print
-    trending_news = [item.text for item in news_items[:5]]  # Extract the top 5 news items
-    return trending_news
+    try:
+        response = requests.get(url, headers=headers)
+        if response.status_code == 200:
+            soup = BeautifulSoup(response.content, "html.parser")
+            # Adjusted to use more generic selectors that might be more stable
+            news_items = soup.find_all("div", class_="ZINbbc xpd O9g5cc uUPGi")
+            if not news_items:
+                print("No news items found, check your selectors.")
+                return []
+            trending_news = [item.find("div", class_="BNeawe vvjwJb AP7Wnd").text for item in news_items[:5]]
+            return trending_news
+        else:
+            print(f"Failed to fetch news, status code: {response.status_code}")
+            return []
+    except Exception as e:
+        print(f"Error fetching news: {e}")
+        return []
+
+
 
 # Define the pages
 def page_trending_niche():