Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import svgwrite
 from PIL import Image
-import urllib.request
 from io import BytesIO
+import os
 
 DEFAULT_CARD_WIDTH = 75
 DEFAULT_CARD_HEIGHT = 100
@@ -21,7 +21,7 @@ def draw_card(background, suit, value, level, card_width, card_height):
     dwg.add(dwg.rect((0, 0), (card_width, card_height), rx=10, ry=10, fill="white", stroke="black", stroke_width=2))
 
     # Add background image, if provided
-    if background:
+    if background and os.path.exists(background):
         img = Image.open(background)
         print("Image opened successfully")
         img = img.resize((card_width, card_height))
@@ -62,12 +62,22 @@ def display_size_selection(default_width, default_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(BytesIO(uploaded_file.read()))
-        return image
+        image_name = uploaded_file.name
+        with open(image_name, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        return image_name
     else:
         return None
         
 def display_card(selected_parts, selected_level, card_width, card_height, background):
+    if background and os.path.exists(background):
+        img = Image.open(background)
+        img = img.resize((card_width, card_height))
+        img_bytes = BytesIO()
+        img.save(img_bytes, format="PNG")
+        img_data = img_bytes.getvalue()
+        img_href = "data:image/png;base64," + img_data.hex()
+        background = img_href
     card_svg = draw_card(background, selected_parts["suit"], selected_parts["value"], selected_level, card_width, card_height)
     st.write(f'<svg viewBox="0 0 {card_width} {card_height}">{card_svg}</svg>', unsafe_allow_html=True)
 
@@ -83,14 +93,11 @@ def display_card(selected_parts, selected_level, card_width, card_height, backgr
             selected_level = new_level
             st.sidebar.warning(f"Card has been leveled up to {selected_level}")
 
+
 st.set_page_config(page_title="Card Evolution Game", page_icon=":hearts:")
 
 selected_parts = display_parts_selection()
 selected_level = display_level_selection()
 card_width, card_height = display_size_selection(DEFAULT_CARD_WIDTH, DEFAULT_CARD_HEIGHT)
 background = display_image_upload()
-
-if background is not None:
-    background = background.resize((card_width, card_height))
-
 display_card(selected_parts, selected_level, card_width, card_height, background)