Spaces:
Sleeping
Sleeping
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"<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) | |
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=" | |
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() |