Update app.py

app.py

@@ -111,17 +111,17 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
 
         # return {"predicted_label": prediction, "nutrition_info": response}
 
-        nutrition_prompt = f"Nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} in 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} in a formatted list only."
         nutrition_info = call_llm(llm_client, nutrition_prompt)
 
-        # Second prompt: Information about the prediction label
-        info_prompt = f"Provide information about the label {prediction}, including its origin, general uses, and key characteristics."
-        prediction_info = call_llm(llm_client, info_prompt)
+        # Second prompt: Health benefits and tips
+        health_benefits_prompt = f"Provide the health benefits and considerations for {prediction} and give tips for making it healthier."
+        health_benefits_and_tips = call_llm(llm_client, health_benefits_prompt)
 
         return {
             "predicted_label": prediction, 
             "nutrition_info": nutrition_info,
-            "prediction_info": prediction_info
+            "health_benefits_and_tips": health_benefits_and_tips
         }
     except Exception as e:
         return JSONResponse(