what is next

inferencer/strategy/__init__.py

@@ -1,3 +1,4 @@
+from .what_is_next import what_is_next
 from .what_is_dish import what_is_dish
 from .what_is_step_img_doing import what_is_step_img_doing
 from .what_are_step_imgs_doing import what_are_step_imgs_doing
@@ -12,4 +13,5 @@
     'what_are_components_nested': what_are_components_nested,
     'what_are_components_flat': what_are_components_flat,
     'how_to_sort_step_imgs': how_to_sort_step_imgs,
+    'what_is_next': what_is_next,
 }

inferencer/strategy/what_is_next.py

@@ -0,0 +1,45 @@
+import os
+from typing import List, Any, Dict
+
+from utils import download_img, make_data_dict, ID_COUNTER, LOGGER
+from configs import IMG_SAVE_PATH
+
+def what_is_next(data: Dict[str, Any], **kwargs) -> List[Any]:
+    results = []
+    for idx, cur_step in enumerate(data['steps']):
+        img_file = os.path.join(IMG_SAVE_PATH, f"{data['id']}_{str(ID_COUNTER)}_{what_is_next.__name__}_{idx}.jpg")
+
+        if os.path.isfile(img_file):
+            LOGGER.warning(f'img has been downloaded in {what_is_next.__name__}: [{img_file}]')
+
+        if not download_img(
+            cur_step['img'],
+            img_file
+            ):
+            LOGGER.debug(f"img download failed, url: [{cur_step['img']}]")
+            continue
+
+        if idx == len(data['steps'])-1:
+            results.append(
+                make_data_dict(
+                    cur_id=str(ID_COUNTER),
+                    cur_conversations=[
+                        f"图：<img>{img_file}</img>，接下来做什么？",
+                        f"已经完成了"
+                    ]
+                )
+            )
+
+        else:
+            results.append(
+                make_data_dict(
+                    cur_id=str(ID_COUNTER),
+                    cur_conversations=[
+                        f"图：<img>{img_file}</img>，接下来做什么？",
+                        f"接下来应该" + ''.join(f"{data['steps'][i]['description']}"[1:] for i in
+                                                range(idx+1, len(data['steps'])))
+                    ]
+                )
+            )
+        ID_COUNTER.increment()
+    return results