Update app.py

app.py

@@ -1,32 +1,54 @@
+import base64
+import graphviz as gv
 import streamlit as st
-import svgwrite
-
-def apply_circle_mask(image_path):
-    # Load image
-    image = svgwrite.Drawing('output.svg')
-    image.add(image.image(href=image_path, size=("100%", "100%")))
-    
-    # Add circular mask
-    viewbox = image.viewbox()
-    width, height = viewbox[2], viewbox[3]
-    half_x, half_y = width/2, height/2
-    line_color = "white"
-    mask = image.defs.add(image.mask(id="bg_wrapper"))
-    mask.add(image.circle(center=(half_x, half_y), r=335, fill=line_color, opacity=".4"))
-    image.add(image.image(href=image_path, size=("100%", "100%"), mask="url(#bg_wrapper)"))
-
-    return image.tostring()
-
-# Streamlit app
-st.title("Circular Image Mask")
-
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-
-if uploaded_file is not None:
-    # Save uploaded image to file
-    with open("image.jpg", "wb") as f:
-        f.write(uploaded_file.getbuffer())
-    
-    # Apply circular mask and display result
-    output_svg = apply_circle_mask("image.jpg")
-    st.image(output_svg, output_format="SVG")
+
+# Define a function that renders the card using Graphviz
+def render_card(image_path, title, subtitle):
+    # Define the Graphviz graph
+    g = gv.Graph(format='png')
+    g.attr(bgcolor='white', margin='0')
+
+    # Add the background image to the graph
+    with open(image_path, 'rb') as f:
+        data = f.read()
+    b64_image = base64.b64encode(data).decode()
+    g.node('background', image=f'data:image/png;base64,{b64_image}', shape='box', width='3.5', height='5')
+
+    # Add the title and subtitle to the graph
+    g.node('title', label=title, fontname='Arial', fontsize='16', shape='none', width='2')
+    g.node('subtitle', label=subtitle, fontname='Arial', fontsize='12', shape='none', width='2')
+
+    # Add edges between the nodes
+    g.edge('background', 'title', color='white', style='invis')
+    g.edge('background', 'subtitle', color='white', style='invis')
+
+    # Render the graph and return the PNG data
+    return g.pipe().decode('utf-8')
+
+# Define the Streamlit app
+def app():
+    st.title('Card Builder Evolver')
+
+    # Define the file uploader
+    uploaded_file = st.file_uploader('Upload an image', type=['jpg', 'jpeg', 'png'])
+
+    # Define the card title and subtitle inputs
+    title = st.text_input('Title')
+    subtitle = st.text_input('Subtitle')
+
+    # Render the card if an image has been uploaded and a title has been provided
+    if uploaded_file and title:
+        image_path = f'tmp/{uploaded_file.name}'
+        with open(image_path, 'wb') as f:
+            f.write(uploaded_file.getbuffer())
+        card = render_card(image_path, title, subtitle)
+        st.image(card, use_column_width=True)
+
+        # Define the download button
+        b64_card = base64.b64encode(card.encode()).decode()
+        href = f'<a href="data:image/png;base64,{b64_card}" download="card.png">Download card</a>'
+        st.markdown(href, unsafe_allow_html=True)
+
+# Run the Streamlit app
+if __name__ == '__main__':
+    app()