|
|
|
import os |
|
import json |
|
|
|
pathToSettings = '../../env/ai.json' |
|
if os.path.exists(pathToSettings): |
|
|
|
print(f'Reading settings from {pathToSettings}') |
|
f = open(pathToSettings) |
|
settingsJson = json.load(f) |
|
del f |
|
|
|
for key in settingsJson: |
|
os.environ[key] = settingsJson[key] |
|
|
|
del settingsJson |
|
|
|
|
|
|
|
|
|
|
|
import requests |
|
from bs4 import BeautifulSoup |
|
import json |
|
import validators |
|
|
|
pageCache = {} |
|
|
|
def json_safe_loads(jsonString): |
|
try: |
|
obj = json.loads(jsonString) |
|
except ValueError as e: |
|
return None |
|
return obj |
|
|
|
def remove_keys(dictionary: dict, keyList: list): |
|
for key in keyList: |
|
if key in dictionary: |
|
del dictionary[key] |
|
|
|
def get_recipe_as_json(url: str) -> dict: |
|
|
|
obj = json_safe_loads(url) |
|
if obj is not None: |
|
if 'url' in obj: |
|
url = obj['url'] |
|
|
|
url = url.replace("'", "").replace('"', '') |
|
|
|
if not validators.url(url): |
|
return {'error': 'Invalid url format', 'url': url } |
|
|
|
if url in pageCache: |
|
return pageCache[url] |
|
|
|
html = requests.get(url).text |
|
soup = BeautifulSoup(html) |
|
script = soup.find_all("script", {"id": "allrecipes-schema_1-0"}) |
|
|
|
recipeDict = json.loads(script[0].text)[0] |
|
remove_keys(recipeDict, ['review', 'image', 'mainEntityOfPage', 'publisher']) |
|
|
|
pageCache[url] = recipeDict |
|
|
|
return recipeDict |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
x = json.loads('{ "url": "https://www.allrecipes.com/recipe/235153/easy-baked-chicken-thighs/"}') |
|
|
|
print(type(x)) |
|
|
|
|
|
|
|
|
|
|
|
dessertList = [ |
|
"https://www.allrecipes.com/chocolate-snack-cake-recipe-8350343", |
|
"https://www.allrecipes.com/charred-spiced-pears-with-smoky-vanilla-cherry-sauce-recipe-8347080", |
|
"https://www.allrecipes.com/meringue-topped-banana-pudding-recipe-8347040", |
|
"https://www.allrecipes.com/white-chocolate-cinnamon-toast-crunch-bars-recipe-7556790", |
|
"https://www.allrecipes.com/plum-cobbler-for-two-recipe-8304143", |
|
"https://www.allrecipes.com/pumpkin-cheesecake-cookies-recipe-7972485", |
|
"https://www.allrecipes.com/chocolate-whipped-cottage-cheese-recipe-8303272", |
|
"https://www.allrecipes.com/nutella-ice-cream-recipe-7508716", |
|
"https://www.allrecipes.com/3-ingredient-banana-oatmeal-cookies-recipe-7972686", |
|
"https://www.allrecipes.com/caramel-apple-pie-cookies-recipe-7642173" |
|
] |
|
|
|
chickenDishList = [ |
|
"https://www.allrecipes.com/recipe/228363/crispy-roasted-chicken/", |
|
"https://www.allrecipes.com/recipe/254877/roasted-spatchcocked-chicken-with-potatoes/", |
|
"https://www.allrecipes.com/recipe/235153/easy-baked-chicken-thighs/", |
|
"https://www.allrecipes.com/recipe/258878/crispy-baked-chicken-thighs/", |
|
"https://www.allrecipes.com/recipe/235151/crispy-and-tender-baked-chicken-thighs/", |
|
"https://www.allrecipes.com/recipe/233953/million-dollar-chicken/", |
|
"https://www.allrecipes.com/recipe/70679/simple-whole-roasted-chicken/", |
|
"https://www.allrecipes.com/recipe/214618/beer-can-chicken/", |
|
"https://www.allrecipes.com/recipe/272858/air-fryer-chicken-thighs/", |
|
"https://www.allrecipes.com/recipe/214478/happy-roast-chicken/" |
|
] |
|
|
|
|
|
|
|
|
|
|
|
|
|
from langchain.agents import Tool |
|
|
|
|
|
|
|
def list_chicken_recipes(query: str): |
|
return chickenDishList |
|
|
|
list_chicken_recipes_tool = Tool(name='Chicken Recipes tool', func= list_chicken_recipes, description="This tools lists the available Chicken Recipes it returns a list of recipes with a TITLE and a URL where you can fetch the recipe.") |
|
|
|
|
|
def list_dessert_recipes(query: str): |
|
return dessertList |
|
|
|
list_dessert_recipes_tool = Tool(name='Dessert Recipes tool', func=list_dessert_recipes, |
|
description="This tools lists the available Dessert Recipes it returns a list of recipes with a TITLE and a URL where you can fetch the recipe.") |
|
|
|
|
|
def get_recipe(url: str): |
|
return get_recipe_as_json(url) |
|
|
|
get_recipe_as_json_tool = Tool(name='Get a Recipe tool', func=get_recipe, description=""" |
|
Useful for fetching a particular recipe, you can only fetch a recipe with it's url, you must get that using another tool |
|
The tool uses the https://schema.org/Recipe format to store it's recipes. |
|
""") |
|
|
|
|
|
tools = [list_chicken_recipes_tool, list_dessert_recipes_tool, get_recipe_as_json_tool] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from langchain.agents import load_tools |
|
from langchain.agents import initialize_agent |
|
from langchain.llms import OpenAI |
|
from langchain.chat_models import ChatOpenAI |
|
from langchain.load.dump import dumps |
|
|
|
def ask_query(query): |
|
|
|
|
|
llm = ChatOpenAI(temperature=0.2, model_name='gpt-4') |
|
agent = initialize_agent(agent="zero-shot-react-description", tools=tools, llm=llm, verbose=True, max_iterations=7, return_intermediate_steps=True) |
|
system = """ |
|
If the answer is not in the tools or context passed to you then don't answer. \n |
|
If you don't know the answer then say so. \n |
|
""" |
|
response = agent({"input": f"{system} [[RECIPENAME]] {query}"}) |
|
|
|
|
|
stepsDict = json.loads(dumps(response["intermediate_steps"], pretty=True)) |
|
resp = 'Below are the steps the agent took to get to the Final Answer. \n"Thought" is the LLMs internal dialogue, \n"Action" is the tool it will use to fetch the next piece of information. \n"Action Input" is the input it passes the tool to fetch this information. \n"Action Response" is what was returned from the tool to the LLM at that given step. ' |
|
resp += '\n\n' |
|
resp += 'Steps to solve answer using ReAct\n' |
|
for i in range(len(stepsDict)): |
|
resp += '##########################################\n' |
|
resp += f'Step: {i+1} of {len(stepsDict)}\n' |
|
resp += f"Thought: {stepsDict[i][0]['kwargs']['log']}\n" |
|
resp += 'Below is what the tool returned...\n' |
|
resp += f"Action response: {stepsDict[i][1]}\n" |
|
resp += '\n' |
|
|
|
resp += '\nFinal Thought:\n' |
|
resp += response['output'] |
|
return resp |
|
|
|
|
|
|
|
|
|
|
|
import gradio as gr |
|
|
|
with gr.Blocks() as demo: |
|
gr.Markdown("Start typing below and then click **Run** to see the output.") |
|
with gr.Row(): |
|
with gr.Column(): |
|
inp = gr.Textbox(placeholder="Type your question here?", label="Question") |
|
btn = gr.Button("Run") |
|
examples = ["Can you show tell me what ingredients I need for the first baked chicken recipe?", |
|
"I have 15 guests coming to visit, if I make the 'Chocolate Snack Cake' recipe will there be enough? ", |
|
"Can you show tell me what ingredients I need for the last baked chicken recipe? ", |
|
"What is the best baked chicken recipe? Please look across all recipes with the word 'baked' in the title", |
|
"Is there a Chicken recipe that's prepared with an alchohol? And if so how long does it take in total from start time to finish?", |
|
"Which is healthier the Caramel Apple Pie Cookies or the beer chicken? Please explain how you got to your answer.", |
|
"Is the moon closer to earth or the sun?", |
|
"How good are the apple pie cookies?", |
|
"What tools do I need for the Nutella Ice Cream?", |
|
"My bowl is broken, can I still make Nutella Ice Cream? Answer as a yes/no." |
|
] |
|
gr.Examples(examples, [inp]) |
|
with gr.Column(): |
|
out = gr.TextArea(label="ReAct Answer", placeholder="The answer will go here...") |
|
|
|
btn.click(fn=ask_query, inputs=inp, outputs=out) |
|
|
|
demo.launch(show_error=True) |
|
|
|
|
|
|