Update app.py

app.py

@@ -7,6 +7,12 @@ st.title("SongScope")
 st.write("Welcome to SongScope. This is a web app that lets you download lyrics and metadata of your favourite artist.\n")
 st.write("Enter an artist name, and you can even select up to 5 songs to retrieve their data. If you don't specify any songs,\ndata on all songs by the artist will be retrieved.")
 
+# delay times
+delay_min = int(st.text_input("Minimum delay per iteration"))
+delay_max = int(st.text_input("Maximum delay per iteration"))
+# delay_min = 15
+# delay_max = 30
+
 # user-provided data
 artist_name = st.text_input("Enter artist name here")
 song_1 = st.text_input("Enter 1st song title here")
@@ -19,6 +25,11 @@ song_5 = st.text_input("Enter 5th song title here")
 #################################################################
 
 # PUT GET DATA FUNCTION HERE
+artist_df = get_all_data(artist_name = artist_name, 
+             song_titles = song_1 + song_2 + song_3 + song_4 + song_5,
+             delay = (delay_min, delay_max), # random float value between (delay_min, delay_max) seconds
+             print_progress = True)
+# artist_df.to_csv(f"{artist_name.lower()}_data.csv")
 
 #################################################################
 
@@ -26,9 +37,9 @@ song_5 = st.text_input("Enter 5th song title here")
 csv_dl_button = st.download_button('Download Data', DATA_TO_CSV, 'text/csv')
 # st.download_button('Download Data', DATA_TO_CSV)  # Defaults to 'text/plain'
 
-with open(f'{artist_name.lower()}_data.csv') as f:
-    st.download_button('Download CSV', f)  # Defaults to 'text/plain'
-    st.write("Your data should now be downloading.")
+# with open(f'{artist_name.lower()}_data.csv') as f:
+#     st.download_button('Download CSV', f)  # Defaults to 'text/plain'
+#     st.write("Your data should now be downloading.")
     
 # fun facts
 fun_fax = []
@@ -38,6 +49,9 @@ with open("fun_fax.txt", "r", encoding = "utf-8") as f:
             fun_fax.append(word)
 f.close() # closes connection to file
 
+if csv_dl_button:
+    st.write("Your data should be downloaded. If not, please reload the page and try again.")
+
 st.write(fun_fax[random.randint(0, len(fun_fax) - 1)])
 
 # "thank you" blurb