Add GenFood into generated recipe

app.py

@@ -9,78 +9,149 @@ from PIL import (
     ImageFont,
     ImageDraw
 )
+import requests
 
-import numpy as np
-from datetime import datetime
-import glob
+import os
 import re
 import random
 import textwrap
 from examples import EXAMPLES
+import dummy
 import meta
 from utils import (
     remote_css,
     local_css,
-    load_image,
+    load_image_from_url,
+    load_image_from_local,
+    image_to_base64,
     pure_comma_separation
 )
 
-# np.random.seed(42)
+
+def generate_cook_image(query, app_id, app_key):
+    api_url = f"https://api.edamam.com/api/recipes/v2?type=public&q={query}&app_id={app_id}&app_key={app_key}&field=image"
+
+    try:
+        r = requests.get(api_url)
+        if r.status_code != 200:
+            return None
+
+        rj = r.json()
+        if "hits" not in rj or not len(rj["hits"]) > 0:
+            return None
+
+        data = rj["hits"]
+        data = data[random.randint(0, min(4, len(data) - 1))] if len(data) > 1 else data[0]
+
+        if "recipe" not in data or "image" not in data["recipe"]:
+            return None
+
+        image = data["recipe"]["image"]
+        return image
+    except Exception as e:
+        return None
+
+
+def generate_food_with_logo_image(bg_path, logo_path, food_url, no_food="asset/frame/no_food.png"):
+    bg = Image.open(bg_path)
+    width, height = bg.size
+
+    logo = Image.open(logo_path)
+    logo_width, logo_height, logo_ratio, logo_rb, logo_mb = logo.size + (3, -20, 45)
+    logo_width, logo_height = (logo_width // logo_ratio, logo_height // logo_ratio)
+    logo = logo.resize((logo_width, logo_height))
+
+    food = load_image_from_url(food_url, rgba_mode=True, default_image=no_food)
+
+    food_width, food_height = (300, 300)
+    food = food.resize((food_width, food_height))
+
+    bg.paste(food, (0, 0), food)
+    bg.paste(logo, (width - logo_width - logo_rb, height - logo_height - logo_mb), logo)
+
+    return bg
+
+
+def generate_recipe_image(
+        recipe_data,
+        bg_path,
+        food_logo_ia,
+        fonts,
+        bg_color="#ffffff"
+):
+    bg = Image.open(bg_path)
+    bg.paste(food_logo_ia, (50, 50), food_logo_ia)
+    bg_color = Image.new("RGBA", bg.size, bg_color)
+    bg_color.paste(bg, mask=bg)
+
+    im_editable = ImageDraw.Draw(bg_color)
+    im_editable.text(
+        (418, 30),
+        textwrap.fill(recipe_data["title"], 15).replace(" \n", "\n"),
+        (61, 61, 70),
+        font=fonts["title"],
+    )
+
+    im_editable.text(
+        (100, 450),
+        "Ingredients",
+        (61, 61, 70),
+        font=fonts["body_bold"],
+    )
+    ingredients = recipe_data["ingredients"]
+    ingredients = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in ingredients]
+
+    im_editable.text(
+        (50, 520),
+        "\n".join([f"- {item}" for item in ingredients]),
+        (61, 61, 70),
+        font=fonts["body"],
+    )
+
+    im_editable.text(
+        (700, 450),
+        "Directions",
+        (61, 61, 70),
+        font=fonts["body_bold"],
+    )
+
+    directions = recipe_data["directions"]
+    directions = [textwrap.fill(item, 70).replace("\n", "\n   ") for item in directions]
+    im_editable.text(
+        (430, 520),
+        "\n".join([f"{i + 1}. {item}" for i, item in enumerate(directions)]).strip(),
+        (61, 61, 70),
+        font=fonts["body"],
+    )
+    return bg_color
 
 
 class TextGeneration:
     def __init__(self):
-        self.debug = False
-        self.dummy_output = {
-            'directions': [
-                "for the dough",
-                "in a small bowl, combine the warm water and yeast.",
-                "let it sit for 5 minutes.",
-                "add the flour, salt, and olive oil.",
-                "mix well and knead until the dough is smooth and elastic.",
-                "cover the dough with a damp towel and let it rise for about 1 hour.",
-                "for the filling",
-                "heat a large skillet over medium high heat.",
-                "cook the beef and onion until the beef is browned and the onion is translucent. browned and the onion is translucent.",
-                "drain off any excess grease.",
-                "stir in the pepper and salt and black pepper to taste.",
-                "remove from the heat and set aside.",
-                "preheat the oven to 425 degrees f.",
-                "roll out the dough on a lightly floured surface into a 12 inch circle.",
-                "spread the beef mixture over the dough, leaving a 1 inch border.",
-                "top with the feta, parsley, and lemon juice.",
-                "bake for 20 minutes or until the crust is golden brown.",
-                "cut into wedges and serve.",
-            ],
-            'ingredients': [
-                '1 potato',
-                '1 slice cheese 1 slice cheese',
-                '1 potato 1 slice cheese 1 slice cheese',
-                '1 slice cheese'
-                '1 potato',
-                '1 slice cheese',
-                '1 slice cheese',
-                '1 potato',
-                '1 slice cheese'
-                '1 potato',
-                '1 slice cheese',
-            ],
-            'title': 'Cheese Potatoes with Some other items'
-        }
+        self.debug = True
+        self.dummy_outputs = dummy.recipes
         self.tokenizer = None
         self.generator = None
+        self.api_ids = []
+        self.api_keys = []
+        self.api_test = 2
         self.task = "text2text-generation"
         self.model_name_or_path = "flax-community/t5-recipe-generation"
-        self.frames = list(glob.glob("asset/images/frames/*.jpg"))
+        self.color_frame = "#ffffff"
+        self.main_frame = "asset/frame/recipe-bg.png"
+        self.no_food = "asset/frame/no_food.png"
+        self.logo_frame = "asset/frame/logo.png"
+        self.chef_frames = {
+            "scheherazade": "asset/frame/food-image-logo-bg-s.png",
+            "giovanni": "asset/frame/food-image-logo-bg-g.png",
+        }
         self.fonts = {
-            "title": ImageFont.truetype("asset/fonts/Poppins-Bold.ttf", 90),
+            "title": ImageFont.truetype("asset/fonts/Poppins-Bold.ttf", 70),
             "sub_title": ImageFont.truetype("asset/fonts/Poppins-Medium.ttf", 30),
-            "body_bold": ImageFont.truetype("asset/fonts/Montserrat-Bold.ttf", 50),
-            "body": ImageFont.truetype("asset/fonts/Montserrat-Regular.ttf", 30),
+            "body_bold": ImageFont.truetype("asset/fonts/Montserrat-Bold.ttf", 22),
+            "body": ImageFont.truetype("asset/fonts/Montserrat-Regular.ttf", 18),
 
         }
-        self.list_division = 5
-        self.point = "•"
         set_seed(42)
 
     def _skip_special_tokens_and_prettify(self, text):
@@ -109,81 +180,43 @@ class TextGeneration:
 
         return data
 
+    def load_pipeline(self):
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
+        self.generator = pipeline(self.task, model=self.model_name_or_path, tokenizer=self.model_name_or_path)
+
+    def load_api(self):
+        app_ids = os.getenv("EDAMAM_APP_ID")
+        app_ids = app_ids.split(",") if app_ids else []
+        app_keys = os.getenv("EDAMAM_APP_KEY")
+        app_keys = app_keys.split(",") if app_keys else []
+
+        if len(app_ids) != len(app_keys):
+            self.api_ids = []
+            self.api_keys = []
+
+        self.api_ids = app_ids
+        self.api_keys = app_keys
+
     def load(self):
+        self.load_api()
         if not self.debug:
-            self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
-            self.generator = pipeline(self.task, model=self.model_name_or_path, tokenizer=self.model_name_or_path)
-
-    def prepare_frame(self, recipe, frame):
-        im_editable = ImageDraw.Draw(frame)
-        # Info
-        im_editable.text(
-            (1570, 180),
-            datetime.now().strftime("%Y-%m-%d %H:%M"),
-            (61, 61, 70),
-            font=self.fonts["sub_title"],
-        )
-        im_editable.text(
-            (1570, 170 + 70),
-            "By " + recipe["by"],
-            (61, 61, 70),
-            font=self.fonts["sub_title"],
+            self.load_pipeline()
+
+    def prepare_frame(self, recipe, chef_name):
+        frame_path = self.chef_frames[chef_name.lower()]
+        food_logo = generate_food_with_logo_image(frame_path, self.logo_frame, recipe["image"])
+        frame = generate_recipe_image(
+            recipe,
+            self.main_frame,
+            food_logo,
+            self.fonts,
+            bg_color="#ffffff"
         )
-        # Title
-        im_editable.text(
-            (400, 650),
-            textwrap.fill(recipe["title"], 30),
-            (61, 61, 70),
-            font=self.fonts["title"],
-        )
-        # Ingredients
-        im_editable.text(
-            (100, 1000),
-            "Ingredients",
-            (61, 61, 70),
-            font=self.fonts["body_bold"],
-        )
-        ingredients = recipe["ingredients"]
-        ingredients = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in ingredients]
-
-        im_editable.text(
-            (100, 1080),
-            "\n".join([f"{self.point} {item}" for item in ingredients]),
-            (61, 61, 70),
-            font=self.fonts["body"],
-        )
-        # Directions
-        im_editable.text(
-            (700, 1000),
-            "Directions",
-            (61, 61, 70),
-            font=self.fonts["body_bold"],
-        )
-        directions = recipe["directions"]
-        directions = [textwrap.fill(item, 80).replace("\n", "\n   ") for item in directions]
-        im_editable.text(
-            (700, 1080),
-            "\n".join([f"{i + 1}. {item}" for i, item in enumerate(directions)]).strip(),
-            (61, 61, 70),
-            font=self.fonts["body"],
-        )
-        # directions_col1 = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in directions[:self.list_division]]
-        # directions_col2 = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in directions[self.list_division:]]
-        # im_editable.text(
-        #     (700, 1130),
-        #     "\n".join([f"{i + 1}. {item}" for i, item in enumerate(directions_col1)]).strip(),
-        #     (61, 61, 70),
-        #     font=self.fonts["body"],
-        # )
-        # im_editable.text(
-        #     (1300, 1130),
-        #     "\n".join([f"{i + 1 + self.list_division}. {item}" for i, item in enumerate(directions_col2)]).strip(),
-        #     (61, 61, 70),
-        #     font=self.fonts["body"],
-        # )
         return frame
 
     def generate(self, items, generation_kwargs):
+        recipe = self.dummy_outputs[random.randint(0, len(self.dummy_outputs) - 1)]
+
         if not self.debug:
             generation_kwargs["num_return_sequences"] = 1
             # generation_kwargs["return_full_text"] = False
@@ -196,13 +229,25 @@ class TextGeneration:
             )[0]["generated_token_ids"]
             recipe = self.tokenizer.decode(generated_ids, skip_special_tokens=False)
             recipe = self._skip_special_tokens_and_prettify(recipe)
-            return recipe
 
-        return self.dummy_output
+        if self.api_ids and self.api_keys and len(self.api_ids) == len(self.api_keys):
+            test = 0
+            for i in range(len(self.api_keys)):
+                if test > self.api_test:
+                    recipe["image"] = None
+                    break
+                image = generate_cook_image(recipe["title"].lower(), self.api_ids[i], self.api_keys[i])
+                test += 1
+                if image:
+                    recipe["image"] = image
+                    break
+        else:
+            recipe["image"] = None
+
+        return recipe
 
-    def generate_frame(self, recipe, frame_path):
-        frame = load_image(frame_path)
-        return self.prepare_frame(recipe, frame)
+    def generate_frame(self, recipe, chef_name):
+        return self.prepare_frame(recipe, chef_name)
 
 
 @st.cache(allow_output_mutation=True)
@@ -240,17 +285,17 @@ def main():
         initial_sidebar_state="expanded"
     )
     generator = load_text_generator()
-    if hasattr(st, "session_state"):
-        if 'get_random_frame' not in st.session_state:
-            st.session_state.get_random_frame = generator.frames[0]
-    else:
-        get_random_frame = generator.frames[0]
+    # if hasattr(st, "session_state"):
+    #     if 'get_random_frame' not in st.session_state:
+    #         st.session_state.get_random_frame = generator.frames[0]
+    # else:
+    #     get_random_frame = generator.frames[0]
 
     local_css("asset/css/style.css")
 
-    col1, col2 = st.beta_columns([5, 3])
+    col1, col2 = st.beta_columns([4, 3])
     with col2:
-        st.image(load_image("asset/images/chef-transformer-transparent.png"), width=300)
+        st.image(load_image_from_local("asset/images/chef-transformer-transparent.png"), width=300)
         st.markdown(meta.SIDEBAR_INFO, unsafe_allow_html=True)
 
         with st.beta_expander("Where did this story start?"):
@@ -288,10 +333,10 @@ def main():
         unsafe_allow_html=True
     )
     if recipe_button:
-        if hasattr(st, "session_state"):
-            st.session_state.get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
-        else:
-            get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
+        # if hasattr(st, "session_state"):
+        #     st.session_state.get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
+        # else:
+        #     get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
 
         entered_items.markdown("**Generate recipe for:** " + items)
         with st.spinner("Generating recipe..."):
@@ -306,6 +351,9 @@ def main():
                 generated_recipe = generator.generate(items, gen_kw)
 
                 title = generated_recipe["title"]
+                food_image = generated_recipe["image"]
+                food_image = load_image_from_url(food_image, rgba_mode=True, default_image=generator.no_food)
+                food_image = image_to_base64(food_image)
                 ingredients = generated_recipe["ingredients"]
                 directions = [textwrap.fill(item, 70).replace("\n", "\n   ") for item in
                               generated_recipe["directions"]]
@@ -315,15 +363,17 @@ def main():
 
                 with r1:
                     # st.write(st.session_state.get_random_frame)
-                    if hasattr(st, "session_state"):
-                        recipe_post = generator.generate_frame(generated_recipe, st.session_state.get_random_frame)
-                    else:
-                        recipe_post = generator.generate_frame(generated_recipe, get_random_frame)
+                    # if hasattr(st, "session_state"):
+                    #     recipe_post = generator.generate_frame(generated_recipe, st.session_state.get_random_frame)
+                    # else:
+                    #     recipe_post = generator.generate_frame(generated_recipe, get_random_frame)
+
+                    recipe_post = generator.generate_frame(generated_recipe, chef.split()[-1])
 
                     st.image(
                         recipe_post,
                         # width=500,
-                        caption="Click 🔎 to enlarge",
+                        caption="Save image and share on your social media",
                         use_column_width="auto",
                         output_format="PNG"
                     )
@@ -332,7 +382,11 @@ def main():
                     st.markdown(
                         " ".join([
                             "<div class='r-text-recipe'>",
+                            "<div class='food-title'>",
+                            f"<img src='{food_image}' />",
                             f"<h2>{title}</h2>",
+                            "</div>",
+                            '<div class="divider"><div class="divider-mask"></div></div>',
                             "<h3>Ingredients</h3>",
                             "<ul class='ingredients-list'>",
                             " ".join([f'<li>{item}</li>' for item in ingredients]),

asset/css/style.css

@@ -35,14 +35,46 @@ body {
     text-decoration: underline;
 }
 
+.story-box {
+    overflow-y: scroll;
+    max-height: 300px;
+}
+
 .r-text-recipe {
     padding-left: 30px;
     margin-left: 10px;
     border-left: 1px dashed #eee;
 }
-.r-text-recipe > h2 {
-    margin-top: 0;
-    padding-top: 0;
+
+.divider {
+    margin: 5px auto;
+    width: 400px;
+    max-width: 100%;
+    position:relative;
+}
+
+.divider-mask {
+    overflow: hidden;
+    height: 20px;
+}
+
+.divider-mask:after {
+    content: '';
+    display: block;
+    margin: 0 auto;
+    width: 170px;
+    height: 0px;
+    border-bottom: 2px solid #e9a726;
+    border-radius: 10px;
+}
+
+.r-text-recipe .food-title {
+    text-align: center;
+}
+.r-text-recipe .food-title img {
+    max-width: 120px;
+}
+.r-text-recipe .food-title h2 {
 }
 .ingredients-list {
     columns: 2;

dummy.py

@@ -0,0 +1,85 @@
+recipes = [
+    {
+        'directions': [
+            "preheat oven to 350.",
+            "grease a 13 x 9 x 2 inch baking pan.",
+            "place 1 sheet of the dough on a work surface.",
+            "brush with melted butter.",
+            "top with another sheet of dough and brush with butter. repeat with 3 more sheets of dough, brushing each sheet with butter between each layer.",
+            "combine walnuts and cinnamon in a small bowl.",
+            "sprinkle half of the walnut mixture over the dough.",
+            "repeat with the remaining dough and walnut mixture.",
+            "starting at the long side, roll up the dough into a log.",
+            "cut the log into 1 inch slices.",
+            "arrange the slices, cut side down, in the prepared baking dish.",
+            "bake until golden brown, about 30 minutes.",
+            "meanwhile, bring water and honey to a boil.",
+            "remove from heat and stir in chocolate until melted.",
+            "pour over the cooled strudel.",
+            "serve warm or at room temperature.",
+            "makes 12 servings.",
+        ],
+        'ingredients': [
+            "1 lb. phyllo dough, thawed",
+            "1 c. unsalted butter, melted",
+            "2 c chopped walnuts",
+            "1/2 tsp. cinnamon",
+            "1 1/2 c water",
+            "3/4 c honey",
+            "1/4 c melted chocolate",
+        ],
+        'title': 'Baklava'
+    },
+    {
+        'directions': [
+            "in a large skillet, heat oil to 375.",
+            "season chops on both sides with salt and pepper.",
+            "dredge chops in flour, shaking off excess.",
+            "dip chops into eggs, then coat with breadcrumbs.",
+            "fry chops until golden brown, about 3 minutes per side.",
+            "transfer to a platter and keep warm.",
+            "pour off all but 1 tablespoon of fat from skillet.",
+            "add gravy and cook over medium heat, stirring occasionally, until thickened, about 5 minutes.",
+            "spoon gravy over chops and sprinkle with parsley.",
+            "makes 4 servings.",
+        ],
+        'ingredients': [
+            "1 lb beef, cubed",
+            "2 tablespoons oil",
+            "1 large onion, chopped",
+            "2 medium tomatoes, peeled and chopped",
+            "1 teaspoon turmeric powder",
+            "2 limes, juice of",
+            "1 cup water",
+            "salt",
+            "pepper",
+            "1 15 ounce can red beans, drained and rinsed",
+            "1 tablespoon dried herb",
+        ],
+        'title': 'Beef And Red Beans'
+    },
+    {
+        'directions': [
+            "heat the oil in a large saucepan.",
+            "add the onion and saute until golden brown.",
+            "stir in the tomatoes, turmeric powder, lime juice, water, salt and pepper.",
+            "bring to a boil.",
+            "reduce the heat and simmer for 10 minutes.",
+            "mix in the beef and beans.",
+            "cover and cook over low heat for 30 minutes or until the beef is tender.",
+            "garnish with the dried herbs.",
+        ],
+        'ingredients': [
+            "oil for frying",
+            "4 6 oz. boneless pork loin chops",
+            "salt",
+            "pepper",
+            "1/2 c. flour",
+            "2 eggs, lightly beaten",
+            "2 tbsp. dry breadcrumbs",
+            "gravy",
+            "chopped parsley",
+        ],
+        'title': 'Breaded Pork Chops With Gravy'
+    }
+]

meta.py

@@ -15,26 +15,40 @@ Chef Giovanni. Scheherazade is known for being more creative whereas Giovanni is
 """.strip()
 PROMPT_BOX = "Add custom ingredients here (separated by `,`): "
 STORY = """
-Hello everyone 👋, I am **Chef Transformer**, the owner of this restaurant. I was made by a group of [NLP Engineers](
-https://huggingface.co/flax-community/t5-recipe-generation#team-members) to train my two prodigy recipe creators: **Chef Scheherazade** and **Chef Giovanni**. Both of my students participated in my rigorous culinary program, [T5 fine-tune training](https://huggingface.co/flax-community/t5-recipe-generation), to learn how to prepare exquisite cuisines from a wide variety of ingredients. 
-I've never been more proud of my students -- both can produce exceptional dishes but I regard Scheherazade as being *creative* while Giovanni is *meticulous*. If you give each of them the same ingredients, they'll usually come up with something different.
-At the start of the program the chefs read cookbooks containing thousands of recipes of varying difficulties and from cultures all over the world. The NLP engineers helped guide the learning process so that the chefs could actually learn which ingredients work well together rather than just memorize recipes.
+<div class="story-box">
+<p>
+Hello everyone 👋, I am <strong>Chef Transformer</strong>, 
+the owner of this restaurant. I was made by a group of <a href="https://huggingface.co/flax-community/t5-recipe-generation#team-members">NLP Engineers</a> to train my two prodigy recipe creators: <strong>Chef Scheherazade</strong> and <strong>Chef Giovanni</strong>. 
+Both of my students participated in my rigorous culinary program, <a href="https://huggingface.co/flax-community/t5-recipe-generation">T5 fine-tune training</a>, 
+to learn how to prepare exquisite cuisines from a wide variety of ingredients. 
+I've never been more proud of my students -- both can produce exceptional dishes but I regard Scheherazade as being <em>creative</em> while Giovanni is <em>meticulous</em>. 
+If you give each of them the same ingredients, they'll usually come up with something different. <br /><br />
+At the start of the program the chefs read cookbooks containing thousands of recipes of varying difficulties and from cultures all over the world. 
+The NLP engineers helped guide the learning process so that the chefs could actually learn which ingredients work well together rather than just memorize recipes. 
 I trained my chefs by asking them to generate a title, a list of ingredients (including amounts!), and a list of directions after giving them just a simple list of food items. 
-```text
+</p>
+
+<pre>
 [Input]
     {food items*: separated by comma}
      
 [Targets]
-    title: {TITLE} <section> 
-    ingredients: {INGREDIENTS: separated by <sep>} <section> 
+    title: {TITLE} <section>
+    ingredients: {INGREDIENTS: separated by <sep>} <section>
     directions: {DIRECTIONS: separated by <sep>}.
-```
-
-*In the cookbooks (a.k.a [dataset](https://huggingface.co/datasets/recipe_nlg)), the food items were referred to as 
-NER.*
+</pre>
 
+<p>
+  <em>In the cookbooks (a.k.a <a href="https://huggingface.co/datasets/recipe_nlg">dataset</a>), the food items were referred to as NER. </em>
+</p>
+<p>
+In the span of a week, my chefs went from spitting out nonsense to creating masterpieces. 
+Their learning rate was exceptionally high and each batch of recipes was better than the last. <br />
+In their final exam, they achieved <a href="https://huggingface.co/flax-community/t5-recipe-generation#evaluation">high scores</a> 💯 in a 
+standardized industry test and established this restaurant 🍲. Please tell your friends and family about us! 
+We create each recipe with a smile on our faces 🤗 Everyone at the restaurant is grateful for the generous support of 
+HuggingFace and Google for hosting Flax Community week. 
+</p>
 
-In the span of a week, my chefs went from spitting out nonsense to creating masterpieces. Their learning rate was exceptionally high and each batch of recipes was better than the last. In their final exam, they achieved [high scores](
-https://huggingface.co/flax-community/t5-recipe-generation#evaluation) 💯 in a standardized industry test and established this restaurant 🍲. Please tell your friends and family about us! We create each recipe with a smile on our faces 🤗
-Everyone at the restaurant is grateful for the generous support of Hugging Face and Google for hosting Flax Community week.
+</div>
 """.strip()

requirements.txt

@@ -1,4 +1,5 @@
 streamlit==0.84.1
 transformers
 torch
-Pillow
+Pillow
+requests

utils.py

@@ -1,15 +1,44 @@
 import streamlit as st
 import json
 from PIL import Image
+from io import BytesIO
+import base64
+import requests
 
 
-def load_image(image_path, image_resize=None):
+def load_image_from_local(image_path, image_resize=None):
     image = Image.open(image_path)
+
     if isinstance(image_resize, tuple):
-        image.resize(image_resize)
+        image = image.resize(image_resize)
+    return image
+
+
+def load_image_from_url(image_url, rgba_mode=False, image_resize=None, default_image=None):
+    try:
+        image = Image.open(requests.get(image_url, stream=True).raw)
+
+        if rgba_mode:
+            image = image.convert("RGBA")
+
+        if isinstance(image_resize, tuple):
+            image = image.resize(image_resize)
+
+    except Exception as e:
+        image = None
+        if default_image:
+            image = load_image_from_local(default_image, image_resize=image_resize)
+
     return image
 
 
+def image_to_base64(image_array):
+    buffered = BytesIO()
+    image_array.save(buffered, format="PNG")
+    image_b64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return f"data:image/png;base64, {image_b64}"
+
+
 def load_text(text_path):
     text = ''
     with open(text_path) as f: