victorialslocum's picture
refactor
1e794df
raw
history blame
No virus
4.05 kB
from dis import Instruction
import requests
from decouple import config
import gradio as gr
import spacy
import re
# for generating requirements.txt, use:
# pip install pipreqs
# pipreqs ./ --force
RAPIDAPI_KEY = config('RAPIDAPI_KEY')
ner = gr.Blocks.load(name='models/victorialslocum/en_reciparse_model')
def get_recipe_data(recipe_url):
rapid_api_url = "https://mycookbook-io1.p.rapidapi.com/recipes/rapidapi"
headers = {
"content-type": "text/plain",
"X-RapidAPI-Host": "mycookbook-io1.p.rapidapi.com",
"X-RapidAPI-Key": RAPIDAPI_KEY
}
response = requests.request("POST", rapid_api_url,
data=recipe_url, headers=headers).json()
# print(response)
instructions_list = response[0]['instructions'][0]['steps']
ingredients_list = response[0]['ingredients']
recipe_title = response[0]['name']
instructions = ""
ingredients = ""
for i in range(len(instructions_list)):
instructions += f'{i+1}. {instructions_list[i]} \n'
for i in range(len(ingredients_list)):
ingredients += f'{i+1}. {ingredients_list[i]} \n'
return instructions, ingredients, recipe_title
def get_data(data):
nlp = spacy.load("en_core_web_sm")
doc = ner(data)
ingredients_list = []
for item in doc:
if item[1] == 'INGREDIENT':
item_lemma = ' ' .join([tok.lemma_ for tok in nlp(item[0])])
if item_lemma not in ingredients_list:
ingredients_list.append(item_lemma)
ingredients_out = ""
for i in range(len(ingredients_list)):
ingredients_out += f'{i+1}. {ingredients_list[i]} \n'
return ingredients_out
def final_ingredients(inst_ingredients_list, ingr_ingredients_list):
instructions_guess = [re.sub(r'[0-9]+', '', item)[2:-1] for item in inst_ingredients_list.split('\n')if len(item) > 3]
ingredients_guess = [re.sub(r'[0-9]+', '', item)[2:-1] for item in ingr_ingredients_list.split('\n')if len(item) > 3]
final_guess_list = []
for ingr_item in ingredients_guess:
for inst_item in instructions_guess:
if inst_item in ingr_item and ingr_item not in final_guess_list:
final_guess_list.append(ingr_item)
final_guess = ""
for i in range(len(final_guess_list)):
final_guess += f'{i+1}. {final_guess_list[i]} \n'
return final_guess
with gr.Blocks() as demo:
gr.Markdown("# Reciparse Visualizer")
gr.Markdown("### Powered by spaCy and Gradio")
with gr.Box():
ingredients_list = gr.Variable()
with gr.Column():
gr.Markdown("## Recipe Info")
with gr.Row():
recipe_link = gr.Textbox(label="Recipe link", placeholder="Input your recipe link")
recipe_title = gr.Textbox(label="Recipe title")
with gr.Row():
instructions = gr.Textbox(label="Instructions")
ingredients = gr.Textbox(label="Ingredients")
recipe_btn = gr.Button("Get recipe info")
with gr.Box():
with gr.Column():
gr.Markdown("## Ingredient Info")
ingredient_btn = gr.Button("Get the list of ingredients")
final_btn = gr.Button("Create the final list")
with gr.Row():
ingr_ingredients_list = gr.Textbox(label="Ingredients in recipe based off of ingredients list")
inst_ingredients_list = gr.Textbox(label="Ingredients in recipe based off of instructions")
final_ingredients_list = gr.Textbox(label="Final ingredient guess")
recipe_btn.click(fn=get_recipe_data, inputs=recipe_link, outputs=[instructions, ingredients, recipe_title])
ingredient_btn.click(fn=get_data, inputs=[ingredients], outputs=[ingr_ingredients_list])
ingredient_btn.click(fn=get_data, inputs=[instructions], outputs=[inst_ingredients_list])
final_btn.click(fn=final_ingredients, inputs=[inst_ingredients_list, ingr_ingredients_list], outputs=[final_ingredients_list])
demo.launch()