Update app.py

app.py

@@ -6,7 +6,6 @@ from PIL import Image
 import base64
 from transformers import pipeline
 import spacy
-import googleapiclient
 import numpy as np
 from sentence_transformers import SentenceTransformer
 from matplotlib import colormaps
@@ -199,17 +198,23 @@ if (st.session_state.rerun_button == "QUERYING") and (st.session_state["video_li
  yt_parser.scrape_comments(st.session_state['video_link'])
  yt_parser.scrape_video_title()
  except:
- st.error("Error: Unable to query comments, incorrect YouTube URL or maximum \
- API call limit reached.")
+ st.error("Error: Unable to query comments, incorrect YouTube URL.")
  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"