changes

ui/pages/rating.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from streamlit_star_rating import st_star_rating
 st.set_page_config(initial_sidebar_state="collapsed")
+from mega import Mega
 import glob
 import shutil
 import sys
@@ -23,6 +24,34 @@ rating = st_star_rating(label="rating", maxValue=10, defaultValue=3)
 submit_rating = st.button("Submit Rating")
 
 if submit_rating:
-    shutil.copy(generated_songs[0],f"../DataSet/Song/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.wav")
-    shutil.copy(spectrograms[0],f"../DataSet/Spec/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.npy")
-    st.switch_page("app.py")
+    # shutil.copy(generated_songs[0],f"../DataSet/Song/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.wav")
+    # shutil.copy(spectrograms[0],f"../DataSet/Spec/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.npy")
+    # st.switch_page("app.py")
+
+    
+    # uplaod to mega
+    mega = Mega()
+    m = mega.login('','')
+    
+    user = st.session_state['user_id']  # Assuming 'srija' is the user
+    model_name = st.session_state['model_name']
+    song_name = st.session_state['song_name']
+    similarity = st.session_state['similarity']
+    
+    # Construct the dynamic filenames
+    dynamic_song_name = f"{user}_{model_name}_{song_name}_{similarity}_{rating}.wav"
+    dynamic_spec_name = f"{user}_{model_name}_{song_name}_{similarity}_{rating}.npy"
+    
+    folder = mega.find('orpheus_data')
+    
+    # Rename and upload the generated song
+    generated_song_path = generated_songs[0]
+    mega.upload(generated_song_path, folder[0], dest_filename=dynamic_song_name)
+
+    # Rename and upload the spectrogram
+    spectrogram_path = spectrograms[0]
+    mega.upload(spectrogram_path, folder[0], dest_filename=dynamic_spec_name)
+
+    # Provide user feedback (optional)
+    st.success("Files uploaded successfully!")
+    st.switch_page("app.py")