Update app.py

app.py

@@ -36,59 +36,75 @@ def create_search_url_lyrics(artist_song):
 def songupdate():
     st.write(st.session_state.EnhancedChordSheet)
 
-def load_song_file(filename):
+def load_song_file2(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 )
+        st.session_state['chord_sheet'] = chord_sheet  # Load the content into session state
         processed_sheet = process_chord_sheet(chord_sheet)
         st.markdown(processed_sheet, unsafe_allow_html=True)
 
+def load_song_file(filename):
+    with open(filename, "r") as file:
+        chord_sheet = file.read()
+        # Instead of directly modifying the session state, return the chord sheet content
+        return chord_sheet
+
+def song_update():
+    if 'selected_file' in st.session_state:
+        song_name, artist_name = parse_filename(st.session_state.selected_file)
+        st.session_state.song_name = song_name
+        st.session_state.artist_name = artist_name
+
+def parse_filename(filename):
+    base_name = os.path.splitext(filename)[0]
+    song_name, artist_name = base_name.split(' by ')
+    return song_name.replace("_", " "), artist_name.replace("_", " ")
 
-# Automatic save function
 def auto_save():
-    song_name = st.session_state['song_name']
-    artist_name = st.session_state['artist_name']
-    chord_sheet = st.session_state['chord_sheet']
+    song_name = st.session_state.get('song_name', '')
+    artist_name = st.session_state.get('artist_name', '')
+    chord_sheet = st.session_state.get('chord_sheet', '')
 
-    if song_name and artist_name:
-        filename = f"{song_name} by {artist_name}.txt".replace(" ", "_")
+    if song_name and artist_name and chord_sheet:
+        filename = song_name + " by " + artist_name + ".txt"
         with open(filename, "w") as file:
-            file.write(chord_sheet)
-
-        # Display the character count below the text area
+            
+            
+            chord_sheet_text = st.session_state.get('chord_sheet', '')
+            file.write(chord_sheet_text)
+            
         st.session_state['char_count'] = len(chord_sheet)
+        st.success(f"Auto-saved to {filename}")
 
 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)
+        selected_file = st.selectbox("Choose a file", all_files, on_change=song_update, key='selected_file')
 
     col1, col2 = st.columns([4, 1])
+
     with col1:
-        song_name_input = st.text_input("🎵 Song Name", key='song_name')
-        artist_name_input = st.text_input("🎤 Artist Name", key='artist_name')
+        song_name_input = st.text_input("🎵 Song Name", key='song_name', on_change=auto_save)
+        artist_name_input = st.text_input("🎤 Artist Name", key='artist_name', on_change=auto_save)
+
+        if 'selected_file' in st.session_state and st.session_state.selected_file:
+            # Update the session state before creating the text area widget
+            st.session_state['chord_sheet'] = load_song_file(st.session_state.selected_file)
+
+        # Now create the text area widget with the updated session state
+        chord_sheet_area = st.text_area("Chord Sheet", value=st.session_state.get('chord_sheet', ''), height=300, key='chord_sheet', on_change=auto_save)
 
-        # Load chord sheet from selected file into the text area
-        if selected_file:
-            with open(selected_file, "r") as file:
-                chord_sheet_input = file.read()
-                st.session_state['chord_sheet'] = chord_sheet_input
-            chord_sheet_area = st.text_area("Chord Sheet", value=chord_sheet_input, height=300, key='chord_sheet', on_change=auto_save)
-        else:
-            chord_sheet_area = st.text_area("Chord Sheet", height=300, key='chord_sheet', on_change=auto_save)
-
-        # Display the character count below the text area
         char_count_msg = f"Character Count: {st.session_state.get('char_count', 0)}"
         st.write(char_count_msg)
         
         # 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(" ", "_")
+                filename = song_name_input + " by " + artist_name_input + ".txt"
                 with open(filename, "w") as file:
                     file.write(chord_sheet_area)
-                st.success("Chord sheet saved.")
+                st.success("Chord sheet saved to file: " + filename)
             else:
                 st.error("Both Song Name and Artist Name are required.")
 
@@ -116,6 +132,11 @@ def main():
             st.markdown(f"[🎸Chords]({create_search_url_chords(song_info)})")
             st.markdown(f"[🎶Lyrics]({create_search_url_lyrics(song_info)})")
 
+
+    # Load chord sheet from selected file into the text area
+    if 'selected_file' in st.session_state and st.session_state.selected_file:
+        load_song_file(st.session_state.selected_file)
+
         
 if __name__ == '__main__':
     main()