first

app.py

@@ -15,8 +15,12 @@ def search_callback():
         if len(st.session_state.search_term) == 0:
             st.error("Please enter a search term")
             return
-        st.session_state.df = hf.get_tweets(st.session_state.search_term, st.session_state.num_tweets)
-        st.session_state.df = hf.get_sentiment(st.session_state.df)
+        try:
+            st.session_state.df = hf.get_tweets(st.session_state.search_term, st.session_state.num_tweets)
+            st.session_state.df = hf.get_sentiment(st.session_state.df)
+        except:
+            st.error("Please enter a valid search term")
+            return
 
 def twitter_form():
     with st.form(key="search_form"):
@@ -43,7 +47,6 @@ with st.sidebar:
     )
 
 if "df" in st.session_state:
-
     def make_dashboard(tweet_df, bar_color, wc_color):
         # first row
         col1, col2, col3 = st.columns([28, 34, 38])
@@ -98,16 +101,19 @@ if "df" in st.session_state:
 
     st.write(adjust_tab_font, unsafe_allow_html=True)
 
-    tab1, tab2, tab3, tab4 = st.tabs(["All", "Positive 😊", "Negative ☹️", "Neutral 😐"])
-    with tab1:
-        tweet_df = st.session_state.df
-        make_dashboard(tweet_df, bar_color="#1F77B4", wc_color="Blues")
-    with tab2:
-        tweet_df = st.session_state.df.query("Sentiment == 'Positive'")
-        make_dashboard(tweet_df, bar_color="#54A24B", wc_color="Greens")
-    with tab3:
-        tweet_df = st.session_state.df.query("Sentiment == 'Negative'")
-        make_dashboard(tweet_df, bar_color="#FF7F0E", wc_color="Oranges")
-    with tab4:
-        tweet_df = st.session_state.df.query("Sentiment == 'Neutral'")
-        make_dashboard(tweet_df, bar_color="#1F77B4", wc_color="Blues")
+    try:
+        tab1, tab2, tab3, tab4 = st.tabs(["All", "Positive 😊", "Negative ☹️", "Neutral 😐"])
+        with tab1:
+            tweet_df = st.session_state.df
+            make_dashboard(tweet_df, bar_color="#1F77B4", wc_color="Blues")
+        with tab2:
+            tweet_df = st.session_state.df.query("Sentiment == 'Positive'")
+            make_dashboard(tweet_df, bar_color="#54A24B", wc_color="Greens")
+        with tab3:
+            tweet_df = st.session_state.df.query("Sentiment == 'Negative'")
+            make_dashboard(tweet_df, bar_color="#FF7F0E", wc_color="Oranges")
+        with tab4:
+            tweet_df = st.session_state.df.query("Sentiment == 'Neutral'")
+            make_dashboard(tweet_df, bar_color="#1F77B4", wc_color="Blues")
+    except:
+        st.error("No plots to display.")