Update app.py

app.py

@@ -107,24 +107,30 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
         
         # Use the prediction to generate nutrition information
         # 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"
-        # response = call_llm(llm_client, prompt)
+        prompt = f"Provide all the nutrition information for {prediction}, including Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, and Pantothenic acid. Please present the information in a clear, formatted list only, without additional explanations."
+        response = call_llm(llm_client, prompt)
 
-        # return {"predicted_label": prediction, "nutrition_info": response}
+        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} in a formatted list only."
-        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} 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,
-            "health_benefits_and_tips": health_benefits_and_tips
-        }
+        
     except Exception as e:
         return JSONResponse(
             status_code=500,
             content={"error": f"An error occurred: {str(e)}"}
         )
+
+
+
+
+        #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: 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,
+        #     "health_benefits_and_tips": health_benefits_and_tips
+        # }