Transfer from git

app.py

@@ -5,11 +5,8 @@ from transformers import pipeline, set_seed
 from transformers import AutoTokenizer
 
 from PIL import (
-    Image,
     ImageFont,
-    ImageDraw
 )
-import requests
 
 import os
 import re
@@ -124,7 +121,7 @@ class TextGeneration:
 
     def generate(self, items, generation_kwargs):
         recipe = self.dummy_outputs[0]
-        recipe = self.dummy_outputs[random.randint(0, len(self.dummy_outputs) - 1)]
+        # recipe = self.dummy_outputs[random.randint(0, len(self.dummy_outputs) - 1)]
 
         if not self.debug:
             generation_kwargs["num_return_sequences"] = 1
@@ -208,7 +205,7 @@ def main():
         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?"):
+        with st.beta_expander("Where did this story start?", expanded=True):
             st.markdown(meta.STORY, unsafe_allow_html=True)
 
     with col1:
@@ -270,10 +267,10 @@ def main():
                     generated_recipe["ingredients"],
                     pure_comma_separation(items, return_list=True)
                 )
+                # ingredients = [textwrap.fill(item, 10).replace("\n", "<br />   ") for item in ingredients]
 
-                directions = [textwrap.fill(item, 70).replace("\n", "\n   ") for item in
-                              generated_recipe["directions"]]
-                directions = ext.directions(directions)
+                directions = ext.directions(generated_recipe["directions"])
+                # directions = [textwrap.fill(item, 70).replace("\n", "<br />   ") for item in directions]
 
                 generated_recipe["by"] = chef
 

dummy.py

@@ -20,7 +20,7 @@ recipes = [
             "makes 12 servings.",
         ],
         'ingredients': [
-            "1 lb. phyllo dough, thawed",
+            "1 lb. phyllo dough, thawed 1 lb. phyllo dough, thawed",
             "1 c. unsalted butter, melted",
             "2 c chopped walnuts",
             "1/2 tsp. cinnamon",

requirements.txt

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

utils/draw.py

@@ -1,16 +1,22 @@
 from PIL import (
     Image,
-    ImageFont,
     ImageDraw
 )
 import textwrap
-from .utils import load_image_from_url
-from .ext import (
+from utils.utils import load_image_from_url
+from utils.ext import (
     ingredients as ext_ingredients,
     directions as ext_directions
 )
 
 
+# from .utils import load_image_from_url
+# from .ext import (
+#     ingredients as ext_ingredients,
+#     directions as ext_directions
+# )
+
+
 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
@@ -58,8 +64,8 @@ def generate_recipe_image(
         font=fonts["body_bold"],
     )
     ingredients = recipe_data["ingredients"]
-    ingredients = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in ingredients]
     ingredients = ext_ingredients(ingredients, [], without_mapping=True)
+    ingredients = [textwrap.fill(item, 30).replace("\n", "\n   ") for item in ingredients]
 
     im_editable.text(
         (50, 520),
@@ -76,7 +82,9 @@ def generate_recipe_image(
     )
 
     directions = recipe_data["directions"]
-    directions = [textwrap.fill(item, 70).replace("\n", "\n   ").capitalize() for item in directions]
+    directions = ext_directions(directions)
+    directions = [textwrap.fill(item, 70).replace("\n", "\n    ").capitalize() for item in directions]
+
     im_editable.text(
         (430, 520),
         "\n".join([f"{i + 1}. {item}" for i, item in enumerate(directions)]).strip(),

utils/ext.py

@@ -1,5 +1,6 @@
 import re
-from .utils import replace_regex
+from utils.utils import replace_regex
+# from .utils import replace_regex
 
 DEFAULT_MAP_DICT = {
     " c ": " c. ",