Update app.py

app.py

@@ -29,13 +29,25 @@ def get_ingredients_llama(food_name):
     Returns a clean, comma-separated list of ingredients.
     """
     prompt = (
-        f"List only the main ingredients typically used to prepare {food_name}. "
-        "Respond with just the ingredients in a comma-separated list without any extra text or explanations."
+        f"List only the ingredients for {food_name}. "
+        "Output the ingredients as a simple, comma-separated list such as: ingredient1, ingredient2, ingredient3."
     )
     try:
         response = pipe_llama(prompt, max_length=50, num_return_sequences=1)
         generated_text = response[0]["generated_text"].strip()
         
+        # Post-process to extract only the list of ingredients
+        ingredients = generated_text.split(":")[-1].strip()  # Handle cases like "Ingredients: ..."
+        ingredients = ingredients.replace(".", "").strip()  # Remove periods and extra spaces
+        
+        # Ensure the response is a proper list format
+        if "," not in ingredients:  # If the model didn't output a list, fallback to generic handling
+            ingredients = "No ingredients found or response unclear."
+        
+        return ingredients
+    except Exception as e:
+        return f"Error generating ingredients: {e}"
+        
         # Process the response to ensure it's a clean, comma-separated list
         ingredients = generated_text.split(":")[-1].strip()  # Handle cases like "Ingredients: ..."
         ingredients = ingredients.replace(".", "").strip()  # Remove periods and extra spaces