Update app.py

app.py

@@ -9,8 +9,6 @@ import hmac
 import os
 import time
 
-feed_url1 = 'https://trends.google.de/trends/trendingsearches/daily/rss?geo=AT'
-
 def parse_url(url):
     response = requests.get(url)
 
@@ -118,20 +116,19 @@ def display_trends_from_yesterday():
 def display_articles_for_category(category):
     checkbox_statuses = {}
     urls = []
-    
     # Dictionary zur Verwaltung des Status jeder Checkbox
-    for index, row in st.session_state["real_trending_searches"][category].iterrows():
-        articles = find_details(st.session_state["base_data"][category], row['title'])
+    for index, row in st.session_state["real_trending_searches"][selected_country][category].iterrows():
+        articles = find_details(st.session_state["base_data"][selected_country][category], row['title'])
         for count2, url in enumerate(articles, start=1):
             checkbox_label = f"{category}_{index}_{count2}"
             if st.session_state.reset:
                 st.session_state[checkbox_label] = False
             checkbox_statuses[checkbox_label] = st.session_state.get(checkbox_label, False)
 
-    for index, row in st.session_state["real_trending_searches"][category].iterrows():
+    for index, row in st.session_state["real_trending_searches"][selected_country][category].iterrows():
         count = index + 1
         with st.expander(f"{count}•   {row['title']} "):
-            articles = find_details(st.session_state["base_data"][category], row['title'])
+            articles = find_details(st.session_state["base_data"][selected_country][category], row['title'])
             for count2, url in enumerate(articles, start=1):
                 checkbox_label = f"{category}_{index}_{count2}"
                 disabled = not checkbox_statuses[checkbox_label] and sum(checkbox_statuses.values()) >= MAX_CHECKED
@@ -192,6 +189,32 @@ categories = {
     "Technik": "t",
 }
 
+country_list = {
+    "Germamy" : "DE",
+    "Austria" : "AT"
+}
+
+pytrend = TrendReq(hl='de-AT', tz=360, timeout=(10,50))
+
+if 'base_load_finished' not in st.session_state:
+    st.session_state["real_trending_searches"] = {}
+    st.session_state["base_data"] = {}
+    st.session_state["pn"] = "AT"
+
+if 'base_load_finished' not in st.session_state or st.session_state.reset:
+    with st.spinner("Loading Trends"):
+        st.session_state["today"] = {}
+        for country_name, pn_option in country_list.items():
+            st.session_state["base_data"][pn_option] = {}
+            st.session_state["real_trending_searches"][pn_option] = {}
+            st.session_state["today"][pn_option] = pytrend.today_searches(pn=pn_option)
+
+            for category_name, category_code in categories.items():
+                    st.session_state["base"] = pytrend.realtime_trending_searches(pn=pn_option, cat=category_code, count=75)
+                    st.session_state["base_data"][pn_option][category_name] = st.session_state["base"]
+                    st.session_state["real_trending_searches"][pn_option][category_name] = convert_into_pd(st.session_state["base"])
+
+
 MAX_CHECKED = 3
 
 def check_password():
@@ -221,46 +244,39 @@ def check_password():
 if not check_password():
     st.stop()  # Do not continue if check_password is not True.
 
-pytrend = TrendReq(hl='de-AT', tz=360, timeout=(10,50))
 
-if 'base_load_finished' not in st.session_state:
-    st.session_state["real_trending_searches"] = {}
-    st.session_state["base_data"] = {}
-for category_name, category_code in categories.items():
-    if 'base_load_finished' not in st.session_state:
-        st.session_state["base"] = pytrend.realtime_trending_searches(pn='AT', cat=category_code, count=75)
-        st.session_state["base_data"][category_name] = st.session_state["base"]
-        st.session_state["real_trending_searches"][category_name] = convert_into_pd(st.session_state["base"])
-    elif st.session_state.reset:
-        st.session_state["base"] = pytrend.realtime_trending_searches(pn='AT', cat=category_code, count=75)
-        st.session_state["base_data"][category_name] = st.session_state["base"]
-        st.session_state["real_trending_searches"][category_name] = convert_into_pd(st.session_state["base"])
+    
+
 
 if 'selected_option' not in st.session_state:
     st.session_state['selected_option'] = "default_value"  # You can set a default value as needed
 
 # Now, you can safely use st.session_state['selected_option']
+# Selectbox to choose a country
+selected_country = st.sidebar.selectbox("Choose a Country", ["AT", "DE"])
+feed_url1 = f'https://trends.google.de/trends/trendingsearches/daily/rss?geo={selected_country}'
+
+# Button to trigger actions
+if st.sidebar.button("Change Country"):
+    if selected_country == "AT":
+        st.session_state["pn"] = selected_country
+    elif selected_country == "DE":
+        st.session_state["pn"] = selected_country
+
 selected_option = st.sidebar.radio("Choose an option", ["Realzeit Anfragen", "Tagesaktuelle Anfragen", "Trends von Gestern"])
 st.warning("Die aufgelisteten Keywörter für erhöhte Reichweite in den Überschriften verwenden")
 if selected_option == "Tagesaktuelle Anfragen":    
-    if 'today' not in st.session_state:
-        st.session_state["today"] = pytrend.today_searches(pn="AT")
-    elif st.session_state.reset:
-        st.session_state["today"] = pytrend.today_searches(pn="AT")
-
-    #today = pytrend.today_searches(pn="AT")
-    #trending_searches = pytrend.trending_searches(pn="austria")
-    for count, index in enumerate(st.session_state["today"], start=0):
+
+    for count, index in enumerate(st.session_state["today"][selected_country], start=0):
         try:
             display_articles_for_today(count, index)
         except Exception as e:
             st.code(e)
             continue
 elif selected_option == "Realzeit Anfragen":
-    choices_list = list(st.session_state["real_trending_searches"].keys())
+    choices_list = list(st.session_state["real_trending_searches"][selected_country].keys())
     if len(categories) == len(choices_list):
         st.session_state["base_load_finished"] = True
-    print(choices_list)
     auswahl = st.selectbox("Select Ressort", choices_list)
 
     display_articles_for_category(auswahl)