Update app.py

app.py

@@ -5,8 +5,14 @@ import datetime
 import os
 import time
 import base64
+import re
+import glob
 from camera_input_live import camera_input_live
 
+# Set wide layout
+st.set_page_config(layout="wide")
+
+# Function Definitions for Camera Feature
 def save_image(image):
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     filename = f"captured_image_{timestamp}.png"
@@ -19,43 +25,67 @@ def get_image_base64(image_path):
     with open(image_path, "rb") as image_file:
         return base64.b64encode(image_file.read()).decode()
 
+# Function Definitions for Chord Sheet Feature
+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)
+
+# Main Function
 def main():
-    st.title("Streamlit Camera Real Time Live Streming")
-    st.header("QR code reader is ready.")
-
-    # Slider for snapshot interval
-    snapshot_interval = st.sidebar.slider("Snapshot Interval (seconds)", 1, 10, 5)
-
-    # Display container for live image
-    image_placeholder = st.empty()
-
-    # Initialize session state for captured images and last captured time
-    if 'captured_images' not in st.session_state:
-        st.session_state['captured_images'] = []
-    if 'last_captured' not in st.session_state:
-        st.session_state['last_captured'] = time.time()
-
-    # Capture and display live image
-    image = camera_input_live()
-    if image is not None:
-        image_placeholder.image(image)
-
-        # Save image based on interval
-        if time.time() - st.session_state['last_captured'] > snapshot_interval:
-            filename = save_image(image)
-            st.session_state['captured_images'].append(filename)
+    # Layout Configuration
+    col1, col2, col3 = st.columns([2, 3, 5])
+
+    # Camera Section
+    with col1:
+        st.title("Real-Time Camera Stream")
+        st.header("QR code reader ready.")
+
+        snapshot_interval = st.slider("Snapshot Interval (seconds)", 1, 10, 5)
+        image_placeholder = st.empty()
+
+        if 'captured_images' not in st.session_state:
+            st.session_state['captured_images'] = []
+        if 'last_captured' not in st.session_state:
             st.session_state['last_captured'] = time.time()
 
-    # Update sidebar with captured images
-    sidebar_html = "<div style='display:flex;flex-direction:column;'>"
-    for img_file in st.session_state['captured_images']:
-        image_base64 = get_image_base64(img_file)
-        sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
-    sidebar_html += "</div>"
-    st.sidebar.markdown("## Captured Images")
-    st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
+        image = camera_input_live()
+        if image is not None:
+            image_placeholder.image(image)
+
+            if time.time() - st.session_state['last_captured'] > snapshot_interval:
+                filename = save_image(image)
+                st.session_state['captured_images'].append(filename)
+                st.session_state['last_captured'] = time.time()
+
+        sidebar_html = "<div style='display:flex;flex-direction:column;'>"
+        for img_file in st.session_state['captured_images']:
+            image_base64 = get_image_base64(img_file)
+            sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
+        sidebar_html += "</div>"
+        st.sidebar.markdown("## Captured Images")
+        st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
+
+    # Chord Sheet Section
+    with col2:
+        st.title("Chord Sheet Manager")
+
+        all_files = [f for f in glob.glob("*.txt") if ' by ' in f]
+        selected_file = st.selectbox("Choose a Song:", all_files)
+
+        if selected_file:
+            with open(selected_file, 'r', encoding='utf-8') as file:
+                chord_sheet = file.read()
+            st.markdown(process_chord_sheet(chord_sheet), unsafe_allow_html=True)
 
-    # Trigger a rerun of the app to update the live feed
+    # Trigger a rerun to update the live feed
     st.experimental_rerun()
 
 if __name__ == "__main__":