Update app.py

app.py

@@ -3,117 +3,89 @@ import streamlit.components.v1 as components
 import re
 import os
 import glob
+
+# Set Streamlit page configuration
 st.set_page_config(layout="wide")
+
 def process_line(line):
+    # Process each line to replace chords with images
     if re.search(r'\b[A-G][#b]?m?\b', line):
         line = re.sub(r'\b([A-G][#b]?m?)\b', r"<img src='\1.png' style='height:20px;'>", line)
     return line
+
 def process_chord_sheet(chord_sheet):
-    processed_lines = []
-    for line in chord_sheet.split('\n'):
-        processed_line = process_line(line)
-        processed_lines.append(processed_line)
+    # Process the entire chord sheet
+    processed_lines = [process_line(line) for line in chord_sheet.split('\n')]
     return '<br>'.join(processed_lines)
-def create_search_url_wikipedia(artist_song):
-    base_url = "https://www.wikipedia.org/search-redirect.php?family=wikipedia&language=en&search="
+
+def create_search_url(base_url, artist_song):
+    # Create search URL with appropriate character replacements
     return base_url + artist_song.replace(' ', '+').replace('–', '%E2%80%93').replace('&', 'and')
-def create_search_url_youtube(artist_song):
-    base_url = "https://www.youtube.com/results?search_query="
-    return base_url + artist_song.replace(' ', '+').replace('–', '%E2%80%93').replace('&', 'and')
-def create_search_url_chords(artist_song):
-    base_url = "https://www.ultimate-guitar.com/search.php?search_type=title&value="
-    return base_url + artist_song.replace(' ', '+').replace('–', '%E2%80%93').replace('&', 'and')
-def create_search_url_lyrics(artist_song):
-    base_url = "https://www.google.com/search?q="
-    return base_url + artist_song.replace(' ', '+').replace('–', '%E2%80%93').replace('&', 'and') + '+lyrics'
-def songupdate():
-    st.write(st.session_state.EnhancedChordSheet)
-def load_song_file2(filename):
-    with open(filename, "r") as file:
-        chord_sheet = file.read()
-        st.session_state['chord_sheet'] = chord_sheet
-        processed_sheet = process_chord_sheet(chord_sheet)
-        st.markdown(processed_sheet, unsafe_allow_html=True)
+
 def load_song_file(file_path):
+    # Load a song file and return its content
     with open(file_path, 'r', encoding='utf-8') as file:
-        chord_sheet = file.read()
-    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
+        return file.read()
+
 def parse_filename(filename):
+    # Parse the filename to extract song and artist names
     base_name = os.path.splitext(filename)[0]
     song_name, artist_name = base_name.split(' by ')
     return song_name.replace("_", " "), artist_name.replace("_", " ")
-def auto_save():
-    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 and chord_sheet:
-        filename = song_name + " by " + artist_name + ".txt"
-        with open(filename, "w") as file:
-            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():
     col1, col3 = st.columns([3, 5])
+
     with col1:
         st.markdown('### 🎵 📚Prompt🎥🎸Chord Sheet🎶 AI Prompt Authoring App')
+
+        # Expander for selecting the song
         with st.expander("Select Song:", expanded=True):
             all_files = [f for f in glob.glob("*.txt") if ' by ' in f]
-            selected_file = st.selectbox("Choose: ", all_files, on_change=song_update, key='selected_file')
-        song_name_input = st.text_input("🎵 Song:", key='song_name', on_change=auto_save)
-        artist_name_input = st.text_input("🎤 Artist:", 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)
-        st.header("🎼 Current Song")
-        load_song_file(selected_file)
-        song_info = os.path.splitext(selected_file)[0].replace("_", " ")
-        st.markdown("**" + song_info + "**")
-        table_md = f"""
-        | Wikipedia | YouTube | Chords | Lyrics |
-        | --------- | ------- | ------ | ------ |
-        | [📚]({create_search_url_wikipedia(song_info)}) | [🎥]({create_search_url_youtube(song_info)}) | [🎸]({create_search_url_chords(song_info)}) | [🎶]({create_search_url_lyrics(song_info)}) |
-        """
-        st.markdown(table_md)
-        st.header("🎼 Available Songs")
-        for file in all_files:
-            song_info = os.path.splitext(file)[0].replace("_", " ")
-            icol1, icol2 = st.columns([1, 3])
-            with icol1:
-                st.markdown("**" + song_info + "**")
-                load_song_file(file)
+            selected_file = st.selectbox("Choose: ", all_files, key='selected_file')
+
+        # Input fields for song and artist names
+        song_name_input = st.text_input("🎵 Song:", key='song_name')
+        artist_name_input = st.text_input("🎤 Artist:", key='artist_name')
+
+        if selected_file:
+            # Display the current song
+            song_info = os.path.splitext(selected_file)[0].replace("_", " ")
+            st.header("🎼 Current Song")
+            st.markdown("**" + song_info + "**")
+            processed_sheet = process_chord_sheet(load_song_file(selected_file))
+            components.html(processed_sheet, height=600)  # Using HTML component for chord sheet
+
+            # Markdown table with links
+            table_md = f"""
+            | Wikipedia | YouTube | Chords | Lyrics |
+            | --------- | ------- | ------ | ------ |
+            | [📚]({create_search_url("https://www.wikipedia.org/search-redirect.php?family=wikipedia&language=en&search=", song_info)}) | [🎥]({create_search_url("https://www.youtube.com/results?search_query=", song_info)}) | [🎸]({create_search_url("https://www.ultimate-guitar.com/search.php?search_type=title&value=", song_info)}) | [🎶]({create_search_url("https://www.google.com/search?q=", song_info + '+lyrics')}) |
+            """
+            st.markdown(table_md)
+
+            st.header("🎼 Available Songs")
+            for file in all_files:
                 song_info = os.path.splitext(file)[0].replace("_", " ")
-            with icol2:
-                # Create a markdown table with links for each song file
-                table_md = f"""
-                | Wikipedia | YouTube | Chords | Lyrics |
-                | --------- | ------- | ------ | ------ |
-                | [📚]({create_search_url_wikipedia(song_info)}) | [🎥]({create_search_url_youtube(song_info)}) | [🎸]({create_search_url_chords(song_info)}) | [🎶]({create_search_url_lyrics(song_info)}) |
-                """
-                st.markdown(table_md)
+                icol1, icol2 = st.columns([1, 3])
+                with icol1:
+                    st.markdown("**" + song_info + "**")
+                    processed_sheet = process_chord_sheet(load_song_file(file))
+                    components.html(processed_sheet, height=150)  # Displaying processed chord sheets
+
     with col3:
-        subcol1, subcol2 = st.columns([1, 5])
-        with subcol2:
-            chord_sheet_area = st.text_area("Chord Sheet", value=st.session_state.get('chord_sheet', ''), height=1600, key='chord_sheet', on_change=auto_save)
-        with subcol1:
-            # Save functionality
-            if st.button("💾 Save", key="save_song"):
-                if song_name_input and artist_name_input:
-                    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 to file: " + filename)
-                else:
-                    st.error("Both Song Name and Artist Name are required.")
-            char_count_msg = f"Character Count: {st.session_state.get('char_count', 0)}"
-            st.write(char_count_msg)
-    # 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)
+        # Chord sheet text area
+        chord_sheet_area = st.text_area("Chord Sheet", value=load_song_file(selected_file) if selected_file else '', height=1600, key='chord_sheet')
+
+        # Save functionality
+        if st.button("💾 Save", key="save_song"):
+            if song_name_input and artist_name_input:
+                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 to file: " + filename)
+            else:
+                st.error("Both Song Name and Artist Name are required.")
+
 if __name__ == '__main__':
-    main()
+    main()