Default to 1 items

This commit is contained in:
jableader 2024-01-13 16:48:09 +11:00
parent df7a41e028
commit bff5607381

View file

@ -18,13 +18,13 @@ def parse_ingredient_from_nlp(ingredients: List[str]) -> Ingredient:
for ingredient in parse_multiple_ingredients(ingredients): for ingredient in parse_multiple_ingredients(ingredients):
name = ingredient.name.text if ingredient.name else '' name = ingredient.name.text if ingredient.name else ''
quantity, unit = 0, units.ITEMS.name quantity, unit = 1, units.ITEMS.name
if ingredient.amount: if ingredient.amount:
amount = ingredient.amount[0] amount = ingredient.amount[0]
real_unit = units.get_unit(unit) real_unit = units.get_unit(unit)
unit = real_unit.name if real_unit else units.ITEMS.name unit = real_unit.name if real_unit else units.ITEMS.name
quantity = amount.quantity if amount.quantity else 0 quantity = amount.quantity if amount.quantity else 1
results.append(Ingredient(id=0, results.append(Ingredient(id=0,
line=ingredient.sentence, line=ingredient.sentence,