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)):