Update app.py

app.py

@@ -14,13 +14,10 @@ start_time = time.time()
 dataset = load_dataset("corbt/all-recipes", split="train[:20000]")
 
 # Preprocessing functions to extract features from the raw text
-def extract_title_and_ingredients(sample):
-    extraction = sample['input'][:sample['input'].find("Directions")]
-    return {"text_for_embedding": extraction}
-
 def extract_each_feature(sample):
     full_text = sample['input']
     
+    # User's fix: Use "\n" instead of "\\n" to correctly find the title
     title = full_text[:full_text.find("\n")]
     
     ingredients = "Not available"
@@ -51,7 +48,6 @@ def extract_each_feature(sample):
         "directions": directions,
     }
 
-
 # Apply preprocessing
 dataset = dataset.map(extract_each_feature)
 
@@ -89,9 +85,11 @@ def get_recommendations_and_generate(query_ingredients, k=3):
     while len(results) < 3:
         results.append({"title": "No recipe found", "ingredients": "", "directions": ""})
         
-    # 2. Generate 10 new recipe ideas
-    prompt = f"Create a simple and delicious recipe using the following ingredients: {query_ingredients}."
-    generated_outputs = generator(prompt, max_new_tokens=200, num_return_sequences=10, pad_token_id=50256)
+    # 2. Generate 10 new recipe ideas with a simpler, more direct prompt
+    prompt = f"Write a complete recipe that includes a title, a list of ingredients, and step-by-step directions. The recipe must use the following ingredients: {query_ingredients}."
+    
+    # Optimized for speed by reducing max_new_tokens
+    generated_outputs = generator(prompt, max_new_tokens=180, num_return_sequences=10, pad_token_id=50256)
     
     # 3. Find the best recipe out of the 10 generated
     generated_texts = [output['generated_text'].replace(prompt, "").strip() for output in generated_outputs]
@@ -116,7 +114,7 @@ def format_recipe(recipe):
     return f"### {recipe['title']}\n**Ingredients:**\n{recipe['ingredients']}\n\n**Directions:**\n{recipe['directions']}"
 
 def format_generated_recipe(recipe_text):
-    # Formats the AI-generated recipe as simple text
+    # Formats the AI-generated recipe as simple text, without extra markdown
     return recipe_text
 
 def recipe_wizard(ingredients):
@@ -143,7 +141,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             output_rec3 = gr.Markdown()
         with gr.Column(scale=1):
             gr.Markdown("### ✨ AI-Generated Idea")
-            output_gen = gr.Markdown()
+            output_gen = gr.Textbox(label="AI Generated Recipe", lines=15) # Changed to Textbox for plain text
 
     submit_btn.click(
         fn=recipe_wizard,