Update app.py

app.py

@@ -55,7 +55,47 @@ class VectorStore:
 dataset = load_dataset('Thefoodprocessor/recipe_new_with_features_full')
 vector_store = VectorStore("embedding_vector")
 vector_store.populate_vectors(dataset)
+def format_recipe(input_string):
+    # Clean up the input
+    cleaned_text = input_string.strip("[]'").replace('\\n', '\n')
+    
+    # Split the text into lines
+    lines = cleaned_text.split('\n')
+    
+    # Initialize sections
+    title = lines[0]
+    ingredients = []
+    instructions = []
+    substitutions = []
 
+    # Extract ingredients and instructions
+    in_instructions = False
+    for line in lines[1:]:
+        if line.startswith("Instructions:"):
+            in_instructions = True
+            continue
+        
+        if in_instructions:
+            if line.strip():  # Check for non-empty lines
+                instructions.append(line.strip())
+        else:
+            if line.strip():  # Check for non-empty lines
+                ingredients.append(line.strip())
+    
+    # Gather substitutions from the last few lines
+    for line in lines:
+        if ':' in line:
+            substitutions.append(line.strip())
+
+    # Format output
+    formatted_recipe = f"## {title}\n\n### Ingredients:\n"
+    formatted_recipe += '\n'.join(f"- {item}" for item in ingredients) + "\n\n"
+    formatted_recipe += "### Instructions:\n" + '\n'.join(f"{i + 1}. {line}" for i, line in enumerate(instructions)) + "\n\n"
+    
+    if substitutions:
+        formatted_recipe += "### Substitutions:\n" + '\n'.join(f"- **{line.split(':')[0].strip()}**: {line.split(':')[1].strip()}" for line in substitutions) + "\n"
+    return formatted_recipe
+    # print(formatted_recipe)
 def generate_text(
     message,
     history: list[tuple[str, str]],
@@ -87,7 +127,7 @@ def generate_text(
         stream=True,
     )
     for out in output:
-        temp += out["choices"][0]["text"]
+        temp += format_recipe(out["choices"][0]["text"])
         yield temp
 
 # Define the Gradio interface