Spaces:
Runtime error
Runtime error
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() |