Update app.py

app.py

@@ -207,7 +207,7 @@ if (st.session_state.rerun_button == "QUERYING") and (st.session_state["video_li
         try:
             yt_parser.query_comments(st.session_state['video_link'], st.session_state['max_comments'])
         except googleapiclient.errors.HttpError:
-            st.error("Error: Unable to query comments, please check your API key.")
+            st.error("Error: Unable to query comments, incorrect YouTube URL or check your API key.")
             st.stop()
         except:
             st.error("Error: Unable to query comments, incorrect YouTube URL or maximum \
@@ -215,12 +215,20 @@ if (st.session_state.rerun_button == "QUERYING") and (st.session_state["video_li
             st.stop()
 
         # Run formatting and models
-        yt_parser.format_comments()
-        yt_parser.clean_comments()
-        yt_parser.run_sentiment_pipeline(sentiment_pipeline)
-        yt_parser.run_topic_modelling_pipeline(embedding_model,
-                                               nlp=spacy_nlp,
-                                               max_topics=st.session_state['max_topics'])
+        try:
+            yt_parser.format_comments()
+            yt_parser.clean_comments()
+            yt_parser.run_sentiment_pipeline(sentiment_pipeline)
+            yt_parser.run_topic_modelling_pipeline(embedding_model,
+                                                   nlp=spacy_nlp,
+                                                   max_topics=st.session_state['max_topics'])
+        except ValueError:
+            st.error("Error: Oops there are not enough comments to analyse, please try a different video.")
+            st.stop()
+        except:
+            st.error("Error: Oops there's an issue on our end, please wait a moment and try again.")
+            st.stop()
+            
     # Set "QUERY COMPLETE" to bypass running this section on script re-run
     st.session_state.rerun_button = "QUERY COMPLETE"