Update app.py

app.py

@@ -1,30 +1,36 @@
 import streamlit as st
+import re
+import json
 import os
 import pandas as pd
 from huggingface_hub import login, InferenceClient
 import pickle
-import json
+from sklearn.metrics.pairwise import cosine_similarity
 
 st.set_page_config(layout="wide")
 
-# Authenticate with Hugging Face
 login(token=os.getenv("TOKEN"))
 
-# Load similarity data
-with open('l.pkl', 'rb') as file:
-    similarity = pickle.load(file)
+try:
+    with open('l.pkl', 'rb') as file:
+        similarity = pickle.load(file)
+except FileNotFoundError:
+    st.error("The similarity file was not found.")
 
-# Load items dictionary
-with open('items_dict.pkl', 'rb') as file:
-    items_dict = pickle.load(file)
+try:
+    with open('items_dict.pkl', 'rb') as file:
+        items_dict = pickle.load(file)
+except FileNotFoundError:
+    st.error("The items dictionary file was not found.")
 
-# Load dataset
-data = pd.read_csv('marketing_sample_for_walmart_com-product_details__20200101_20200331__30k_data.csv')
+try:
+    data = pd.read_csv('marketing_sample_for_walmart_com-product_details__20200101_20200331__30k_data.csv')
+except FileNotFoundError:
+    st.error("The CSV file was not found.")
 
 # Initialize the inference client for the Mixtral model
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Initialize session state variables
 if 'recipe' not in st.session_state:
     st.session_state.recipe = None
 
@@ -32,7 +38,7 @@ if 'recipe_saved' not in st.session_state:
     st.session_state.recipe_saved = None
 
 if 'user_direction' not in st.session_state:
-    st.session_state.user_direction = ""
+    st.session_state.user_direction = None
 
 if 'serving_size' not in st.session_state:
     st.session_state.serving_size = 2
@@ -41,7 +47,7 @@ if 'selected_difficulty' not in st.session_state:
     st.session_state.selected_difficulty = "Quick & Easy"
 
 if 'exclusions' not in st.session_state:
-    st.session_state.exclusions = ""
+    st.session_state.exclusions = None
 
 
 def create_detailed_prompt(user_direction, exclusions, serving_size, difficulty):
@@ -74,6 +80,60 @@ def generate_recipe(user_inputs):
         prompt = create_detailed_prompt(user_inputs['user_direction'], user_inputs['exclusions'],
                                         user_inputs['serving_size'], user_inputs['difficulty'])
 
+        functions = [
+            {
+                "name": "provide_recipe",
+                "description": "Provides a detailed recipe strictly adhering to the user input/specifications, especially ingredient exclusions and the recipe difficulty",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "name": {
+                            "type": "string",
+                            "description": "A creative name for the recipe"
+                        },
+                        "description": {
+                            "type": "string",
+                            "description": "a brief one-sentence description of the provided recipe"
+                        },
+                        "ingredients": {
+                            "type": "array",
+                            "items": {
+                                "type": "object",
+                                "properties": {
+                                    "name": {
+                                        "type": "string",
+                                        "description": "Quantity and name of the ingredient"
+                                    }
+                                }
+                            }
+                        },
+                        "instructions": {
+                            "type": "array",
+                            "items": {
+                                "type": "object",
+                                "properties": {
+                                    "step_number": {
+                                        "type": "number",
+                                        "description": "The sequence number of this step"
+                                    },
+                                    "instruction": {
+                                        "type": "string",
+                                        "description": "Detailed description of what to do in this step"
+                                    }
+                                }
+                            }
+                        }
+                    },
+                    "required": [
+                        "name",
+                        "description",
+                        "ingredients",
+                        "instructions"
+                    ],
+                },
+            }
+        ]
+
         generate_kwargs = dict(
             temperature=0.9,
             max_new_tokens=10000,
@@ -82,17 +142,16 @@ def generate_recipe(user_inputs):
             do_sample=True,
         )
 
+        prompt += f"\nPlease format the output in JSON. The JSON should include fields for 'name', 'description', 'ingredients', and 'instructions', with each field structured as described below.\n\n{json.dumps(functions)}"
+
         response = client.text_generation(prompt, **generate_kwargs)
-        try:
-            st.session_state.recipe = json.loads(response)
-        except json.JSONDecodeError:
-            st.session_state.recipe = None
+        st.session_state.recipe = response
         st.session_state.recipe_saved = False
 
 
 def clear_inputs():
-    st.session_state.user_direction = ""
-    st.session_state.exclusions = ""
+    st.session_state.user_direction = None
+    st.session_state.exclusions = None
     st.session_state.serving_size = 2
     st.session_state.selected_difficulty = "Quick & Easy"
 
@@ -101,7 +160,7 @@ st.title("Let's get cooking")
 st.session_state.user_direction = st.text_area(
     "What do you want to cook? Describe anything - a dish, cuisine, event, or vibe.",
     value=st.session_state.user_direction,
-    placeholder="quick snack, Asian style bowl with either noodles or rice, something Italian",
+    placeholder="quick snack, asian style bowl with either noodles or rice, something italian",
 )
 
 st.session_state.serving_size = st.number_input(
@@ -126,14 +185,10 @@ difficulty_dictionary = {
 
 st.session_state.selected_difficulty = st.radio(
     "Choose a difficulty level for your recipe.",
-    options=list(difficulty_dictionary.keys()),
+    list(difficulty_dictionary.keys()),
     index=list(difficulty_dictionary).index(st.session_state.selected_difficulty)
 )
 
-for level, details in difficulty_dictionary.items():
-    if level == st.session_state.selected_difficulty:
-        st.caption(details["description"])
-
 st.session_state.exclusions = st.text_area(
     "Any ingredients you want to exclude?",
     value=st.session_state.exclusions,
@@ -141,6 +196,7 @@ st.session_state.exclusions = st.text_area(
 )
 
 fancy_exclusions = ""
+
 if st.session_state.selected_difficulty == "Professional":
     exclude_fancy = st.checkbox(
         "Exclude cliche professional ingredients? (gold leaf, truffle, edible flowers, microgreens)",
@@ -150,7 +206,7 @@ if st.session_state.selected_difficulty == "Professional":
 
 user_inputs = {
     "user_direction": st.session_state.user_direction,
-    "exclusions": f"{st.session_state.exclusions}, {fancy_exclusions}",
+    "exclusions": f"{st.session_state.exclusions}, {fancy_exclusions}".strip(", "),
     "serving_size": st.session_state.serving_size,
     "difficulty": st.session_state.selected_difficulty
 }
@@ -164,22 +220,27 @@ with button_cols_submit[1]:
 with button_cols_submit[2]:
     st.empty()
 
-if st.session_state.recipe:
+if st.session_state.recipe is not None:
     st.divider()
-    print(st.session_state.recipe)
-    recipe = st.session_state.recipe
-    name_and_dis = f'# {recipe["name"]}\n\n'
-    name_and_dis += f'{recipe["description"]}\n\n'
-    ingredients = '## Ingredients:\n'
-    for ingredient in recipe["ingredients"]:
-        ingredients += f"- {ingredient['name']}\n"
-    instructions = '\n## Instructions:\n'
-    for instruction in recipe["instructions"]:
-        instructions += f"{instruction['step_number']}. {instruction['instruction']}\n"
-
-    st.write(name_and_dis)
-    col01, col02 = st.columns(2)
-    with col01:
-        st.write(ingredients)
-    with col02:
-        st.write(instructions)
+    try:
+        print(st.session_state.recipe)
+        recipe = json.loads(st.session_state.recipe)
+        name_and_dis = f'# {recipe["name"]}\n\n'
+        name_and_dis += f'{recipe["description"]}\n\n'
+        ingredients = '## Ingredients:\n'
+        for ingredient in recipe["ingredients"]:
+            ingredients += f"- {ingredient['name']}\n"
+        instructions = '\n## Instructions:\n'
+        for instruction in recipe["instructions"]:
+            instructions += f"{instruction['step_number']}. {instruction['instruction']}\n"
+
+        st.write(name_and_dis)
+        col01, col02 = st.columns(2)
+        with col01:
+            cont = st.container()
+            cont.write(ingredients)
+        with col02:
+            cont = st.container()
+            cont.write(instructions)
+    except (json.JSONDecodeError, KeyError) as e:
+        st.error(f"Failed to parse recipe: {e}")