Update app.py

app.py

@@ -1,54 +1,27 @@
-import base64
-import graphviz as gv
 import streamlit as st
+import pandas as pd
+from io import StringIO
+from PIL import Image
+import graphviz as gv
 
-# 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')
+uploaded_files = st.file_uploader("Choose an image file", accept_multiple_files=True, type=["jpg", "jpeg", "png"])
+images = []
 
-    # Define the file uploader
-    uploaded_file = st.file_uploader('Upload an image', type=['jpg', 'jpeg', 'png'])
+for uploaded_file in uploaded_files:
+    # To read file as bytes:
+    bytes_data = uploaded_file.getvalue()
 
-    # Define the card title and subtitle inputs
-    title = st.text_input('Title')
-    subtitle = st.text_input('Subtitle')
+    # To convert to a PIL image:
+    image = Image.open(uploaded_file)
+    images.append(image)
 
-    # Render the card if an image has been uploaded and a title has been provided
-    if uploaded_file and title:
-        image_path = f'{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)
+# Create the Graphviz model using the uploaded images
+dot = gv.Digraph()
+for i, image in enumerate(images):
+    dot.node(str(i), image=image)
 
-        # 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)
+for i in range(len(images) - 1):
+    dot.edge(str(i), str(i+1))
 
-# Run the Streamlit app
-if __name__ == '__main__':
-    app()
+# Render the Graphviz model using the Streamlit framework
+st.graphviz_chart(dot.source)