Update songscope.py

songscope.py

@@ -1,3 +1,4 @@
+import streamlit as st
 import pandas as pd
 import requests
 from bs4 import BeautifulSoup
@@ -65,7 +66,7 @@ def random_delay(min_val: float, max_val: float, print_delay: bool = False):
     val = random.uniform(min_val, max_val)
     time.sleep(val)
     if print_delay == True:
-        print(f"Delayed {val} seconds")
+        st.write(f"Delayed {val} seconds")
     return val
 
 def find_artist(artist_name: str):
@@ -170,9 +171,9 @@ def follow_lyrics(lyric_url: str):
             # Clean up the lyrics by removing unnecessary HTML tags and whitespace
             lyrics_str = lyrics_div.get_text(strip = False)
         else:
-            print(f"Error: Unable to find the lyrics for '{lyric_url}'.")
+            st.write(f"Error: Unable to find the lyrics for '{lyric_url}'.")
     else:
-        print(f"Error: Unable to fetch the webpage. Status code: {response.status_code}")
+        st.write(f"Error: Unable to fetch the webpage. Status code: {response.status_code}")
 
     return lyrics_str
 
@@ -199,7 +200,7 @@ def find_artist(artist_name: str) -> str:
     
     # The target URL
     url = f"https://www.azlyrics.com/{first_letter}.html"
-    print(url)
+    st.write(url)
     
     # Send an HTTP request to the URL
     response = requests.get(url)
@@ -226,7 +227,7 @@ def find_artist(artist_name: str) -> str:
     for url in artist_links:
         artist_urls.append(str(url).split("/")[-1][:-5])
 
-    print(artist_urls)
+    st.write(artist_urls)
 
     if artist_name in artist_urls:
         return f"https://www.azlyrics.com/{artist_links[artist_urls.index(artist_name)]}"
@@ -462,12 +463,12 @@ def get_all_data(artist_name: str, song_titles: list = None,
         A pandas DataFrame containing metadata and sentiment analysis for each song found.
     """
     if print_progress == True:
-        print(f"------------------------\n\nFinding song data for '{artist_name}'. This may take a few moments...")
+        st.write(f"------------------------\n\nFinding song data for '{artist_name}'. This may take a few moments...")
 
     artist_data = get_metadata(artist_name = artist_name, song_titles = song_titles)
 
     if print_progress == True:
-        print(f"\n\t- All metadata found")
+        st.write(f"\n\t- All metadata found")
 
     times = []
 
@@ -477,7 +478,7 @@ def get_all_data(artist_name: str, song_titles: list = None,
         #     lyrics = follow_lyrics(lyric_url = artist_data[title]['url'])
         #     artist_data[title]['lyrics'] = sectionize(lyrics)
         # except: (UnboundLocalError, TypeError, AttributeError)
-        #     print(f"\tCouldn't find lyrics to {title}. Moving to next song.")
+        #     st.write(f"\tCouldn't find lyrics to {title}. Moving to next song.")
         #     pass
         
         lyrics = follow_lyrics(lyric_url = artist_data[title]['url'])
@@ -495,13 +496,13 @@ def get_all_data(artist_name: str, song_titles: list = None,
         if print_progress == True:
             if remaining >= 60: # more than one minute remaining
                 remaining = round(remaining / 60, 2)
-                print(f"\t- Lyrics to '{title}' found. Estimated time remaining: {remaining} minutes")
+                st.write(f"\t- Lyrics to '{title}' found. Estimated time remaining: {remaining} minutes")
             else: # less than one minute remaining
                 remaining = round(remaining, 2)
-                print(f"\t- Lyrics to '{title}' found. Estimated time remaining: {remaining} seconds")
+                st.write(f"\t- Lyrics to '{title}' found. Estimated time remaining: {remaining} seconds")
 
     if print_progress == True:
-        print(f"\nAll lyrics and metadata found. Returning structured data.")
+        st.write(f"\nAll lyrics and metadata found. Returning structured data.")
 
     df_dict = {}
     df_dict['artist_name'] = []
@@ -546,6 +547,6 @@ def get_all_data(artist_name: str, song_titles: list = None,
                          'sentiment_valence', "album_name", "release_type", "lyrics_url"]]
 
     if print_progress == True:
-        print(f"Data retrieval complete!\n\n------------------------")
+        st.write(f"Data retrieval complete!\n\n------------------------")
 
     return sents_df