Update backupapp.py

backupapp.py

@@ -39,12 +39,21 @@ def songupdate():
 def load_song_file(filename):
     with open(filename, "r") as file:
         chord_sheet = file.read()
-        st.text_area(label="Enhanced Chord Sheet", value=chord_sheet, height=300, key="EnhancedChordSheet", help="This text can be read due to alternating chord lines and lyric lines.", label_visibility="visible", on_change=songupdate )
-        processed_sheet = process_chord_sheet(chord_sheet)
-        st.markdown(processed_sheet, unsafe_allow_html=True)
+        
+    st.text_area(label="Enhanced Chord Sheet", value=chord_sheet, height=300, key="EnhancedChordSheet", help="This text can be read due to alternating chord lines and lyric lines.", label_visibility="visible", on_change=songupdate )
+    processed_sheet = process_chord_sheet(chord_sheet)
+    st.markdown(processed_sheet, unsafe_allow_html=True)
+    
+    st.session_state['selected_file'] = filename
+    st.session_state['chord_sheet'] = chord_sheet
+    st.session_state['song_name'], st.session_state['artist_name'] = os.path.splitext(filename)[0].split(' by ')
+
+def update_display():
+    st.session_state['display_sheet'] = process_chord_sheet(st.session_state['chord_sheet'])
 
 def main():
     st.title('🎵 Song Files')
+    
     with st.expander("Select Song File", expanded=True):
         all_files = [f for f in glob.glob("*.txt") if ' by ' in f]
         selected_file = st.selectbox("Choose a file", all_files)
@@ -54,25 +63,31 @@ def main():
             artist_name = artist_name.replace("_", " ")
         else:
             song_name, artist_name = "", ""
+            
     col1, col2 = st.columns([4, 1])
     with col1:
-        song_name_input = st.text_input("🎵 Song Name", value=song_name)
-        artist_name_input = st.text_input("🎤 Artist Name", value=artist_name)
+        song_name = st.text_input("🎵 Song Name", value=st.session_state.get('song_name', ''))
+        artist_name = st.text_input("🎤 Artist Name", value=st.session_state.get('artist_name', ''))
+        chord_sheet_area = st.text_area("Chord Sheet", value=st.session_state.get('chord_sheet', ''), height=300, on_change=update_display)
 
         # Load chord sheet from selected file into the text area
         chord_sheet_input = ""
         if selected_file:
             with open(selected_file, "r") as file:
                 chord_sheet_input = file.read()
-        chord_sheet_area = st.text_area("Chord Sheet", value=chord_sheet_input, height=300)
+        #chord_sheet_area = st.text_area("Chord Sheet", value=chord_sheet_input, height=300)
 
         # Save functionality
         if st.button("💾 Save", key="save_song"):
-            if song_name_input and artist_name_input:
-                filename = f"{song_name_input} by {artist_name_input}.txt".replace(" ", "_")
+            if song_name and artist_name:
+                filename = f"{song_name} by {artist_name}.txt".replace(" ", "_")
                 with open(filename, "w") as file:
                     file.write(chord_sheet_area)
                 st.success("Chord sheet saved.")
+                # Refresh the file list and selected file
+                st.session_state['selected_file'] = filename
+                st.session_state['chord_sheet'] = chord_sheet_area
+                update_display()
             else:
                 st.error("Both Song Name and Artist Name are required.")
 
@@ -98,5 +113,13 @@ def main():
             st.markdown(f"[🎸Chords]({create_search_url_chords(song_info)})")
             st.markdown(f"[🎶Lyrics]({create_search_url_lyrics(song_info)})")
 
+# Initialize session state variables if not already set
+if 'display_sheet' not in st.session_state:
+    st.session_state['display_sheet'] = ''
+    st.session_state['chord_sheet'] = ''
+    st.session_state['selected_file'] = ''
+    st.session_state['song_name'] = ''
+    st.session_state['artist_name'] = ''
+    
 if __name__ == '__main__':
     main()