feat: change image style

.gitignore

@@ -0,0 +1,25 @@
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+.idea/
+nacos/
+logs/
+build/
+
+# IntelliJ related
+*.iml
+*.ipr
+*.iws
+*.db
+
+.DS_Store
+.vscode/

app.py

@@ -0,0 +1,79 @@
+import gradio as gr
+import requests
+
+prefix = "https://smartfeed-custom-tools.hf.space/file="
+
+
+def change_image_style(image_url, style_image_url):
+    data = {
+        "image_url": image_url,
+        "style_image_url": style_image_url
+    }
+
+    response = requests.post(
+        "https://api.hkhappymobile.com/tools/change-image-style",
+        json=data,
+        headers={"content-type": "application/json"}
+    )
+
+    if response.status_code == 200:
+        resultImageUrl = response.json().get("data").get("image_url")
+        return resultImageUrl
+    else:
+        raise Exception(f"Error: {response.status_code} - {response.text}")
+
+def generate_image(input_image, style_images):
+
+    if not input_image:
+        raise gr.Error(f"Please upload an input image! Refer to step 1️⃣")
+
+    if style_images is None:
+        raise gr.Error(f"Cannot find any style image! Please refer to step 1️⃣")
+
+    inputImageUrl = prefix + input_image
+
+    result_images = list[str]()
+    for style_image in style_images:
+        if not style_image:
+            raise gr.Error(f"Cannot find any style image! Please refer to step 1️⃣")
+        styleImageUrl = prefix + style_image[0]
+        result_images.append(change_image_style(inputImageUrl, styleImageUrl))
+
+    return result_images
+    
+
+def swap_to_gallery(images):
+    return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False)
+
+def remove_back_to_files():
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(label="Input Image", type="filepath", interactive=True)
+
+            files = gr.File(
+                        label="Drag (Select) 1 or more style images",
+                        file_types=["image"],
+                        file_count="multiple"
+                    )
+            uploaded_files = gr.Gallery(label="Your images", visible=False, columns=5, rows=1, height=200)
+            with gr.Column(visible=False) as clear_button:
+                remove_and_reupload = gr.ClearButton(value="Remove and upload new ones", components=files, size="sm")
+            submit = gr.Button("Submit")
+
+        with gr.Column():
+            gallery = gr.Gallery(label="Generated Images")
+
+        files.upload(fn=swap_to_gallery, inputs=files, outputs=[uploaded_files, clear_button, files])
+        remove_and_reupload.click(fn=remove_back_to_files, outputs=[uploaded_files, clear_button, files])
+
+        submit.click(
+            fn=generate_image,
+            inputs=[input_image, uploaded_files],
+            outputs=[gallery]
+        )
+
+    
+demo.launch()