update stuff

demo.md

@@ -0,0 +1,14 @@
+---
+replace:
+- name: Taro Tanaka
+- group: Google Inc.
+- email: [email protected]
+- phone: 090 1234 5678
+- link: https://example.com
+---
+Name: {name}
+Group: {group}
+Contact:
+- Email: {email}
+- Phone: {phone}
+- Link: {link}

src/modules/bitly-api.min.py

@@ -9,6 +9,11 @@ import shutil
 import qrcode
 from io import BytesIO
 import time
+import textwrap
+import markdown
+import frontmatter
+from PIL import Image, ImageDraw, ImageFont, ImageOps
+from io import BytesIO
 import gradio as gr
 
 description = f'''
@@ -23,7 +28,7 @@ For more information, see
 BITLY_ACCESS_TOKEN = os.getenv('BITLY_ACCESS_TOKEN', None)
 BITLY_API_URL = "https://api-ssl.bitly.com/v4"
 
-class ShortenUrl:
+class UrlProcessor:
     def __init__(self, access_token=None, api_url=None):
         self.access_token = access_token if access_token else BITLY_ACCESS_TOKEN
         self.api_url = api_url if api_url else BITLY_API_URL
@@ -88,105 +93,228 @@ class ShortenUrl:
         except Exception as e:
             return None, None, {"error": str(e)}
 
-def shorten_and_generate_qr(access_token, api_url, text, shorten, generate_qr):
-    bitly = ShortenUrl(access_token, api_url)
-    urls = re.findall(r'http[s]?://\S+', text)
-    results = []
-    ts, tmpd = Interface.get_tempdir()
-
-    markdown_links = text
-    all_json_responses = []
-    qr_svgs = []
-    qr_pngs = []
-
-    for idx, url in enumerate(urls):
-        shorten_url, shorten_response_json = None, None
-        png_file_name = f'qr_code_{idx}.png'
-        png_output_path = os.path.join(tmpd, png_file_name)
-        try:
-            if shorten:
-                shorten_url, shorten_response_json = bitly.shorten(url)
-                all_json_responses.append(shorten_response_json)
-
-            if shorten and shorten_url:
-                url_to_process = shorten_url
-            else:
-                url_to_process = url
-
-            if generate_qr and url_to_process:
-                if shorten_url:
-                    # qr_html, qr_png, qr_json = bitly.generate_qr(url_to_process)
-                    qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
-                else:
-                    qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
+    @staticmethod
+    def shorten_and_generate_qr(access_token, api_url, text, shorten, generate_qr, common_images, individual_images):
+        bitly = UrlProcessor(access_token, api_url)
+        urls = re.findall(r'http[s]?://\S+', text)
+        results = []
+        ts, tmpd = Interface.get_tempdir()
 
-                if qr_png:
-                    markdown_links = markdown_links.replace(
-                        # url, f'{qr_html} [{url_to_process}]({url}) '
-                        url, f'<img src="{png_file_name}" height="40"> [{url}](https://{url_to_process}) '
-                    )
-                    qr_pngs.append(qr_png)
-                    qr_svgs.append(qr_html)
-                    all_json_responses.append(qr_json)
+        markdown_links = text
+        all_json_responses = []
+        qr_svgs = []
+        qr_pngs = []
+        card_htmls = []
+
+        for idx, url in enumerate(urls):
+            shorten_url, shorten_response_json = None, None
+            try:
+                if shorten:
+                    shorten_url, shorten_response_json = bitly.shorten(url)
+                    all_json_responses.append(shorten_response_json)
+
+                if shorten and shorten_url:
+                    url_to_process = shorten_url
                 else:
-                    results.append(("Error generating QR code", url_to_process))
-            elif generate_qr:
-                qr_html, qr_png, qr_json = bitly.generate_qr_local(url)
-                if qr_png:
+                    url_to_process = url
+
+                if generate_qr and url_to_process:
+                    if shorten_url:
+                        qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
+                    else:
+                        qr_html, qr_png, qr_json = bitly.generate_qr_local(url_to_process)
+
+                    if qr_png:
+                        qr_png_path = os.path.join(tmpd, f'qr_code_{idx}.png')
+                        with open(qr_png_path, 'wb') as f:
+                            f.write(qr_png)
+                        markdown_links = markdown_links.replace(
+                            url, f'<img src="{qr_png_path}" height="40"> [{url_to_process}]({url}) '
+                        )
+                        qr_pngs.append(qr_png_path)
+                        qr_svgs.append(qr_html)
+                        all_json_responses.append(qr_json)
+                    else:
+                        results.append(("Error generating QR code", url_to_process))
+                elif generate_qr:
+                    qr_html, qr_png, qr_json = bitly.generate_qr_local(url)
+                    if qr_png:
+                        qr_png_path = os.path.join(tmpd, f'qr_code_{idx}.png')
+                        with open(qr_png_path, 'wb') as f:
+                            f.write(qr_png)
+                        markdown_links = markdown_links.replace(
+                            url, f'<img src="{qr_png_path}" height="40"> [{url}]({url}) '
+                        )
+                        qr_pngs.append(qr_png_path)
+                        qr_svgs.append(qr_html)
+                        all_json_responses.append(qr_json)
+                    else:
+                        results.append(("Error generating QR code", url))
+                elif shorten_url:
                     markdown_links = markdown_links.replace(
-                        # url, f'{qr_html} [{url}]({url}) '
-                        url, f'<img src="{png_file_name}" height="40") [{url}]({url}) '
+                        url, f'[{shorten_url}]({url}) '
                     )
-                    qr_pngs.append(qr_png)
-                    qr_svgs.append(qr_html)
-                    all_json_responses.append(qr_json)
+                    results.append((None, shorten_url))
                 else:
-                    results.append(("Error generating QR code", url))
-            elif shorten_url:
-                markdown_links = markdown_links.replace(
-                    url, f'[{shorten_url}]({url}) '
-                )
-                results.append((None, shorten_url))
-            else:
-                results.append(("Error shortening URL", url))
+                    results.append(("Error shortening URL", url))
+            except Exception as e:
+                results.append((str(e), url))
+
+        layer = ImageProcessor.create_layer()
+
+        for idx, png in enumerate(qr_pngs):
+            try:
+                png = ImageProcessor.combine_images(layer, [png])
+                if common_images:
+                    png = ImageProcessor.combine_images(png, common_images)
+                if individual_images:
+                    for im in individual_images:
+                        png = ImageProcessor.combine_images(png, [im.name])
+                        with open(png, "rb") as image_file:
+                            encoded_string = base64.b64encode(image_file.read()).decode()
+
+                        card_html = f'<img src="data:image/png;base64,{encoded_string}" alt="Card"> '
+                        card_htmls.append(card_html)
+            except Exception as e:
+                raise Exception(f"Error combining images: {e}")
+
+
+        try:
+            json_output_path = os.path.join(tmpd, 'response.json')
+            with open(json_output_path, 'w') as json_file:
+                json.dump(all_json_responses, json_file)
+
+            markdown_output_path = os.path.join(tmpd, 'links.md')
+            with open(markdown_output_path, 'w') as markdown_file:
+                markdown_file.write(markdown_links)
+
+            qr_svg_output_paths = []
+            for idx, svg in enumerate(qr_svgs):
+                svg_output_path = os.path.join(tmpd, f'qr_code_{idx}.html')
+                with open(svg_output_path, 'w') as svg_file:
+                    svg_file.write(svg)
+                qr_svg_output_paths.append(svg_output_path)
+
+            qr_png_output_paths = []
+            for idx, png in enumerate(qr_pngs):
+                qr_png_output_paths.append(png)
+
+            card_output_paths = []
+            for idx, card_html in enumerate(card_htmls):
+                card_output_path = os.path.join(tmpd, f'card_{idx}.html')
+                with open(card_output_path, 'w') as card_file:
+                    card_file.write(card_html)
+                card_output_paths.append(card_output_path)
+
+            index_html = '<html><head><link rel="stylesheet" type="text/css" href="style.css"></head><body>'
+            for card_html in card_htmls:
+                index_html += card_html
+            index_html += '</body></html>'
+
+            index_output_path = os.path.join(tmpd, 'index.html')
+            with open(index_output_path, 'w') as index_file:
+                index_file.write(index_html)
+
+            zip_output_path = os.path.join(tmpd, f"{ts}.zip")
+            zip_content_list = [
+                json_output_path,
+                markdown_output_path,
+                *qr_svg_output_paths,
+                *qr_png_output_paths,
+                *card_output_paths,
+                index_output_path
+            ]
+            zip_output_path = Interface.create_zip(zip_content_list, zip_output_path)
         except Exception as e:
-            results.append((str(e), url))
+            return f"An error occurred: {str(e)}", None, None, []
 
-    try:
-        json_output_path = os.path.join(tmpd, 'response.json')
-        with open(json_output_path, 'w') as json_file:
-            json.dump(all_json_responses, json_file)
+        return markdown_links, qr_png_output_paths, zip_output_path, all_json_responses
 
-        markdown_output_path = os.path.join(tmpd, 'links.md')
-        with open(markdown_output_path, 'w') as markdown_file:
-            markdown_file.write(markdown_links)
 
-        qr_svg_output_paths = []
-        for idx, svg in enumerate(qr_svgs):
-            svg_output_path = os.path.join(tmpd, f'qr_code_{idx}.html')
-            with open(svg_output_path, 'w') as svg_file:
-                svg_file.write(svg)
-            qr_svg_output_paths.append(svg_output_path)
+class TextProcessor:
+    def apply_frontmatter(text, replacements):
+        for key, value in replacements.items():
+            text = text.replace(f'{{{key}}}', value)
+        return text
 
-        qr_png_output_paths = []
-        for idx, png in enumerate(qr_pngs):
-            png_output_path = os.path.join(tmpd, f'qr_code_{idx}.png')
-            with open(png_output_path, 'wb') as png_file:
-                png_file.write(png)
-            qr_png_output_paths.append(png_output_path)
-
-        zip_output_path = os.path.join(tmpd, f"{ts}.zip")
-        zip_content_list = [
-            json_output_path,
-            markdown_output_path,
-            *qr_svg_output_paths,
-            *qr_png_output_paths
-        ]
-        zip_output_path = Interface.create_zip(zip_content_list, zip_output_path)
-    except Exception as e:
-        return f"An error occurred: {str(e)}", None, None, []
+    def apply_style(input_markdown_path, style_css_path):
+        with open(input_markdown_path, 'r') as file:
+            md_content = file.read()
+
+        fm = frontmatter.load(input_markdown_path)
+        replacements = {key: list(item.values())[0] \
+            for item in fm.metadata['replace'] for key in item}
+        processed_text = TextProcessor.apply_frontmatter(md_content, replacements)
+
+        with open(style_css_path, 'r') as file:
+            style_css = file.read()
+
+        html_content = markdown.markdown(processed_text)
+        html_content = f'<html><head><style>{style_css}</style></head><body>{html_content}</body></html>'
+
+        temp_html_path = os.path.join(tempfile.gettempdir(), 'processed.html')
+        with open(temp_html_path, 'w') as file:
+            file.write(html_content)
+
+        return temp_html_path
+
+class ImageProcessor:
+
+
+    def create_layer(layout='horizontal', size=(91 * 3.7795, 55 * 3.7795)):
+        size = (int(size[0]), int(size[1]))  # Convert to integers
+        img = Image.new('RGBA', size, (255, 255, 255, 0))
+        draw = ImageDraw.Draw(img)
+        if layout == 'vertical':
+            draw.rectangle([(0, 0), (size[0], size[1])], outline="black", width=3)
+        elif layout == 'horizontal':
+            draw.rectangle([(0, 0), (size[1], size[0])], outline="black", width=3)
 
-    return markdown_links, qr_png_output_paths, zip_output_path, all_json_responses
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmpfile:
+            img.save(tmpfile.name, format="PNG")
+            return tmpfile.name
+
+    @staticmethod
+    def combine_images(base_image_path, overlay_image_paths):
+        with open(base_image_path, 'rb') as f:
+            base_img = Image.open(f).convert("RGBA")
+        for overlay_path in overlay_image_paths:
+            with open(overlay_path, 'rb') as f:
+                overlay_img = Image.open(f).convert("RGBA")
+            # overlay_img = overlay_img.resize(base_img.size, Image.NEAREST )
+            base_img = ImageOps.contain(overlay_img, base_img.size)
+
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmpfile:
+            base_img.save(tmpfile.name, format="PNG")
+            return tmpfile.name
+
+    # @staticmethod
+    # def insert_txt(png_path, text):
+    #     img = Image.open(png_path)
+    #     draw = ImageDraw.Draw(img)
+    #     font = ImageFont.load_default()  # Update to a specific font if needed
+    #     width, height = img.size
+    #     text_width, text_height = draw.textlength(text, font=font)
+    #     text_position = (width - text_width - 10, height - text_height - 10)
+    #     draw.text(text_position, text, font=font, fill="black")
+
+    #     with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmpfile:
+    #         img.save(tmpfile.name, format="PNG")
+    #         return tmpfile.name
+
+    # @staticmethod
+    # def insert_txt(png_path, text):
+    #     img = Image.open(png_path)
+    #     draw = ImageDraw.Draw(img)
+    #     font = ImageFont.load_default()  # Update to a specific font if needed
+    #     width, height = img.size
+    #     text_width, text_height = draw.textlength(text, font=font)
+    #     text_position = (width - text_width - 10, height - text_height - 10)
+    #     draw.text(text_position, text, font=font, fill="black")
+
+    #     with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmpfile:
+    #         img.save(tmpfile.name, format="PNG")
+    #         return tmpfile.name
 
 class Interface:
     @staticmethod
@@ -220,11 +348,15 @@ with gr.Blocks() as demo:
             access_token = gr.Textbox(label="Access Token", placeholder="***")
             api_url = gr.Textbox(label="API base URL", placeholder="https://")
             url_input = gr.TextArea(label="Enter Text with URLs")
+
             with gr.Row():
                 shorten_checkbox = gr.Checkbox(label="Shorten URL", value=True)
                 qr_checkbox = gr.Checkbox(label="Generate QR Code", value=True)
             submit_button = gr.Button("Process URLs", variant='primary')
-
+            with gr.Accordion(open=False, label="Insert"):
+                with gr.Row():
+                    cm_img_input = gr.Files(label='Insert common images')
+                    idv_img_input = gr.Files(label='Insert individual images')
         with gr.Column():
             file_output = gr.File(label="Download ZIP")
             with gr.Accordion(open=True, label="Gallery"):
@@ -234,14 +366,29 @@ with gr.Blocks() as demo:
             with gr.Accordion(open=False, label="Preview"):
                 preview_output = gr.TextArea(label="Raw Text")
 
-    gr.Examples([
-        ['http://example.com'],
-        ['http://example.com\nhttp://toscrape.com\nhttp://books.toscrape.com'],
-    ], inputs=url_input)
+    example_temp_html = TextProcessor.apply_style('demo.md', 'style.css')
+
+    examples =     [
+        ['http://example.com\nhttp://toscrape.com\nhttp://books.toscrape.com',
+        ['img/number-1.png'],
+        ['img/number-2.png']
+        ]
+    ]
+    inputs = [
+        url_input, cm_img_input, idv_img_input
+    ]
+    gr.Examples(examples, inputs)
 
     submit_button.click(
-        shorten_and_generate_qr,
-        inputs=[access_token, api_url, url_input, shorten_checkbox, qr_checkbox],
+        UrlProcessor.shorten_and_generate_qr,
+        inputs=[
+            access_token,
+            api_url,
+            url_input,
+            shorten_checkbox,
+            qr_checkbox,
+            cm_img_input,
+            idv_img_input],
         outputs=[preview_output, gallery_output, file_output, json_output]
     )
 

style.css

@@ -0,0 +1,56 @@
+h1 {
+        font-weight: bold;
+}
+
+/* フォントとマージンの設定 */
+p {
+  font-family: 'MS Mincho', serif;
+  margin: 10%;
+}
+
+/* 画像の透明度設定 */
+img {
+  opacity: 0.5;
+}
+
+/* QRコードの位置設定 */
+.qr {
+  position: absolute;
+  bottom: 10px;
+  right: 10px;
+  margin: 10%;
+}
+
+/* カードのサイズとグリッドレイアウト設定 */
+.card {
+  width: 91mm;
+  height: 55mm;
+  display: grid;
+  grid-template-columns: 1fr;
+  grid-template-rows: 1fr;
+  align-items: center;
+  justify-items: center;
+  border: 1px solid black;
+  margin: 5mm;
+  position: relative;
+}
+
+/* 印刷時のA4用紙設定 */
+@media print {
+  .a4 {
+      page-break-after: always;
+      width: 210mm;
+      height: 297mm;
+      display: grid;
+      grid-template-columns: repeat(2, 1fr);
+      grid-template-rows: repeat(5, 1fr);
+      gap: 0;
+  }
+
+  .card {
+      margin: 0;
+      border: none;
+      width: 100%;
+      height: 100%;
+  }
+}