Golden Frame 0.1.24

MANIFEST.in

@@ -1 +1,3 @@
 include assets/*
+include src/*
+include src/golden_frame/*

README.md

@@ -10,7 +10,13 @@ _work in progress, making a Python Package is so complex just like npm_
 <img src="example/zhongxina_before.jpg" width = 200 /> **=**
 <img src="example/zhongxina_after.png" width=200 />
 
-PS. The original picture of golden frame is K-Pop Star (Search: กรอบทอง ทรงพระเจริญ in Google, there is many variant)
+PS. The original picture of golden frame is K-Pop Star (Search: กรอบทอง ทรงพระเจริญ in Google, there are many variant)
+
+### Command for Above Example
+
+```bash
+src/cli.py build golden_frame.png example/zhongxina_before.jpg --output=example/zhongxina_after.png
+```
 
 ## Adding Images
 
@@ -24,3 +30,7 @@ file_name.json Schema
 ```
 
 pos => x1,y1,x2,y2 ; Position to put pictures on, you can get these info using Paint
+
+## TODO
+
+- Upload to PyPi (This is so mucking hard, muck Python)

assets/vladdy_daddy.json

@@ -0,0 +1,4 @@
+{
+    "name": "Vladimir Putin does not #StandWithUkraine but I do",
+    "pos": "172,456,369,803"
+}

setup.cfg

@@ -2,7 +2,8 @@
 name = golden_frame
 description = Golden Frame Generator, Create Golden Frame for given images, it also supports other frame.
 long_description = file: README.md
-version = 0.1.10
+long_description_content_type = text/markdown
+version = 0.1.24
 license = MIT
 author = Leomotors
 author_email = [email protected]

src/cli.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+import typer
+
+from lib import PosOptions, buildFromPreset, listFrames
+
+app = typer.Typer()
+
+
+@app.command()
+def build(frame_name: str, input: str, output="output.png", pos=PosOptions.CENTER):
+    """Build the Golden Frame
+
+    Args:
+
+        frame_name (str): Name of Template Frame
+
+        input (str): Location of your Image
+
+        output (str, optional): Output Location. Defaults to "output.png".
+
+        pos (0 | 1 | 2, optional): Position Type CENTER = 0, START = 1, END = 2. Defaults to CENTER.
+    """
+    buildFromPreset(frame_name, input, output, pos)
+
+
+@app.command()
+def list():
+    """Print List of Available Frames
+    """
+    print(listFrames())
+
+
+def main():
+    app()
+
+
+if __name__ == "__main__":
+    main()

golden_frame/lib.py → src/lib.py

@@ -1,4 +1,4 @@
-from typing import List, Tuple, Union
+from typing import Dict, List, Tuple, Union
 import cv2
 import numpy as np
 import json
@@ -32,7 +32,7 @@ def resizeImage(
     size: List[int],
     posOption=PosOptions.CENTER
 ) -> np.ndarray:
-    px, py = pic.shape[0:2]
+    py, px = pic.shape[0:2]
 
     ratio = max(size[0]/px, size[1]/py)
 
@@ -67,14 +67,53 @@ def buildGoldenFrame(
     return frame
 
 
-def loadConfig(name: str) -> str:
+def loadConfig(name: str) -> Dict:
     with open(".".join(name.split(".")[:-1]) + ".json") as f:
-        return json.load(f)["pos"]
+        return json.load(f)
 
 
 def buildFromPreset(frame: str, image: str, out: str, opt=PosOptions.CENTER):
-    cfg = loadConfig(frame)
-    outim = buildGoldenFrame(cv2.imread(frame), cv2.imread(image),
+    # Temporary Solution
+    frame = f"./assets/{frame}"
+    out = f"./{out}"
+
+    try:
+        frameimg = cv2.imread(frame)
+        if not frameimg.data:
+            raise
+    except:
+        print(
+            f"ERROR: Cannot Read {frame}! Did you execute this script from correct location and frame name is correct?"
+        )
+        return
+
+    try:
+        inputimg = cv2.imread(image)
+        if not inputimg.data:
+            raise
+    except:
+        print(f"ERROR: Cannot Read Input Image {image}!")
+        return
+
+    cfg = loadConfig(frame)["pos"]
+    outim = buildGoldenFrame(frameimg, inputimg,
                              list(int(k) for k in cfg.split(",")), opt)
 
-    cv2.imwrite(out, outim)
+    try:
+        cv2.imwrite(out, outim)
+    except:
+        print(f"Error writing Image!")
+
+
+def listFrames() -> str:
+    import os
+    items = list(filter(lambda x: not x.endswith(
+        ".json"), os.listdir("./assets")))
+
+    text = f"There are {len(items)} frames available.\n"
+
+    for item in items:
+        cfg = loadConfig(f"./assets/{item}")
+        text += f"\n{item} : {cfg['name']}"
+
+    return text