Update app.py

app.py

@@ -2,53 +2,42 @@ import os
 from dotenv import load_dotenv
 import gradio as gr
 from langchain_huggingface import HuggingFaceEndpoint
+from langchain_core.messages import HumanMessage
 
 # Load environment variables
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Initialize the Hugging Face endpoint
+# Initialize the HuggingFace inference endpoint
 llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-7B-Instruct-v0.3",  # Replace with the desired Hugging Face model
+    repo_id="flax-community/t5-recipe-generation",
     huggingfacehub_api_token=HF_TOKEN.strip(),
     temperature=0.7,
-    max_new_tokens=300
+    max_new_tokens=500
 )
 
 # Recipe generation function
 def suggest_recipes(ingredients):
+    # Create a prompt for the recipe generation
     prompt = (
-        f"You are an expert chef. Please suggest 3 unique recipes using the following "
+        f"You are an expert in cooking. Please suggest 3 recipes using the following "
         f"ingredients: {ingredients}. Provide a title for each recipe, include "
         f"preparation time, and list step-by-step directions."
     )
+    # Wrap the prompt in a HumanMessage object
+    input_message = HumanMessage(content=prompt)
     
-    try:
-        response = llm(prompt)
-        # Format response into multiple recipes
-        generated_text = response.content
-        recipes = generated_text.split("Recipe")
-        structured_recipes = []
-        
-        for i, recipe in enumerate(recipes):
-            if recipe.strip():  # Ensure non-empty recipe
-                structured_recipes.append(f"Recipe {i+1}:\n{recipe.strip()}")
-        
-        return "\n\n".join(structured_recipes)
-    
-    except Exception as e:
-        return f"Error: {e}"
+    # Use the HuggingFaceEndpoint model to generate a response
+    response = llm(input_message)
+    return response.content
 
 # Gradio interface
 with gr.Blocks() as app:
-    gr.Markdown("# AI Recipe Generator")
-    gr.Markdown("Enter the ingredients you have, and this app will generate 3 unique recipes along with preparation times!")
+    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"
-        )
+        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)