Update app.py

app.py

@@ -4,33 +4,20 @@ from PIL import Image
 from io import BytesIO
 import os
 
-DEFAULT_CARD_WIDTH = 75
-DEFAULT_CARD_HEIGHT = 100
+st.set_page_config(page_title="Card Evolution Game", page_icon=":hearts:")
 
-PARTS = {"background": ["white", "black", "red", "blue", "green", "yellow"],
-         "suit": ["clubs", "diamonds", "hearts", "spades"],
-         "value": ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "J", "Q", "K"]}
+def get_default_card_size():
+    return 75, 100
+
+def get_default_parts():
+    return {"background": "white", "suit": "clubs", "value": "1"}
 
-LEVELS = {"level 1": ["common", "uncommon", "rare"],
-          "level 2": ["epic", "legendary", "mythic"],
-          "level 3": ["godlike"]}
+def get_default_levels():
+    return ["level 1", "common"]
 
 def draw_card(background, suit, value, level, card_width, card_height):
-    print(f"Background image: {background}")
     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))
-
-    # Add background image, if provided
-    if background and os.path.exists(background):
-        img = Image.open(background)
-        print("Image opened successfully")
-        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()
-        img = svgwrite.image.Image(href=img_href, size=(f"{card_width}px", f"{card_height}px"))
-        dwg.add(img)
+    dwg.add(dwg.rect((0, 0), (card_width, card_height), rx=10, ry=10, fill=background, stroke="black", stroke_width=2))
 
     # Add suit and value labels
     suit = svgwrite.text.Text(suit.upper(), insert=(5, 15), fill="black", font_size="16px", font_weight="bold")
@@ -43,7 +30,6 @@ def draw_card(background, suit, value, level, card_width, card_height):
     svg_string = dwg.tostring()
     return svg_string
 
-
 def display_parts_selection():
     selected_parts = {}
     for part, options in PARTS.items():
@@ -51,7 +37,7 @@ def display_parts_selection():
     return selected_parts
 
 def display_level_selection():
-    selected_level = st.sidebar.selectbox("Select Level", list(LEVELS.keys()))
+    selected_level = st.sidebar.selectbox("Select Level", LEVELS)
     return selected_level
 
 def display_size_selection(default_width, default_height):
@@ -77,8 +63,13 @@ def display_card(selected_parts, selected_level, card_width, card_height, backgr
         img.save(img_bytes, format="PNG")
         img_data = img_bytes.getvalue()
         img_href = "data:image/png;base64," + img_data.hex()
+        background_name = os.path.basename(background)
+        st.sidebar.markdown(f"**Background Image:** {background_name}")
         background = img_href
-    card_svg = draw_card(background, selected_parts["suit"], selected_parts["value"], selected_level, card_width, card_height)
+    else:
+        st.sidebar.markdown(f"**Background Image:** {selected_parts['background'].capitalize()}")
+
+    card_svg = draw_card(selected_parts["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)
 
     # Create a download link for the card SVG
@@ -86,16 +77,25 @@ def display_card(selected_parts, selected_level, card_width, card_height, backgr
     st.download_button("Download Card as SVG", svg_content, "card.svg", "text/plain")
 
     # Create button to level up card
-    if selected_level != "level 3":
+    if selected_level != LEVELS[-1]:
         if st.button("Level Up"):
-            current_level_index = list(LEVELS.keys()).index(selected_level)
-            new_level = list(LEVELS.keys())[current_level_index + 1]
+            current_level_index = LEVELS.index(selected_level)
+            new_level = LEVELS[current_level_index + 1]
             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:")
 
+DEFAULT_CARD_WIDTH, DEFAULT_CARD_HEIGHT = get_default_card_size()
+DEFAULT_PARTS = get_default_parts()
+DEFAULT_LEVELS = get_default_levels()
+
+PARTS = {"background": ["white", "black", "red", "blue", "green", "yellow"],
+         "suit": ["clubs", "diamonds", "hearts", "spades"],
+         "value": ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "J", "Q", "K"]}
+
+LEVELS = ["level 1", "common", "uncommon", "rare", "epic", "legendary", "mythic", "godlike"]
+
 selected_parts = display_parts_selection()
 selected_level = display_level_selection()
 card_width, card_height = display_size_selection(DEFAULT_CARD_WIDTH, DEFAULT_CARD_HEIGHT)