Upload 3 files

app.py

@@ -0,0 +1,45 @@
+import streamlit as st
+import cv2
+from PIL import Image
+import io
+import numpy as np
+from passport_img import merge_all
+
+def grayscale_image(image):
+    # Convert the image to grayscale
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    return gray
+st.title('📸Passport Photo Maker📸')
+def main():
+    # st.title("Grayscale Image Converter")
+
+    # File uploader
+    uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+    num_img = st.slider('Number of Image', 0, 36, 12)
+    if uploaded_file is not None:
+        if st.button('Make'):
+            # Read the image
+            image = Image.open(uploaded_file)
+            img_array = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+
+            # st.image(image, caption='Original Image', use_column_width=True)
+
+            # Convert to grayscale
+            gray_image = merge_all(num_img,img_array)
+            gray_image = cv2.cvtColor(np.array(gray_image), cv2.COLOR_BGR2RGB)
+
+            pil_image = Image.fromarray(gray_image)
+            st.image(pil_image, caption='Passport Image', use_column_width=True)
+
+            # Create a download link for the grayscale image
+            buffered = io.BytesIO()
+            pil_image.save(buffered, format="PNG")
+            download_btn = st.download_button(
+                label='Download Image',
+                data=buffered.getvalue(),
+                file_name='passport_image.png',
+                mime='image/png'
+            )
+
+if __name__ == '__main__':
+    main()

passport_img.py

@@ -0,0 +1,73 @@
+import cv2
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+
+def crop_passport_size(img_arr):
+    # Load the input image
+    # original_image = cv2.imread(input_image_path)
+    original_image = img_arr
+    # Define the dimensions for the passport-size photo (35mm x 45mm)
+    passport_size = (310, 410)
+
+    # Resize the image to the passport-size dimensions
+    resized_image = cv2.resize(original_image, (passport_size[0], passport_size[1]))
+
+    # Save the cropped passport-size image
+#     w_output = add_border(output, 4, 4, 4, 4, (0,0,0))
+#     w_output = add_border(w_output, 15, 15, 15, 15, (255,255,255))
+#     cv2.imwrite(output_image_path, resized_image)
+    return resized_image
+
+def add_border(image, top, bottom, left, right, color):
+    return cv2.copyMakeBorder(image, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color)
+
+def horizontal_merge(number,crop_img):
+    white_image = np.ones((450, 350, 3), dtype=np.uint8) * 255
+    list_img = []
+    # for i in range(5):
+    for i in range(number):
+        list_img.append(crop_img)
+    for i in range(6-number):
+        list_img.append(white_image)
+    merge_img = cv2.hconcat(list_img)
+    return merge_img
+
+def merge_all(number,input_path):
+    
+    input_image_path = input_path  # Replace with the path to your input image
+    # output_image_path = output_path  # Replace with the desired output path
+    output=crop_passport_size(input_image_path)
+
+    w_output = add_border(output, 4, 4, 4, 4, (0,0,0))
+    w_output = add_border(w_output, 15, 15, 15, 15, (255,255,255))
+    w_output = add_border(w_output, 1, 1, 1, 1, (0,0,0))
+
+    img_2 = []
+    num_ = number
+    if num_%6 == 0:
+        for i in range(int(num_/6)):
+            img1 = horizontal_merge(6,w_output)
+            img_2.append(img1)
+    #         print('fill')
+        for i in range(6-int(num_/6)):
+            img1_1 = horizontal_merge(0,w_output)
+            img_2.append(img1_1)
+    #         print('blank')
+    elif num_%6 != 0:
+        for i in range(int(num_//6)):
+            img1 = horizontal_merge(6,w_output)
+            img_2.append(img1)
+    #         print('fill')
+        img1 = horizontal_merge(num_%6,w_output)
+        img_2.append(img1)
+    #     print('some')
+        for i in range(6-int(num_//6)-1):
+            img1 = horizontal_merge(0,w_output)
+            img_2.append(img1)
+    #         print('blank')
+    merge_img_final = cv2.vconcat(img_2)
+    # cv2.imwrite(output_path, merge_img_final)
+    # print(merge_img_final.shape)
+    return merge_img_final

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+opencv