Update app.py

app.py

@@ -3,59 +3,28 @@ import streamlit.components.v1 as components
 import re
 import os
 import glob
+import base64
 
-# Set Streamlit page configuration
 st.set_page_config(layout="wide")
 
-import base64
-
 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):
-    # Define a dictionary for chord-to-image mapping
     chord_images = {
         'Em': 'Em.png',
         'D': 'D.png',
         'C': 'C.png'
     }
-
-    # Function to replace chord with base64 image
     def replace_chord(match):
         chord = match.group(0)
         image_base64 = get_image_base64(chord_images.get(chord, 'default.png'))
-        return f"<img src='data:image/png;base64,{image_base64}' style='height:20px;'>"
-
-    # Use regular expression to find and replace chords
-    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_line_old2(line):
-    # Define a dictionary for chord-to-image mapping
-    chord_images = {
-        'Em': 'Em.png',
-        'D': 'D.png',
-        'C': 'C.png'
-    }
-
-    # Function to replace chord with image
-    def replace_chord(match):
-        chord = match.group(0)
-        return f"<img src='{chord_images.get(chord, 'default.png')}' style='height:20px;'>"
-
-    # Use regular expression to find and replace chords
+        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_line_old(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='\Em.png' style='height:20px;'>", 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)
@@ -81,7 +50,6 @@ def create_search_url_lyrics(artist_song):
     return base_url + artist_song.replace(' ', '+').replace('–', '%E2%80%93').replace('&', 'and') + '+lyrics'
 
 def display_chord_sheet_in_two_page_view(chord_sheet):
-    # CSS for multi-column layout
     css_content = """
     <style>
     .multi-column {
@@ -94,8 +62,6 @@ def display_chord_sheet_in_two_page_view(chord_sheet):
     }
     </style>
     """
-
-    # HTML structure with multi-column layout
     html_content = f"""
     {css_content}
     <div class="multi-column">
@@ -106,7 +72,6 @@ def display_chord_sheet_in_two_page_view(chord_sheet):
 
 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):
@@ -125,7 +90,6 @@ def main():
                 | [📚]({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("_", " ")
@@ -139,13 +103,11 @@ def main():
                 | [📚]({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:
         if 'selected_file' in st.session_state and st.session_state.selected_file:
-            # Display chord sheet in multi-column view
             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()
+    main()