decomposition

requirements.txt

@@ -1,3 +1,7 @@
 gradio==4.32.2
+links==1.0
+Markdown==3.6
+Pillow==10.3.0
+python_frontmatter==1.1.0
 qrcode==7.4.2
 Requests==2.32.3

src/modules/app.py

@@ -0,0 +1,79 @@
+import gradio as gr
+
+from link import UrlProcessor
+# from main import TextProcessor
+
+
+description = f'''
+You must set the BITLY_ACCESS_TOKEN environment variable.
+For more information, see
+- https://app.bitly.com/settings/api
+'''
+
+with gr.Blocks() as demo:
+    gr.Markdown(description)
+    with gr.Row():
+        with gr.Column():
+            access_token = gr.Textbox(label="Access Token", placeholder="***")
+            api_url = gr.Textbox(label="API base URL", placeholder="https://")
+            txt_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.Row():
+                    pos_x = gr.Number(91, label="Position X", minimum=0, maximum=1000, step=1)
+                    pos_y = gr.Number(55, label="Position Y", minimum=0, maximum=1000, step=1)
+                    pos_px = gr.Number(14, label="Position Padding", minimum=0, maximum=1000, step=1)
+                    pos_py = gr.Number(11, label="Position Padding", minimum=0, maximum=1000, step=1)
+
+            # positions = [(10, 10), (int(0.9 * s * 3.7795), int(0.9 * 55 * 3.7795))]
+        with gr.Column():
+            file_output = gr.File(label="Download ZIP")
+            with gr.Accordion(open=True, label="Gallery"):
+                gallery_output = gr.Gallery(label="PNG Gallery")
+            with gr.Accordion(open=False, label="Summary"):
+                json_output = gr.JSON(label="JSON Summary")
+            with gr.Accordion(open=False, label="Preview"):
+                preview_output = gr.TextArea(label="Raw Text")
+
+    # example_temp_html = TextProcessor.apply_style('demo/demo.md', 'style.css') # TODO
+    with open('demo/demo.md') as f:
+        demo_txt = f.read()
+    examples =     [
+        ['https://example.com',
+        ['demo/demo.jpg'],
+        ['demo/demo.jpg']
+        ]
+    ]
+    inputs = [
+        txt_input, cm_img_input, idv_img_input
+    ]
+    gr.Examples(examples, inputs)
+
+    submit_button.click(
+        UrlProcessor.shorten_and_generate_qr,
+        inputs=[
+            access_token,
+            api_url,
+            txt_input,
+            shorten_checkbox,
+            qr_checkbox,
+            cm_img_input,
+            idv_img_input,
+            pos_x,
+            pos_y,
+            pos_px,
+            pos_py,
+            ],
+        outputs=[preview_output, gallery_output, file_output, json_output]
+    )
+
+demo.launch()

src/modules/ia-api.py

@@ -1,164 +0,0 @@
-import os
-import gradio as gr
-import subprocess
-
-class WaybackAPI:
-    def __init__(self):
-        pass
-
-    def configure(self, config_file):
-        try:
-            result = subprocess.run(["ia", "configure", "--config-file", config_file], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def upload(self, identifier, files, urls, metadata):
-        results = []
-        if files:
-            for file in files:
-                try:
-                    cmd = ["ia", "upload", identifier, file.name] + [f"--metadata={m}" for m in metadata.split(",")]
-                    result = subprocess.run(cmd, capture_output=True, text=True)
-                    results.append((file.name, result.stdout if result.returncode == 0 else result.stderr))
-                except Exception as e:
-                    results.append((file.name, str(e)))
-
-        if urls:
-            for url in urls.split(","):
-                try:
-                    cmd = ["ia", "upload", identifier, url] + [f"--metadata={m}" for m in metadata.split(",")]
-                    result = subprocess.run(cmd, capture_output=True, text=True)
-                    results.append((url, result.stdout if result.returncode == 0 else result.stderr))
-                except Exception as e:
-                    results.append((url, str(e)))
-
-        return results
-
-    def download(self, identifier, files=None, glob=None, format=None):
-        try:
-            cmd = ["ia", "download", identifier]
-            if files:
-                cmd += [file.name for file in files]
-            if glob:
-                cmd += ["--glob", glob]
-            if format:
-                cmd += ["--format", format]
-            result = subprocess.run(cmd, capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def delete(self, identifier, file=None, cascade=False, all_files=False):
-        try:
-            cmd = ["ia", "delete", identifier]
-            if file:
-                cmd.append(file)
-            if cascade:
-                cmd.append("--cascade")
-            if all_files:
-                cmd.append("--all")
-            result = subprocess.run(cmd, capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def search(self, query, parameters=None, itemlist=False):
-        try:
-            cmd = ["ia", "search", query]
-            if parameters:
-                cmd += ["--parameters", parameters]
-            if itemlist:
-                cmd.append("--itemlist")
-            result = subprocess.run(cmd, capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def list_files(self, identifier):
-        try:
-            result = subprocess.run(["ia", "list", identifier], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def tasks(self, identifier=None):
-        try:
-            cmd = ["ia", "tasks"]
-            if identifier:
-                cmd.append(identifier)
-            result = subprocess.run(cmd, capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-wayback_api = WaybackAPI()
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            with gr.Accordion("Configure", open=True):
-                config_file = gr.Textbox(os.getenv('IA_ACCESS_KEY', None), label="API keys")
-                configure_button = gr.Button("Configure")
-
-            with gr.Accordion("Upload", open=True):
-                identifier = gr.Textbox(label="Identifier")
-                files = gr.Files(label="Files")
-                urls = gr.Textbox(label="URLs (comma separated)")
-                metadata = gr.Textbox(label="Metadata (comma separated)")
-                upload_button = gr.Button("Upload")
-
-            with gr.Accordion("Download", open=False):
-                identifier = gr.Textbox(label="Identifier")
-                files = gr.Files(label="Files")
-                glob = gr.Textbox(label="Glob Pattern")
-                format = gr.Textbox(label="Format")
-                download_button = gr.Button("Download")
-
-            with gr.Accordion("Delete", open=False):
-                identifier = gr.Textbox(label="Identifier")
-                file = gr.Textbox(label="File to delete")
-                cascade = gr.Checkbox(label="Cascade")
-                all_files = gr.Checkbox(label="Delete all files")
-                delete_button = gr.Button("Delete")
-
-            with gr.Accordion("Search", open=False):
-                query = gr.Textbox(label="Search Query")
-                parameters = gr.Textbox(label="Parameters")
-                itemlist = gr.Checkbox(label="Generate Itemlist")
-                search_button = gr.Button("Search")
-
-            with gr.Accordion("List Files", open=False):
-                identifier = gr.Textbox(label="Identifier")
-                list_button = gr.Button("List Files")
-
-            with gr.Accordion("Tasks", open=False):
-                identifier = gr.Textbox(label="Identifier")
-                tasks_button = gr.Button("Get Tasks")
-        gr.Examples([['https://news.yahoo.co.jp/pickup/6502784']], identifier)
-
-        with gr.Column():
-            configure_output = gr.Textbox(label="Output")
-            configure_error = gr.Error()
-            upload_output = gr.Textbox(label="Output")
-            upload_error = gr.Error()
-            download_output = gr.Textbox(label="Output")
-            download_error = gr.Error()
-            delete_output = gr.Textbox(label="Output")
-            delete_error = gr.Error()
-            search_output = gr.Textbox(label="Output")
-            search_error = gr.Error()
-            list_output = gr.Textbox(label="Output")
-            list_error = gr.Error()
-            tasks_output = gr.Textbox(label="Output")
-            tasks_error = gr.Error()
-
-    configure_button.click(wayback_api.configure, [config_file], [configure_output, ])
-    upload_button.click(wayback_api.upload, [identifier, files, urls, metadata], [upload_output, ])
-    download_button.click(wayback_api.download, [identifier, files, glob, format], [download_output, ])
-    delete_button.click(wayback_api.delete, [identifier, file, cascade, all_files], [delete_output, ])
-    search_button.click(wayback_api.search, [query, parameters, itemlist], [search_output, ])
-    list_button.click(wayback_api.list_files, [identifier], [list_output, ])
-    tasks_button.click(wayback_api.tasks, [identifier], [tasks_output, ])
-
-demo.launch()

src/modules/{bitly-api.min.py → link.py}

@@ -1,28 +1,14 @@
 import os
+import requests
 import re
 import base64
-import requests
 import json
-import tempfile
-import zipfile
-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'''
-You must set the BITLY_ACCESS_TOKEN environment variable.
-For more information, see
-- https://app.bitly.com/settings/api
-- https://dev.bitly.com/api-reference/#createBitlink
-- https://dev.bitly.com/api-reference/#createQRCodePublic
-'''
+from utils import Utils
+from main import ImageProcessor
 
 
 BITLY_ACCESS_TOKEN = os.getenv('BITLY_ACCESS_TOKEN', None)
@@ -93,20 +79,25 @@ class UrlProcessor:
         except Exception as e:
             return None, None, {"error": str(e)}
 
+
     @staticmethod
-    def shorten_and_generate_qr(access_token, api_url, text, shorten, generate_qr, common_images, individual_images, pos_x, pos_y, pos_p):
+    def shorten_and_generate_qr(access_token, api_url, txt, shorten, generate_qr, common_images, individual_images, pos_x, pos_y, pos_px, pos_py):
         bitly = UrlProcessor(access_token, api_url)
-        urls = re.findall(r'http[s]?://\S+', text)
+
+        urls = re.findall(r'http[s]?://\S+', txt)
+
         results = []
-        ts, tmpd = Interface.get_tempdir()
+        ts, tmpd = Utils.get_tempdir()
 
-        markdown_links = text
+        markdown_links = txt
         all_json_responses = []
         qr_svgs = []
         qr_pngs = []
         card_htmls = []
+        index_path = 'index.html'
+        css_path = 'style.css'
 
-        positions = [(pos_p, pos_p), (int(0.9 * pos_x * 3.7795), int(0.9 * pos_y * 3.7795))]
+        positions = [(pos_px, pos_py), (int(0.9 * pos_x * 3.7795), int(0.9 * pos_y * 3.7795))]
 
         for idx, url in enumerate(urls):
             shorten_url, shorten_response_json = None, None
@@ -127,13 +118,15 @@ class UrlProcessor:
                         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:
+                        os.makedirs(f'{tmpd}/img/qr', exist_ok=True)
+                        qr_png_path = f'img/qr/QR{idx:05d}.png'
+                        qr_path = os.path.join(tmpd, qr_png_path)
+                        with open(qr_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_pngs.append(qr_path)
                         qr_svgs.append(qr_html)
                         all_json_responses.append(qr_json)
                     else:
@@ -141,13 +134,15 @@ class UrlProcessor:
                 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:
+                        os.makedirs(f'{tmpd}/img/qr', exist_ok=True)
+                        qr_png_path = f'img/qr/QR{idx:05d}.png'
+                        qr_path = os.path.join(tmpd, qr_png_path)
+                        with open(qr_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_pngs.append(qr_path)
                         qr_svgs.append(qr_html)
                         all_json_responses.append(qr_json)
                     else:
@@ -186,13 +181,14 @@ class UrlProcessor:
             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')
+            markdown_output_path = os.path.join(tmpd, 'index.md')
             with open(markdown_output_path, 'w') as markdown_file:
                 markdown_file.write(markdown_links)
 
             qr_svg_output_paths = []
+            os.makedirs(f'{tmpd}/chunks/qr', exist_ok=True)
             for idx, svg in enumerate(qr_svgs):
-                svg_output_path = os.path.join(tmpd, f'qr_code_{idx}.html')
+                svg_output_path = os.path.join(tmpd, f'chunks/qr/QR{idx:05d}.html')
                 with open(svg_output_path, 'w') as svg_file:
                     svg_file.write(svg)
                 qr_svg_output_paths.append(svg_output_path)
@@ -202,18 +198,19 @@ class UrlProcessor:
                 qr_png_output_paths.append(png)
 
             card_output_paths = []
+            os.makedirs(f'{tmpd}/chunks/cards', exist_ok=True)
             for idx, card_html in enumerate(card_htmls):
-                card_output_path = os.path.join(tmpd, f'card_{idx}.html')
+                card_output_path = os.path.join(tmpd, f'chunks/cards/CD{idx:05d}.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>'
+            index_html = f'<html><head><link rel="stylesheet" type="text/css" href="{css_path}"></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')
+            index_output_path = os.path.join(tmpd, index_path)
             with open(index_output_path, 'w') as index_file:
                 index_file.write(index_html)
 
@@ -224,211 +221,12 @@ class UrlProcessor:
                 *qr_svg_output_paths,
                 *qr_png_output_paths,
                 *card_output_paths,
-                index_output_path
+                index_output_path,
+                css_path
             ]
-            zip_output_path = Interface.create_zip(zip_content_list, zip_output_path)
+            zip_output_path = Utils.create_zip(zip_content_list, zip_output_path)
         except Exception as e:
             return f"An error occurred: {str(e)}", None, None, []
 
         return markdown_links, qr_png_output_paths, zip_output_path, all_json_responses
 
-
-class TextProcessor:
-    def apply_frontmatter(text, replacements):
-        for key, value in replacements.items():
-            text = text.replace(f'{{{key}}}', value)
-        return text
-
-    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:
-
-
-    @staticmethod
-    def create_layer(layout='horizontal', size=(91 * 3.7795, 55 * 3.7795)):
-        ts, temp_dir = Interface.get_tempdir()
-        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="lightgray")
-        elif layout == 'horizontal':
-            draw.rectangle([(0, 0), (size[0], size[1])], outline="lightgray")
-
-        file_path = os.path.join(temp_dir, 'layer.png')
-        img.save(file_path, format="PNG")
-        return file_path
-
-    @staticmethod
-    def combine_images(base_image_path, overlay_image_paths, positions=None, temp_dir=None):
-        try:
-            if temp_dir is None:
-                _, temp_dir = Interface.get_tempdir()
-            with open(base_image_path, 'rb') as f:
-                base_img = Image.open(f).convert("RGBA")
-            for i, overlay_path in enumerate(overlay_image_paths):
-                with open(overlay_path, 'rb') as f:
-                    overlay_img = Image.open(f).convert("RGBA")
-                if positions and i < len(positions):
-                    position = positions[i]
-                else:
-                    position = (0, 0)
-                overlay_img_resized = ImageOps.contain(overlay_img, base_img.size)
-                base_img.paste(overlay_img_resized, position, overlay_img_resized)
-
-            file_path = os.path.join(temp_dir, 'combined.png')
-            base_img.save(file_path, format="PNG")
-            return file_path
-        except Exception as e:
-            raise Exception(f"Error combining images: {e}")
-    # @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
-
-    @staticmethod
-    def move_image(image_path, position, temp_dir=None):
-        try:
-            if temp_dir is None:
-                _, temp_dir = Interface.get_tempdir()
-            with open(image_path, 'rb') as f:
-                img = Image.open(f).convert("RGBA")
-            result_img = Image.new("RGBA", img.size, (255, 255, 255, 0))
-            result_img.paste(img, position, img)
-            file_path = os.path.join(temp_dir, 'moved.png')
-            result_img.save(file_path, format="PNG")
-            return file_path
-        except Exception as e:
-            raise Exception(f"Error moving image: {e}")
-
-class Interface:
-    @staticmethod
-    def get_tempdir():
-        try:
-            timestamp = int(time.time())
-            temp_dir = tempfile.mkdtemp()
-            return timestamp, temp_dir
-        except Exception as e:
-            raise RuntimeError(f"Failed to create temporary directory: {str(e)}")
-
-    @staticmethod
-    def create_zip(filelist, tmp_fname):
-        if not filelist:
-            return None
-        try:
-            tmp_dir = os.path.dirname(filelist[0])
-            zip_name = os.path.join(tmp_dir, tmp_fname)
-            with zipfile.ZipFile(zip_name, "w") as f:
-                for file in filelist:
-                    f.write(file, os.path.basename(file))
-            return zip_name
-        except Exception as e:
-            raise RuntimeError(f"Failed to create zip file: {str(e)}")
-
-
-with gr.Blocks() as demo:
-    gr.Markdown(description)
-    with gr.Row():
-        with gr.Column():
-            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.Row():
-                    pos_x = gr.Number(91, label="Position X", minimum=0, maximum=1000, step=1)
-                    pos_y = gr.Number(55, label="Position Y", minimum=0, maximum=1000, step=1)
-                    pos_p = gr.Number(10, label="Position Padding", minimum=0, maximum=1000, step=1)
-
-            # positions = [(10, 10), (int(0.9 * s * 3.7795), int(0.9 * 55 * 3.7795))]
-        with gr.Column():
-            file_output = gr.File(label="Download ZIP")
-            with gr.Accordion(open=True, label="Gallery"):
-                gallery_output = gr.Gallery(label="PNG Gallery")
-            with gr.Accordion(open=False, label="Summary"):
-                json_output = gr.JSON(label="JSON Summary")
-            with gr.Accordion(open=False, label="Preview"):
-                preview_output = gr.TextArea(label="Raw Text")
-
-    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(
-        UrlProcessor.shorten_and_generate_qr,
-        inputs=[
-            access_token,
-            api_url,
-            url_input,
-            shorten_checkbox,
-            qr_checkbox,
-            cm_img_input,
-            idv_img_input,
-            pos_x,
-            pos_y,
-            pos_p,
-            ],
-        outputs=[preview_output, gallery_output, file_output, json_output]
-    )
-
-demo.launch()

src/modules/main.py

@@ -0,0 +1,116 @@
+import os
+import markdown
+import frontmatter
+from PIL import Image, ImageDraw, ImageFont, ImageOps
+
+from utils import Utils
+
+class TextProcessor:
+    def apply_frontmatter(text, replacements):
+        for key, value in replacements.items():
+            text = text.replace(f'{{{key}}}', value)
+        return text
+
+    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:
+
+    @staticmethod
+    def create_layer(layout='horizontal', size=(91 * 3.7795, 55 * 3.7795)):
+        ts, temp_dir = Utils.get_tempdir()
+        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="lightgray")
+        elif layout == 'horizontal':
+            draw.rectangle([(0, 0), (size[0], size[1])], outline="lightgray")
+
+        file_path = os.path.join(temp_dir, 'layer.png')
+        img.save(file_path, format="PNG")
+        return file_path
+
+    @staticmethod
+    def combine_images(base_image_path, overlay_image_paths, positions=None, temp_dir=None):
+        try:
+            if temp_dir is None:
+                _, temp_dir = Utils.get_tempdir()
+            with open(base_image_path, 'rb') as f:
+                base_img = Image.open(f).convert("RGBA")
+            for i, overlay_path in enumerate(overlay_image_paths):
+                with open(overlay_path, 'rb') as f:
+                    overlay_img = Image.open(f).convert("RGBA")
+                if positions and i < len(positions):
+                    position = positions[i]
+                else:
+                    position = (0, 0)
+                overlay_img_resized = ImageOps.contain(overlay_img, base_img.size)
+                base_img.paste(overlay_img_resized, position, overlay_img_resized)
+
+            file_path = os.path.join(temp_dir, 'combined.png')
+            base_img.save(file_path, format="PNG")
+            return file_path
+        except Exception as e:
+            raise Exception(f"Error combining images: {e}")
+    # @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
+
+    @staticmethod
+    def move_image(image_path, position, temp_dir=None):
+        try:
+            if temp_dir is None:
+                _, temp_dir = Utils.get_tempdir()
+            with open(image_path, 'rb') as f:
+                img = Image.open(f).convert("RGBA")
+            result_img = Image.new("RGBA", img.size, (255, 255, 255, 0))
+            result_img.paste(img, position, img)
+            file_path = os.path.join(temp_dir, 'moved.png')
+            result_img.save(file_path, format="PNG")
+            return file_path
+        except Exception as e:
+            raise Exception(f"Error moving image: {e}")
+

src/modules/mega-api.py

@@ -1,117 +0,0 @@
-import os
-import gradio as gr
-import subprocess
-
-class MEGAcmdAPI:
-    def __init__(self):
-        pass
-
-    def login(self, email, password):
-        try:
-            result = subprocess.run(["mega-login", email, password], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def logout(self):
-        try:
-            result = subprocess.run(["mega-logout"], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def upload(self, local_path, remote_path):
-        try:
-            result = subprocess.run(["mega-put", local_path, remote_path], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def download(self, remote_path, local_path):
-        try:
-            result = subprocess.run(["mega-get", remote_path, local_path], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def list(self, remote_path):
-        try:
-            result = subprocess.run(["mega-ls", remote_path], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def delete(self, remote_path):
-        try:
-            result = subprocess.run(["mega-rm", remote_path], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-    def mkdir(self, remote_path):
-        try:
-            result = subprocess.run(["mega-mkdir", remote_path], capture_output=True, text=True)
-            return result.stdout if result.returncode == 0 else result.stderr
-        except Exception as e:
-            return str(e)
-
-mega_api = MEGAcmdAPI()
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            with gr.Accordion("Login", open=False):
-                email = gr.Textbox(label="Email")
-                password = gr.Textbox(getenv('MEGA_API_KEY'), label="Password", type="password")
-                login_button = gr.Button("Login")
-
-            with gr.Accordion("Logout", open=False):
-                logout_button = gr.Button("Logout")
-
-            with gr.Accordion("Upload", open=False):
-                local_path = gr.Textbox(label="Local Path")
-                remote_path = gr.Textbox(label="Remote Path")
-                upload_button = gr.Button("Upload")
-
-            with gr.Accordion("Download", open=False):
-                remote_path = gr.Textbox(label="Remote Path")
-                local_path = gr.Textbox(label="Local Path")
-                download_button = gr.Button("Download")
-
-            with gr.Accordion("List Files", open=False):
-                remote_path = gr.Textbox(label="Remote Path")
-                list_button = gr.Button("List Files")
-
-            with gr.Accordion("Delete", open=False):
-                remote_path = gr.Textbox(label="Remote Path")
-                delete_button = gr.Button("Delete")
-
-            with gr.Accordion("Create Directory", open=False):
-                remote_path = gr.Textbox(label="Remote Path")
-                mkdir_button = gr.Button("Create Directory")
-
-        with gr.Column():
-            login_output = gr.Textbox(label="Output")
-            login_error = gr.Error()
-            logout_output = gr.Textbox(label="Output")
-            logout_error = gr.Error()
-            upload_output = gr.Textbox(label="Output")
-            upload_error = gr.Error()
-            download_output = gr.Textbox(label="Output")
-            download_error = gr.Error()
-            list_output = gr.Textbox(label="Output")
-            list_error = gr.Error()
-            delete_output = gr.Textbox(label="Output")
-            delete_error = gr.Error()
-            mkdir_output = gr.Textbox(label="Output")
-            mkdir_error = gr.Error()
-
-    login_button.click(mega_api.login, [email, password], [login_output, ])
-    logout_button.click(mega_api.logout, [], [logout_output, ])
-    upload_button.click(mega_api.upload, [local_path, remote_path], [upload_output, ])
-    download_button.click(mega_api.download, [remote_path, local_path], [download_output, ])
-    list_button.click(mega_api.list, [remote_path], [list_output, ])
-    delete_button.click(mega_api.delete, [remote_path], [delete_output, ])
-    mkdir_button.click(mega_api.mkdir, [remote_path], [mkdir_output, ])
-
-demo.launch()

src/modules/utils.py

@@ -0,0 +1,42 @@
+import os
+import time
+import tempfile
+import zipfile
+
+
+class Utils:
+    @staticmethod
+    def get_tempdir():
+        try:
+            timestamp = int(time.time())
+            temp_dir = tempfile.mkdtemp()
+            return timestamp, temp_dir
+        except Exception as e:
+            raise RuntimeError(f"Failed to create temporary directory: {str(e)}")
+
+    @staticmethod
+    def create_zip(filelist, tmp_fname, passwd=None):
+        if not filelist:
+            return None
+        try:
+            zip_name = os.path.abspath(tmp_fname)
+            with zipfile.ZipFile(zip_name, "w", compression=zipfile.ZIP_DEFLATED) as f:
+                for file in filelist:
+                    if os.path.isfile(file):
+                        f.write(file, os.path.relpath(file, os.path.dirname(filelist[0])))
+                    elif os.path.isdir(file):
+                        for root, dirs, files in os.walk(file):
+                            for filename in files:
+                                filepath = os.path.join(root, filename)
+                                f.write(filepath, os.path.relpath(filepath, os.path.dirname(filelist[0])))
+            if passwd:
+                zip_name_encrypted = zip_name + ".zip"
+                with zipfile.ZipFile(zip_name_encrypted, "w", compression=zipfile.ZIP_DEFLATED) as f:
+                    f.setpassword(passwd)
+                    f.write(zip_name, os.path.basename(zip_name))
+                os.remove(zip_name)
+                return zip_name_encrypted
+            else:
+                return zip_name
+        except Exception as e:
+            raise RuntimeError(f"Failed to create zip file: {str(e)}")