From 17f59fdaac2bf6e10fbd833db84c3be3a367538f Mon Sep 17 00:00:00 2001 From: jableader Date: Mon, 8 Jan 2024 11:45:22 +1100 Subject: [PATCH] Ingredient parsing --- main.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index a77f074..5db419d 100644 --- a/main.py +++ b/main.py @@ -1,8 +1,9 @@ +import sqlite3 +import product, db, recipe + +from typing import List from fastapi import FastAPI, Depends, HTTPException from fastapi.middleware.cors import CORSMiddleware -import sqlite3 -from recipe import parse_recipe -import product, db app = FastAPI() @@ -25,7 +26,13 @@ async def get_db(): @app.get("/recipes/parse") async def parse_recipe_handler(url: str, conn: sqlite3.Connection = Depends(get_db)): - return await parse_recipe(conn, url) + return await recipe.parse_recipe(conn, url) + +@app.get("/recipes/ingredients/parse") +async def parse_ingredients(lines: List[str], conn: sqlite3.Connection = Depends(get_db)): + ingredients = recipe.parse_ingredient_from_nlp(lines) + recipe.match_existing_products(conn, ingredients) + return ingredients @app.post("/product/") async def create_product(url: str, conn: sqlite3.Connection = Depends(get_db)):