import streamlit as st import streamlit.components.v1 as components import re import os import glob st.set_page_config(layout="wide") def process_line(line): if re.search(r'\b[A-G][#b]?m?\b', line): line = re.sub(r'\b([A-G][#b]?m?)\b', r"", 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) return '
'.join(processed_lines) def create_search_url_wikipedia(artist_song): base_url = "https://www.wikipedia.org/search-redirect.php?family=wikipedia&language=en&search=" 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): 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 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("_", " ") 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') 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) 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) 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) if __name__ == '__main__': main()