Can upload user original card design.

app.py

@@ -7,6 +7,7 @@ import glob
 
 import src.constants as constants
 
+from src.front_card_frame import create_card_frame
 from src.front_card_image import create_card_image
 from src.front_card_image_historic_site import create_historic_site_card_image
 from src.picture_box_model import create_picture_box_model
@@ -29,11 +30,11 @@ def update_card_image(card_design_upload, card_design_dropdown, update_type):
     else:
         return default_card_design_image_dict[card_design_dropdown]
 
-def create_3dmodel(model_no, title, color, mark, historic_site_type, difficulty, description, is_thick, image):
+def create_3dmodel(model_no, title, color, mark, historic_site_type, difficulty, description, is_thick, picture_image, card_design_preview_image):
 
-    img_bytearray = BytesIO()
-    image['background'].save(img_bytearray, "JPEG", quality=95)
-    img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data.
+    picture_img_bytearray = BytesIO()
+    picture_image['background'].save(picture_img_bytearray, "JPEG", quality=95)
+    picture_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data.
 
     option_dict = {
         'タイトル': title,
@@ -50,16 +51,21 @@ def create_3dmodel(model_no, title, color, mark, historic_site_type, difficulty,
     if model_no not in ['A', 'B']:
         # Create the card image (front side)
         if model_no == '1':
-            front_img_bytearray = create_historic_site_card_image(img_bytearray, option_dict)
-        else:
-            front_img_bytearray = create_card_image(model_no, img_bytearray, option_dict)
-
-        # Retrieve the card image (back side)
-        back_path = constants.back_card_img_dict[model_no]
-        back_img = PIL_Image.open(back_path)
-        back_img_bytearray = BytesIO()
-        back_img.convert('RGB').save(back_img_bytearray, "JPEG", quality=95)
-        back_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data
+            front_img_bytearray = create_historic_site_card_image(picture_img_bytearray, option_dict)
+            
+            # Retrieve the card image (back side)
+            back_path = constants.back_card_img_dict[model_no]
+            back_img = PIL_Image.open(back_path)
+            back_img_bytearray = BytesIO()
+            back_img.convert('RGB').save(back_img_bytearray, "JPEG", quality=95)
+            back_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data
+
+        else:            
+            back_img_bytearray = BytesIO()
+            card_design_preview_image.save(back_img_bytearray, "JPEG", quality=95)
+            back_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data.    
+            card_frame_bytearray = create_card_frame(back_img_bytearray)
+            front_img_bytearray = create_card_image(card_frame_bytearray, picture_img_bytearray, option_dict)
 
         # Create a 3D model of the card (return value is the path of the created model)
         model_path = create_card_model(front_img_bytearray, back_img_bytearray, option_dict)
@@ -67,9 +73,9 @@ def create_3dmodel(model_no, title, color, mark, historic_site_type, difficulty,
     else:
         # Create a 3D model of the card (return value is the path of the created model)
         if model_no == 'A':
-            model_path = create_picture_box_model(img_bytearray)
+            model_path = create_picture_box_model(picture_img_bytearray)
         if model_no == 'B':
-            model_path = create_extracted_objects_model(img_bytearray)
+            model_path = create_extracted_objects_model(picture_img_bytearray)
     
     return model_path
 
@@ -90,29 +96,28 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 title = gr.Textbox(label=display_message_dict['label_title'], placeholder=display_message_dict['placeholder_title'])
-                model_type = gr.Radio([(model_type_dict[key], key) for key in model_type_dict], value='2', label=display_message_dict['label_card_type'])
                 
                 with gr.Row():
                     with gr.Column():
-                            card_design_dropdown = gr.Dropdown(list(default_card_design_image_dict.keys()), label=display_message_dict['label_card_design_sample'])
-                            card_design_upload = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="868:1213", label=display_message_dict['label_card_design_upload'])                                                            
+                            card_design_dropdown = gr.Dropdown(list(default_card_design_image_dict.keys()), label=display_message_dict['label_card_design_sample'])                           
+                            card_design_upload = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="868:1213", label=display_message_dict['label_card_design_upload'])
                     with gr.Column():                        
-                        card_design_preview = gr.Image(label=display_message_dict['label_card_design_preview'], type="pil")
+                        card_design_preview_image = gr.Image(label=display_message_dict['label_card_design_preview'], type="pil")
                     card_design_upload.change(fn=lambda card_design_upload, card_design_dropdown: update_card_image(card_design_upload, card_design_dropdown, 'Upload'), 
-                                              inputs=[card_design_upload, card_design_dropdown], outputs=card_design_preview)
+                                              inputs=[card_design_upload, card_design_dropdown], outputs=card_design_preview_image)
                     card_design_dropdown.change(fn=lambda card_design_upload, card_design_dropdown: update_card_image(card_design_upload, card_design_dropdown, 'Drop'), 
-                                                inputs=[card_design_upload, card_design_dropdown], outputs=card_design_preview)
+                                                inputs=[card_design_upload, card_design_dropdown], outputs=card_design_preview_image)
 
             with gr.Column():
                 description = gr.Textbox(lines=2, label=display_message_dict['label_description'], placeholder=display_message_dict['placeholder_description'])
                 is_thick = gr.Checkbox(label=display_message_dict['label_is_thick'], value=False, info=display_message_dict['info_is_thick'])
-                image = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="1:1", label=display_message_dict['label_image'])
+                picture_image = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="1:1", label=display_message_dict['label_image'])
 
         button = gr.Button(display_message_dict['label_button'])
         button.click(
-                fn=lambda model_type, title, description, is_thick, image:
-                    create_3dmodel(model_type, title, None, None, None, None, description, is_thick, image),
-            inputs=[model_type, title, description, is_thick, image],
+                fn=lambda  title, description, is_thick, picture_image, card_design_preview_image:
+                    create_3dmodel(None, title, None, None, None, None, description, is_thick, picture_image, card_design_preview_image),
+            inputs=[title, description, is_thick, picture_image, card_design_preview_image],
             outputs=[gr.Model3D(camera_position=(90, 90, 5))]
         )
     
@@ -133,13 +138,13 @@ with gr.Blocks() as demo:
                 difficulty = gr.Slider(1, 5, 3, step=1, label=display_message_dict['label_difficulty'])
                 description = gr.Textbox(lines=2, label=display_message_dict['label_description'], placeholder=display_message_dict['placeholder_description'])
                 is_thick = gr.Checkbox(label=display_message_dict['label_is_thick'], value=False, info=display_message_dict['info_is_thick'])
-                image = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="1:1", label=display_message_dict['label_image'])
+                picture_image = gr.ImageEditor(image_mode='RGB', sources="upload", type="pil", crop_size="1:1", label=display_message_dict['label_image'])
 
         button = gr.Button(display_message_dict['label_button'])
         button.click(
-                fn=lambda title, color, mark, historic_site_type, difficulty, description, is_thick, image:
-                    create_3dmodel('1', title, color, mark, historic_site_type, difficulty, description, is_thick, image),
-            inputs=[title, color, mark, historic_site_type, difficulty, description, is_thick, image],
+                fn=lambda title, color, mark, historic_site_type, difficulty, description, is_thick, picture_image:
+                    create_3dmodel('1', title, color, mark, historic_site_type, difficulty, description, is_thick, picture_image, None),
+            inputs=[title, color, mark, historic_site_type, difficulty, description, is_thick, picture_image],
             outputs=[gr.Model3D(camera_position=(90, 90, 5))]
         )
 

src/front_card_frame.py

@@ -0,0 +1,65 @@
+from PIL import Image, ImageDraw
+from io import BytesIO
+
+# Pixel position information for each area of the card image
+
+# Card
+CARD_SIZE = (868, 1213)
+
+# Picture
+PICTURE_WHITE_LT_XY = (65, 188)
+PICTURE_WHITE_RB_XY = (802, 925)
+
+# Title
+TITLE_BLEND_WHITE_LT_XY = (34, 29) 
+TITLE_BLEND_WHITE_RB_XY = (833, 149)
+
+# Main Body of Description
+DESCRIPTION_BLEND_WHITELT_XY = (34, 961)
+DESCRIPTION_BLEND_WHITERB_XY = (833, 1185)
+
+def average_with_white(img, area):
+
+    # Change alpha value
+    x1, y1, x2, y2 = area
+    for x in range(x1, x2):
+        for y in range(y1, y2):
+            r, g, b = img.getpixel((x, y))
+            img.putpixel((x, y), ((r + 255) // 2, (g + 255) // 2, (b + 255) // 2))
+
+    return img
+
+def create_card_frame(img_bytearray):
+    card_img = Image.open(img_bytearray).convert('RGB')
+    card_img = card_img.resize(CARD_SIZE)
+    card_imgdraw = ImageDraw.Draw(card_img)
+
+    # Paint White to the Picture Area
+    card_img.paste((255, 255, 255), PICTURE_WHITE_LT_XY + PICTURE_WHITE_RB_XY)
+    
+    # Make translucent to the Title Area
+    card_img = average_with_white(card_img, TITLE_BLEND_WHITE_LT_XY + TITLE_BLEND_WHITE_RB_XY)
+    # Draw black rectangle frame to the Title Area
+    card_imgdraw.rectangle(TITLE_BLEND_WHITE_LT_XY + TITLE_BLEND_WHITE_RB_XY, outline=(0, 0, 0), width=5)
+    
+    # Make translucent to the Description Area
+    card_img = average_with_white(card_img, DESCRIPTION_BLEND_WHITELT_XY + DESCRIPTION_BLEND_WHITERB_XY)
+    # Draw rectangle frame to the Description Area
+    card_imgdraw.rectangle(DESCRIPTION_BLEND_WHITELT_XY + DESCRIPTION_BLEND_WHITERB_XY, outline=(0, 0, 0), width=5)
+    
+    # Outputting Binary Data
+    output_img_bytearray = BytesIO()
+    card_img.save(output_img_bytearray, "PNG", quality=95)
+    output_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data
+
+    return output_img_bytearray
+
+if __name__ == '__main__':
+    with open('data/cards/back/cats.png', 'rb') as f:
+        card_frame_bytearray = BytesIO(f.read())
+    
+    card_frame_bytearray = create_card_frame(card_frame_bytearray)
+    
+    with open('test.jpeg', 'wb') as f:
+        f.write(card_frame_bytearray.read())
+        

src/front_card_image.py

@@ -36,7 +36,7 @@ def crop_center(pil_img, crop_width, crop_height):
 def crop_max_square(pil_img):
     return crop_center(pil_img, min(pil_img.size), min(pil_img.size))
     
-def create_card_image(model_no, img_bytearray, option_dict):
+def create_card_image(card_frame_bytearray, img_bytearray, option_dict):
 
     # Loading and Cropping of Picture
     picture_img = Image.open(img_bytearray)
@@ -44,7 +44,7 @@ def create_card_image(model_no, img_bytearray, option_dict):
     picture_img = picture_img.resize(PICTURE_SIZE)
 
     # Loading of Card Frame
-    card_img = Image.open(front_card_img_dict[model_no])
+    card_img = Image.open(card_frame_bytearray)
 
     # Embedding a Picture in the Card
     card_img.paste(picture_img, PICTURE_LT_XY)