Update app.py

app.py

@@ -36,40 +36,57 @@ def get_product_list(product_name_by_user, data_extractor_url):
     response = find_product(product_name_by_user)
     return response
 
+
 def rda_analysis(product_info_from_db_nutritionalInformation, product_info_from_db_servingSize):
-    data_for_rda_analysis = {'nutritionPerServing' : {}, 'userServingSize' : None}
-    nutrient_name_list = ['energy', 'protein', 'carbohydrates', 'addedSugars', 'dietaryFiber', 'totalFat', 'saturatedFat', 'monounsaturatedFat', 'polyunsaturatedFat', 'transFat', 'sodium']
+    data_for_rda_analysis = {'nutritionPerServing': {}, 'userServingSize': None}
+    nutrient_name_list = ['energy', 'protein', 'carbohydrates', 'addedSugars', 'dietaryFiber', 'totalFat', 
+                          'saturatedFat', 'monounsaturatedFat', 'polyunsaturatedFat', 'transFat', 'sodium']
 
+    # Using the response_format to request output in structured JSON format
     completion = client.chat.completions.create(
         model="gpt-4o",
         messages=[
-            {"role": "system", "content": f"You will be given nutritional information of a food product. Find values of {', '.join(nutrient_name_list)} in the user-defined format"},
-            {"role": "user", "content": 
-             "Nutritional content of food product is " + json.dumps(product_info_from_db_nutritionalInformation) + """. Generate output in this JSON format 
-            {'energy': <value>,
-            'protein': <value>,
-            'carbohydrates': <value>,
-            'addedSugars': <value>,
-            'dietaryFiber': <value>,
-            'totalFat': <value>,
-            'saturatedFat': <value>,
-            'monounsaturatedFat': <value>,
-            'polyunsaturatedFat': <value>,
-            'transFat': <value>,
-            'sodium': <value>,
-            'servingSize': <Serving size of the provided nutrition value>} """}
+            {"role": "system", "content": "You will be given nutritional information of a food product."},
+            {"role": "user", "content": f"Nutritional content of food product is {json.dumps(product_info_from_db_nutritionalInformation)}. Extract the values of the following nutrients: {', '.join(nutrient_name_list)}."}
+        ],
+        response_format="json",  # This specifies we want structured JSON in return
+        functions=[
+            {
+                "name": "rda_analysis",
+                "description": "Return the extracted nutrient information in the correct format",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "energy": {"type": "number"},
+                        "protein": {"type": "number"},
+                        "carbohydrates": {"type": "number"},
+                        "addedSugars": {"type": "number"},
+                        "dietaryFiber": {"type": "number"},
+                        "totalFat": {"type": "number"},
+                        "saturatedFat": {"type": "number"},
+                        "monounsaturatedFat": {"type": "number"},
+                        "polyunsaturatedFat": {"type": "number"},
+                        "transFat": {"type": "number"},
+                        "sodium": {"type": "number"},
+                        "servingSize": {"type": "number"}  # Ensures you get serving size too
+                    },
+                    "required": nutrient_name_list + ["servingSize"]
+                }
+            }
         ]
     )
 
-    data_for_rda_analysis_part_1 = completion.choices[0].message.content
-    print(f"data_for_rda_analysis_part_1 : {data_for_rda_analysis_part_1}")
+    data_for_rda_analysis_part_1 = completion.choices[0].message["content"]
+    print(f"data_for_rda_analysis_part_1: {data_for_rda_analysis_part_1}")
 
-    data_for_rda_analysis = {}
-    data_for_rda_analysis.update({'nutritionPerServing' : data_for_rda_analysis_part_1.replace("```", "").replace("json", "")})
-    data_for_rda_analysis.update({'userServingSize' : product_info_from_db_servingSize})
+    data_for_rda_analysis = {
+        'nutritionPerServing': data_for_rda_analysis_part_1,
+        'userServingSize': product_info_from_db_servingSize
+    }
     
     return data_for_rda_analysis
 
+
 def find_product_nutrients(product_info_from_db):
     #GET Response: {'_id': '6714f0487a0e96d7aae2e839',
     #'brandName': 'Parle', 'claims': ['This product does not contain gold'],