Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import os
 import random
 from generate_jokes import generate_inside_jokes, generate_motivation
+from generate_quote_card import create_quote_card
 
 # Page Config
 st.set_page_config(
@@ -69,7 +70,7 @@ with col1:
     st.header("📸 Past Gallery")
     
     # Load images from 'sarah' directory
-    image_dir = "src/sarah"
+    image_dir = "sarah"
     if os.path.exists(image_dir):
         images = [f for f in os.listdir(image_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif'))]
         
@@ -97,6 +98,16 @@ with col1:
         with st.spinner("Finding the right words..."):
             motivation = generate_motivation()
             st.markdown(f'<div class="motivation-card">{motivation}</div>', unsafe_allow_html=True)
+            
+            # Download button for motivation
+            card_buffer = create_quote_card(motivation, card_type="motivation", title="Daily Motivation")
+            st.download_button(
+                label="📥 Download Card",
+                data=card_buffer,
+                file_name="motivation_card.png",
+                mime="image/png",
+                key="download_motivation"
+            )
 
 with col2:
     st.header("✨ Memory jokes fusion")
@@ -117,13 +128,36 @@ with col2:
 
                 # Display Jokes
                 st.markdown("### 🎭 Inside Jokes")
-                for line in jokes_section.split('\n'):
-                    if line.strip():
-                        st.markdown(f'<div class="joke-card">{line.strip()}</div>', unsafe_allow_html=True)
+                jokes_list = [line.strip() for line in jokes_section.split('\n') if line.strip()]
+                
+                for idx, joke in enumerate(jokes_list):
+                    col_joke, col_download = st.columns([4, 1])
+                    with col_joke:
+                        st.markdown(f'<div class="joke-card">{joke}</div>', unsafe_allow_html=True)
+                    with col_download:
+                        card_buffer = create_quote_card(joke, card_type="joke", title="Inside Joke")
+                        st.download_button(
+                            label="📥",
+                            data=card_buffer,
+                            file_name=f"joke_card_{idx+1}.png",
+                            mime="image/png",
+                            key=f"download_joke_{idx}"
+                        )
 
                 # Display Memory
                 st.markdown("### 💌 A Memory to Cherish")
-                st.markdown(f'<div class="memory-card">"{memory_section}"</div>', unsafe_allow_html=True)
+                col_mem, col_mem_download = st.columns([4, 1])
+                with col_mem:
+                    st.markdown(f'<div class="memory-card">"{memory_section}"</div>', unsafe_allow_html=True)
+                with col_mem_download:
+                    mem_card_buffer = create_quote_card(memory_section, card_type="memory", title="A Memory")
+                    st.download_button(
+                        label="📥",
+                        data=mem_card_buffer,
+                        file_name="memory_card.png",
+                        mime="image/png",
+                        key="download_memory"
+                    )
                 
             except Exception as e:
                 st.error(f"Oof! Something went wrong: {e}")