Spaces:
Running
Running
| import streamlit as st | |
| import streamlit.components.v1 as components | |
| import re | |
| import os | |
| import glob | |
| import base64 | |
| st.set_page_config(layout="wide") | |
| def get_image_base64(image_path): | |
| with open(image_path, "rb") as image_file: | |
| return base64.b64encode(image_file.read()).decode() | |
| def process_line(line): | |
| chord_images = { | |
| 'Em': 'Em.png', | |
| 'D': 'D.png', | |
| 'C': 'C.png' | |
| } | |
| def replace_chord(match): | |
| chord = match.group(0) | |
| image_base64 = get_image_base64(chord_images.get(chord, 'default.png')) | |
| return f"<strong>{chord}</strong><img src='data:image/png;base64,{image_base64}' style='height:20px;'>" | |
| pattern = r'\b(' + '|'.join(re.escape(chord) for chord in chord_images.keys()) + r')\b' | |
| line = re.sub(pattern, replace_chord, line) | |
| return line | |
| def process_chord_sheet(chord_sheet): | |
| processed_lines = [process_line(line) for line in chord_sheet.split('\n')] | |
| return '<br>'.join(processed_lines) | |
| def load_song_file(file_path): | |
| with open(file_path, 'r', encoding='utf-8') as file: | |
| return file.read() | |
| 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 display_chord_sheet_in_two_page_view(chord_sheet): | |
| css_content = """ | |
| <style> | |
| .multi-column { | |
| column-count: 2; | |
| column-gap: 1em; | |
| column-rule: thin solid black; | |
| overflow: auto; | |
| white-space: pre-wrap; | |
| font-size: small; | |
| } | |
| </style> | |
| """ | |
| html_content = f""" | |
| {css_content} | |
| <div class="multi-column"> | |
| {chord_sheet} | |
| </div> | |
| """ | |
| components.html(html_content, height=1200) | |
| def main(): | |
| col1, col3 = st.columns([3, 5]) | |
| with col1: | |
| st.markdown('### 🎵🎥 ChordPrompter 🎸 AI Prompt Authoring - Wiki, YouTube, Chords, Lyrics📚🎶') | |
| 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, key='selected_file') | |
| if selected_file: | |
| song_info = os.path.splitext(selected_file)[0].replace("_", " ") | |
| st.header("🎼 Current Song") | |
| st.markdown("**" + song_info + "**") | |
| chord_sheet = load_song_file(selected_file) | |
| processed_sheet = process_chord_sheet(chord_sheet) | |
| #st.markdown(processed_sheet, unsafe_allow_html=True) | |
| table_md = f""" | |
| | [📚Wikipedia]({create_search_url_wikipedia(song_info)}) | [🎥YouTube]({create_search_url_youtube(song_info)}) | |
| | [🎸Chords]({create_search_url_chords(song_info)}) | [🎶Lyrics]({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, 2]) | |
| with icol1: | |
| st.markdown("**" + song_info + "**") | |
| with icol2: | |
| table_md = f""" | |
| | [📚Wikipedia]({create_search_url_wikipedia(song_info)}) | [🎥YouTube]({create_search_url_youtube(song_info)}) | |
| | [🎸Chords]({create_search_url_chords(song_info)}) | [🎶Lyrics]({create_search_url_lyrics(song_info)}) | | |
| """ | |
| st.markdown(table_md) | |
| with col3: | |
| if 'selected_file' in st.session_state and st.session_state.selected_file: | |
| chord_sheet = load_song_file(st.session_state.selected_file) | |
| processed_sheet = process_chord_sheet(chord_sheet) | |
| display_chord_sheet_in_two_page_view(processed_sheet) | |
| if __name__ == '__main__': | |
| main() |