Update app.py

app.py

@@ -1,60 +1,54 @@
 import streamlit as st
 import svgwrite
+from PIL import Image
 
-# Define the size of the cards
-CARD_WIDTH = 75
-CARD_HEIGHT = 100
+DEFAULT_CARD_WIDTH = 75
+DEFAULT_CARD_HEIGHT = 100
 
-# Define the size of the SVG canvas
-CANVAS_WIDTH = CARD_WIDTH * 5
-CANVAS_HEIGHT = CARD_HEIGHT
+PARTS = {"background": ["white", "black", "red", "blue", "green", "yellow"],
+         "suit": ["clubs", "diamonds", "hearts", "spades"],
+         "value": ["A", "2", "3", "4", "5", "6", "7", "8", "9", "10", "J", "Q", "K"]}
 
-# Define the parts that can be added to the card
-PARTS = {
-    "background": ["white", "black", "red", "blue", "green", "yellow"],
-    "suit": ["clubs", "diamonds", "hearts", "spades"],
-    "value": ["A", "2", "3", "4", "5", "6", "7", "8", "9", "10", "J", "Q", "K"],
-}
-
-# Function to draw the card
-def draw_card(background_color, suit, value):
-    # Create a new SVG drawing
-    dwg = svgwrite.Drawing(size=(f"{CARD_WIDTH}px", f"{CARD_HEIGHT}px"))
-
-    # Draw the card border
-    dwg.add(dwg.rect((0, 0), (CARD_WIDTH, CARD_HEIGHT), rx=10, ry=10, fill=background_color, stroke="black", stroke_width=2))
-
-    # Draw the card suit symbol
+def draw_card(background, suit, value, card_width, card_height):
+    dwg = svgwrite.Drawing(size=(f"{card_width}px", f"{card_height}px"))
+    dwg.add(dwg.rect((0, 0), (card_width, card_height), rx=10, ry=10, fill="white", stroke="black", stroke_width=2))
+    if background:
+        img = svgwrite.image.Image(href=background, size=(f"{card_width}px", f"{card_height}px"))
+        dwg.add(img)
     suit = svgwrite.text.Text(suit.upper(), insert=(5, 15), fill="black", font_size="16px", font_weight="bold")
     dwg.add(suit)
-
-    # Draw the card value
-    value = svgwrite.text.Text(value, insert=(5, CARD_HEIGHT - 10), fill="black", font_size="16px", font_weight="bold")
+    value = svgwrite.text.Text(value, insert=(5, card_height - 10), fill="black", font_size="16px", font_weight="bold")
     dwg.add(value)
-
-    # Convert the SVG drawing to a string
     svg_string = dwg.tostring()
-
     return svg_string
 
-# Function to display the parts selection sidebar
 def display_parts_selection():
     selected_parts = {}
     for part, options in PARTS.items():
-        selected_option = st.sidebar.selectbox(f"Select {part}", options)
-        selected_parts[part] = selected_option
+        selected_parts[part] = st.sidebar.selectbox(f"Select {part}", options)
     return selected_parts
 
-# Function to display the resulting card
-def display_card(selected_parts):
-    card_svg = draw_card(selected_parts["background"], selected_parts["suit"], selected_parts["value"])
-    st.write(f'<svg viewBox="0 0 {CARD_WIDTH} {CARD_HEIGHT}">{card_svg}</svg>', unsafe_allow_html=True)
+def display_size_selection(default_width, default_height):
+    card_width = st.sidebar.slider("Card Width", min_value=50, max_value=150, value=default_width, step=5)
+    card_height = st.sidebar.slider("Card Height", min_value=75, max_value=200, value=default_height, step=5)
+    return card_width, card_height
+
+def display_image_upload():
+    uploaded_file = st.sidebar.file_uploader("Upload Card Background Image", type=["png", "jpg", "jpeg"])
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        image.save("background.png")
+        return "background.png"
+    else:
+        return None
+
+def display_card(selected_parts, card_width, card_height, background):
+    card_svg = draw_card(background, selected_parts["suit"], selected_parts["value"], card_width, card_height)
+    st.write(f'<svg viewBox="0 0 {card_width} {card_height}">{card_svg}</svg>', unsafe_allow_html=True)
 
-# Set the page title and icon
 st.set_page_config(page_title="Card Crafting Game", page_icon=":spades:")
 
-# Display the parts selection sidebar
 selected_parts = display_parts_selection()
-
-# Display the resulting card
-display_card(selected_parts)
+card_width, card_height = display_size_selection(DEFAULT_CARD_WIDTH, DEFAULT_CARD_HEIGHT)
+background = display_image_upload()
+display_card(selected_parts, card_width, card_height, background)