Update app.py

app.py

@@ -114,16 +114,16 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
         # return {"predicted_label": prediction, "nutrition_info": response}
 
         # nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} per 100 grams in a formatted list only."
-        nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} per 100 grams, Output the information as a concise, formatted list without repetition."
+        nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6) for {prediction} per 100 grams, Output the information as a concise, formatted list without repetition."
         nutrition_info = call_llm(llm_client, nutrition_prompt)
         
         # # Second prompt: Health benefits and tips
-        # health_benefits_prompt = f"Provide the health benefits and considerations for {prediction}. Additionally, include practical tips for making {prediction} healthier. Keep the response focused on these two aspects only."
-        # health_benefits_and_tips = call_llm(llm_client, health_benefits_prompt)
+        health_benefits_prompt = f"Provide the health benefits and considerations for {prediction}. Additionally, include practical tips for making {prediction} healthier. Keep the response focused on these two aspects only."
+        health_benefits_and_tips = call_llm(llm_client, health_benefits_prompt)
 
         return {
             "predicted_label": prediction, 
-            # "health_benefits_and_tips": health_benefits_and_tips,
+            "health_benefits_and_tips": health_benefits_and_tips,
             "nutrition_info": nutrition_info
         }
     except Exception as e:
@@ -135,7 +135,7 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
 
 
 
-        #nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} in a formatted list only."
+        #nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6) for {prediction} in a formatted list only."
         # nutrition_info = call_llm(llm_client, nutrition_prompt)
 
         # # Second prompt: Health benefits and tips