pratikshahp's picture
Update app.py
4f13a6e verified
raw
history blame
1.83 kB
import os
from dotenv import load_dotenv
import gradio as gr
from langchain_huggingface import HuggingFaceEndpoint
# Load environment variables
load_dotenv()
HF_TOKEN = os.getenv("HF_TOKEN")
# Initialize the HuggingFace inference endpoint
llm = HuggingFaceEndpoint(
repo_id="mistralai/Mistral-7B-Instruct-v0.3",
huggingfacehub_api_token=HF_TOKEN.strip(),
temperature=0.7,
)
# Recipe generation function
def suggest_recipes(ingredients):
# Create a prompt for the recipe generation
prompt = (
f"You are an expert in cooking. Carefully review the provided ingredients: {ingredients}. "
f"If any ingredient is invalid (e.g., non-food items, questions or any random words ), respond only with a two-line polite apology for not being able to process the request & Please do not provide any recipes, alternative suggestions, or outputs in such cases. "
f"If all ingredients are valid, suggest 2 recipes using them. Provide a title, preparation time, and step-by-step directions. "
f"Do not include the ingredients list in the recipes."
)
# Use the HuggingFaceEndpoint model to generate a response
response = llm(prompt)
return response
# Gradio interface
with gr.Blocks() as app:
gr.Markdown("# Recipe Suggestion App")
gr.Markdown("Provide the ingredients you have, and this app will suggest recipes along with preparation times!")
with gr.Row():
ingredients_input = gr.Textbox(label="Enter Ingredients (comma-separated):", placeholder="e.g., eggs, milk, flour")
recipe_output = gr.Textbox(label="Suggested Recipes:", lines=15, interactive=False)
generate_button = gr.Button("Get Recipes")
generate_button.click(suggest_recipes, inputs=ingredients_input, outputs=recipe_output)
# Launch the app
app.launch()