Update app.py

app.py

@@ -42,6 +42,9 @@ os.environ["TOKENIZERS_PARALLELISM"] = 'true'
 # Initialize LLM
 llm = ChatGroq(model="llama-3.1-8b-instant", temperature=0, max_tokens=1024, max_retries=2)
 
+# JSON response LLM
+json_llm = ChatGroq(model="llama-3.1-70b-versatile", temperature=0, max_tokens=1024, max_retries=2, model_kwargs={"response_format": {"type": "json_object"}})
+
 # Initialize Router
 router = ChatGroq(model="llama-3.2-3b-preview", temperature=0, max_tokens=1024, max_retries=2, model_kwargs={"response_format": {"type": "json_object"}})
 
@@ -180,22 +183,191 @@ def generate_session_key():
     return session_key
 
 
+def json_answer_generator(user_query, context):
+    system_prompt = """
+            Given a recipe context in JSON format, respond to user queries by extracting and returning the requested information in JSON format with an additional `"header"` key containing a response starter. Use the following rules:
+
+            1. **Recipe Information Extraction**: 
+            - If the user query explicitly requests specific recipe data (e.g., ingredients, nutrients, or instructions), return only those JSON objects from the provided recipe context.
+            - For example, if the user asks, “What are the ingredients?” or “Show me the nutrient details,” your output should be limited to only the requested JSON objects (e.g., `recipe_ingredients`, `recipe_nutrients`).
+            - Include `"header": "Here is the information you requested:"` at the start of each response.
+
+            2. **Multiple Information Points**:
+            - If a user query asks for more than one piece of information, return each requested JSON object from the recipe context in a combined JSON response.
+            - For example, if the query is “Give me the ingredients and instructions,” the output should include both `recipe_ingredients` and `recipe_instructions` objects.
+            - Include `"header": "Here is the information you requested:"` at the start of each response.
+
+            3. **Non-Specific Recipe Information**:
+            - If the query does not directly refer to recipe data but instead asks for a general response based on the context, return a JSON object with a single key `"content"` and a descriptive response as its value.
+            - Include `"header": "Here is a suggestion based on the recipe:"` as the response starter.
+            - For example, if the query is “How can I use this recipe for a healthy lunch?” return a response like:
+                ```json
+                {
+                    "header": "Here is a suggestion based on the recipe:",
+                    "content": "This Asian Potato Salad with Seven Minute Egg is a nutritious and light option, ideal for a balanced lunch. It provides protein and essential nutrients with low calories."
+                }
+                ```
+
+            **Example Context**:
+            ```json
+            {
+                "recipe_name": "Asian Potato Salad with Seven Minute Egg",
+                "recipe_time": 0,
+                "recipe_yields": "4 servings",
+                "recipe_ingredients": [
+                    "2 1/2 cup Multi-Colored Fingerling Potato",
+                    "3/4 cup Celery",
+                    "1/4 cup Red Onion",
+                    "2 tablespoon Fresh Parsley",
+                    "1/3 cup Mayonnaise",
+                    "1 tablespoon Chili Garlic Sauce",
+                    "1 teaspoon Hoisin Sauce",
+                    "1 splash Soy Sauce",
+                    "to taste Salt",
+                    "to taste Ground Black Pepper",
+                    "4 Egg"
+                ],
+                "recipe_instructions": "Fill a large stock pot with water. Add the Multi-Colored Fingerling Potato...",
+                "recipe_image": "https://www.sidechef.com/recipe/eeeeeceb-493e-493d-8273-66c800821b13.jpg?d=1408x1120",
+                "blogger": "sidechef.com",
+                "recipe_nutrients": {
+                    "calories": "80 calories",
+                    "proteinContent": "2.1 g",
+                    "fatContent": "6.2 g",
+                    "carbohydrateContent": "3.9 g",
+                    "fiberContent": "0.5 g",
+                    "sugarContent": "0.4 g",
+                    "sodiumContent": "108.0 mg",
+                    "saturatedFatContent": "1.2 g",
+                    "transFatContent": "0.0 g",
+                    "cholesterolContent": "47.4 mg",
+                    "unsaturatedFatContent": "3.8 g"
+                },
+                "tags": [
+                    "Salad",
+                    "Lunch",
+                    "Brunch",
+                    "Appetizers",
+                    "Side Dish",
+                    "Budget-Friendly",
+                    "Vegetarian",
+                    "Pescatarian",
+                    "Eggs",
+                    "Potatoes",
+                    "Easy",
+                    "Dairy-Free",
+                    "Shellfish-Free",
+                    "Entertaining",
+                    "Fish-Free",
+                    "Peanut-Free",
+                    "Tree Nut-Free",
+                    "Sugar-Free",
+                    "Global",
+                    "Tomato-Free",
+                    "Stove",
+                    ""
+                ],
+                "id_": "0000001"
+            }
+
+            **Example Query & Output**:
+
+            **Query**: "What are the ingredients and calories?"
+            **Output**:
+            ```json
+            {
+                "header": "Here is the information you requested:",
+                "recipe_ingredients": [
+                    "2 1/2 cup Multi-Colored Fingerling Potato",
+                    "3/4 cup Celery",
+                    "1/4 cup Red Onion",
+                    "2 tablespoon Fresh Parsley",
+                    "1/3 cup Mayonnaise",
+                    "1 tablespoon Chili Garlic Sauce",
+                    "1 teaspoon Hoisin Sauce",
+                    "1 splash Soy Sauce",
+                    "to taste Salt",
+                    "to taste Ground Black Pepper",
+                    "4 Egg"
+                ],
+                "recipe_nutrients": {
+                    "calories": "80 calories"
+                }
+            }
+
+        Try to format the output as JSON object with key value pairs.
+    """
+
+    formatted_input = f"""
+        User Query: {user_query}
+
+        Recipe data as Context:
+        {context}
+    """
+    response = router.invoke(
+        [SystemMessage(content=system_prompt)]
+        + [
+            HumanMessage(
+                content=formatted_input
+            )
+        ]
+    )
+    res = json.loads(response.content)
+    return res
+
+
 def answer_generator(formated_input, session_id):
     # QA system prompt and chain
     qa_system_prompt = """
     You are an AI assistant developed by Nutrigenics AI, specializing in intelligent recipe information retrieval and recipe suggestions. Your purpose is to help users by recommending recipes, providing detailed nutritional values, listing ingredients, offering step-by-step cooking instructions, and filtering recipes based on context and user queries.
-    Operational Guidelines:
-    1. Input Structure:
-    - Context: You may receive contextual information related to recipes, such as specific data sets, user preferences, dietary restrictions, or previously selected dishes.
-    - User Query: Users will pose questions or requests related to recipes, nutritional information, ingredient substitutions, cooking instructions, and more.
-    2. Response Strategy:
-    - Utilize Provided Context: If the context contains relevant information that addresses the user's query, base your response on this provided data to ensure accuracy and relevance.
-    - Respond to User Query Directly: If the context does not contain the necessary information to answer the user's query, kindly state that you do not have the required information.
-    Core Functionalities:
-    - Nutritional Information: Accurately provide nutritional values for each recipe, including calories, macronutrients (proteins, fats, carbohydrates), and essential vitamins and minerals, using contextual data when available.
-    - Ingredient Details: List all ingredients required for recipes, including substitute options for dietary restrictions or ingredient availability, utilizing context when relevant.
-    - Step-by-Step Cooking Instructions: Deliver clear, easy-to-follow instructions for preparing and cooking meals, informed by any provided contextual data.
-    - Recipe Recommendations: Suggest dishes based on user preferences, dietary restrictions, available ingredients, and contextual data if provided.
+    Operational Guidelines: \n
+    1. Input Structure: \n
+    - Context: You may receive contextual information related to recipes, such as specific recipe name, ingredients, nutritional informations, intsructions, recipe tags, or previously selected dishes. \n
+    - User Query: Users will pose questions or requests related to recipes, nutritional information, ingredient, cooking instructions, and more. \n
+    2. Response Strategy: \n
+    - Utilize Provided Context: If the context contains relevant information that addresses the user's query, base your response on this provided data to ensure accuracy and relevance. \n
+    - Respond to User Query Directly: If the context does not contain the necessary information to answer the user's query, kindly state that you do not have the required information. \n
+    Output Format: \n
+    - The output format should be JSON.
+    - The output should have a key 'header' with response message header such as "Here is your ....",
+    - Then there should be other key with the actual response information. If the user query asks recipe ingredients then the key should be named "ingredients" with
+    JSON object as its value. The JSON object should have ingredient and its measurement as key-value pairs. Similarly if user asked for nutritional information then the output should have 'header' key with header text and 'nutrients' key
+    with a JSON object og nutrient and its content as key-value pairs. Similarly if the user query asks for recipe instructions then JSON output should include 'header key with header text and 
+    'instructions' key with a list of instructions as its value.
+
+    Following are the output formats for some cases:
+        1. if user query asks for all recipe information, then output should be of following format:
+            {
+                header: header text,
+                recipe_name: Recipe Name,
+                recipe_instructions: List of recipe instructions,
+                recipe_nutrients: key-value pairs of nutrients name and its content,
+                recipe_ingredients: key-value pairs of ingredients name and its content,
+                recipe_tags: List of tags related to recipe,
+                .
+                .
+                .
+            }
+    
+        2. if user query asks for recipe nutrients information, then output should be of following format:
+            {
+                header: header text,
+                recipe_nutrients: key-value pairs of nutrients name and its content.
+            }
+        
+        3. if user query asks for recipe instructions information, then output should be of following format:
+            {
+                header: header text,
+                recipe_instructions: List of recipe instructions,
+            }
+
+        4. if user query asks for recipe instructions information, then output should be of following format:
+            {
+                header: header text,
+                recipe_instructions: List of recipe instructions,
+            }
+
+
     Additional Instructions:
     - Precision and Personalization: Always aim to provide precise, personalized, and relevant information to users based on both the provided context and their specific queries.
     - Clarity and Coherence: Ensure all responses are clear, well-structured, and easy to understand, facilitating a seamless user experience.
@@ -326,6 +498,7 @@ def recommendation_node(query):
     Recipe filtering instructions:
     - If a user asked for the highest nutrient recipe such as "high protein or high calories" then filtered recipes should be the top highest recipes from all the recipes with high nutrients.
     - sort or rearrange recipes based on which recipes are more appropriate for the user.
+    - Suggest dishes based on user preferences, dietary restrictions, available ingredients if specified by user.
 
     Your output instructions:
     - The function name should be filter_recipes. The input to the function should be the file name.
@@ -410,25 +583,27 @@ def get_answer(image=[], message='', sessionID='abc123'):
                 'input': message,
                 'context': data
             }
-            response = answer_generator(formated_input, session_id=sessionID)
+            # response = answer_generator(formated_input, session_id=sessionID)
+            response = json_answer_generator(message, data)
         except Exception as e:
             print(e)
             response = {'content':"An error occurred while processing your request."}
     elif (image is None) and (message is not None):
         task = router_node(message)
-        print(task)
         if task == 'recommendation':
             recipes = recommendation_node(message)
             if not recipes:
                 response = {'content': "An error occurred while processing your request."}
             else:
-                response = answer_formatter_node(message, recipes)
+                # response = answer_formatter_node(message, recipes)
+                response = recipes
         else:
             formated_input = {
                 'input': message,
                 'context': CURR_CONTEXT
             }
-            response = answer_generator(formated_input, session_id=sessionID)
+            # response = answer_generator(formated_input, session_id=sessionID)
+            response = json_answer_generator(message, CURR_CONTEXT)
     return response
 
 import json