Update app.py

app.py

@@ -4,10 +4,7 @@ import re, ast, secrets, os, json
 import pandas as pd
 from huggingface_hub import login, InferenceClient
 import pickle,datetime
-from functools import partial
 from sklearn.metrics.pairwise import cosine_similarity
-import datetime
-
 
 st.set_page_config(layout="wide")
 
@@ -56,296 +53,306 @@ def show_recipe(recipe):
 # Initialize the inference client for the Mixtral model
 
 if not mode:
+    if 'current_tab' not in st.session_state:
+    st.session_state.current_tab = 'COOK'
+    
     cook,saved=st.tabs(['COOK','SAVED'])
-    with cook:
-        client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-        
-        if 'recipe' not in st.session_state:
-            st.session_state.recipe = None
-        
-        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 = None
-        
-        if 'serving_size' not in st.session_state:
-            st.session_state.serving_size = 2
-        
-        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 = None
-        
-        def preprocess_text(text):
-            # Remove non-alphabet characters and extra spaces
-            text = re.sub(r'[^a-zA-Z\s]', '', text)
-            text = re.sub(r'\s+', ' ', text).strip()
-            return text.lower()
-        
-        def create_detailed_prompt(user_direction, exclusions, serving_size, difficulty):
-            if difficulty == "Quick & Easy":
-                prompt = (
-                    f"Provide a 'Quick and Easy' recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
-                    f"It should require as few ingredients as possible and should be ready in as little time as possible. "
-                    f"The steps should be simple, and the ingredients should be commonly found in a household pantry. "
-                    f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
-                )
-            elif difficulty == "Intermediate":
-                prompt = (
-                    f"Provide a classic recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
-                    f"The recipe should offer a bit of a cooking challenge but should not require professional skills. "
-                    f"The recipe should feature traditional ingredients and techniques that are authentic to its cuisine. "
-                    f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
-                )
-            elif difficulty == "Professional":
-                prompt = (
-                    f"Provide an advanced recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
-                    f"The recipe should push the boundaries of culinary arts, integrating unique ingredients, advanced cooking techniques, and innovative presentations. "
-                    f"The recipe should be able to be served at a high-end restaurant or would impress at a gourmet food competition. "
-                    f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
-                )
-            return prompt
-        
-        
-        def generate_recipe(user_inputs):
-            with st.spinner('Building the perfect recipe...'):
-                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"
+
+    if cook:
+        st.session_state.current_tab = 'COOK'
+    if saved:
+        st.session_state.current_tab = 'SAVED'
+    if st.session_state.current_tab == 'COOK':
+        with cook:
+            client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+            
+            if 'recipe' not in st.session_state:
+                st.session_state.recipe = None
+            
+            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 = None
+            
+            if 'serving_size' not in st.session_state:
+                st.session_state.serving_size = 2
+            
+            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 = None
+            
+            def preprocess_text(text):
+                # Remove non-alphabet characters and extra spaces
+                text = re.sub(r'[^a-zA-Z\s]', '', text)
+                text = re.sub(r'\s+', ' ', text).strip()
+                return text.lower()
+            
+            def create_detailed_prompt(user_direction, exclusions, serving_size, difficulty):
+                if difficulty == "Quick & Easy":
+                    prompt = (
+                        f"Provide a 'Quick and Easy' recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
+                        f"It should require as few ingredients as possible and should be ready in as little time as possible. "
+                        f"The steps should be simple, and the ingredients should be commonly found in a household pantry. "
+                        f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
+                    )
+                elif difficulty == "Intermediate":
+                    prompt = (
+                        f"Provide a classic recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
+                        f"The recipe should offer a bit of a cooking challenge but should not require professional skills. "
+                        f"The recipe should feature traditional ingredients and techniques that are authentic to its cuisine. "
+                        f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
+                    )
+                elif difficulty == "Professional":
+                    prompt = (
+                        f"Provide an advanced recipe for {user_direction} that excludes {exclusions} and has a serving size of {serving_size}. "
+                        f"The recipe should push the boundaries of culinary arts, integrating unique ingredients, advanced cooking techniques, and innovative presentations. "
+                        f"The recipe should be able to be served at a high-end restaurant or would impress at a gourmet food competition. "
+                        f"Provide a detailed ingredient list and step-by-step guide that explains the instructions to prepare in detail."
+                    )
+                return prompt
+            
+            
+            def generate_recipe(user_inputs):
+                with st.spinner('Building the perfect recipe...'):
+                    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"
+                                    },
+                                    "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"
+                                ],
                             },
-                            "required": [
-                                "name",
-                                "description",
-                                "ingredients",
-                                "instructions"
-                            ],
-                        },
-                    }
-                ]
-        
-                generate_kwargs = dict(
-                    temperature=0.9,
-                    max_new_tokens=10000,
-                    top_p=0.9,
-                    repetition_penalty=1.0,
-                    do_sample=True,
+                        }
+                    ]
+            
+                    generate_kwargs = dict(
+                        temperature=0.9,
+                        max_new_tokens=10000,
+                        top_p=0.9,
+                        repetition_penalty=1.0,
+                        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)
+                    st.session_state.recipe = response
+                    st.session_state.recipe_saved = False
+            
+            
+            def clear_inputs():
+                st.session_state.user_direction = None
+                st.session_state.exclusions = None
+                st.session_state.serving_size = 2
+                st.session_state.selected_difficulty = "Quick & Easy"
+                st.session_state.recipe=None
+            
+            
+            st.title("Let's get cooking")
+            col1,col2=st.columns(2)
+            with col1:
+                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",
                 )
-        
-                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)
-                st.session_state.recipe = response
-                st.session_state.recipe_saved = False
-        
-        
-        def clear_inputs():
-            st.session_state.user_direction = None
-            st.session_state.exclusions = None
-            st.session_state.serving_size = 2
-            st.session_state.selected_difficulty = "Quick & Easy"
-            st.session_state.recipe=None
-        
-        
-        st.title("Let's get cooking")
-        col1,col2=st.columns(2)
-        with col1:
-            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",
+            with col2:
+                st.session_state.serving_size = st.number_input(
+                    "How many servings would you like to cook?",
+                    min_value=1,
+                    max_value=100,
+                    value=st.session_state.serving_size,
+                    step=1
+                )
+            
+            difficulty_dictionary = {
+                "Quick & Easy": {
+                    "description": "Easy recipes with straightforward instructions. Ideal for beginners or those seeking quick and simple cooking.",
+                },
+                "Intermediate": {
+                    "description": "Recipes with some intricate steps that invite a little challenge. Perfect for regular cooks wanting to expand their repertoire with new ingredients and techniques.",
+                },
+                "Professional": {
+                    "description": "Complex recipes that demand a high level of skill and precision. Suited for seasoned cooks aspiring to professional-level sophistication and creativity.",
+                }
+            }
+            
+            st.session_state.selected_difficulty = st.radio(
+                "Choose a difficulty level for your recipe.",
+                [
+                    list(difficulty_dictionary.keys())[0],
+                    list(difficulty_dictionary.keys())[1],
+                    list(difficulty_dictionary.keys())[2]
+                ],
+                captions=[
+                    difficulty_dictionary["Quick & Easy"]["description"],
+                    difficulty_dictionary["Intermediate"]["description"],
+                    difficulty_dictionary["Professional"]["description"]
+                ],
+                index=list(difficulty_dictionary).index(st.session_state.selected_difficulty)
             )
-        with col2:
-            st.session_state.serving_size = st.number_input(
-                "How many servings would you like to cook?",
-                min_value=1,
-                max_value=100,
-                value=st.session_state.serving_size,
-                step=1
+            
+            st.session_state.exclusions = st.text_area(
+                "Any ingredients you want to exclude?",
+                value=st.session_state.exclusions,
+                placeholder="gluten, dairy, nuts, cilantro",
             )
-        
-        difficulty_dictionary = {
-            "Quick & Easy": {
-                "description": "Easy recipes with straightforward instructions. Ideal for beginners or those seeking quick and simple cooking.",
-            },
-            "Intermediate": {
-                "description": "Recipes with some intricate steps that invite a little challenge. Perfect for regular cooks wanting to expand their repertoire with new ingredients and techniques.",
-            },
-            "Professional": {
-                "description": "Complex recipes that demand a high level of skill and precision. Suited for seasoned cooks aspiring to professional-level sophistication and creativity.",
+            
+            fancy_exclusions = ""
+            
+            if st.session_state.selected_difficulty == "Professional":
+                exclude_fancy = st.checkbox(
+                    "Exclude cliche professional ingredients? (gold leaf, truffle, edible flowers, microgreens)",
+                    value=True)
+                if exclude_fancy:
+                    fancy_exclusions = "gold leaf, truffle, edible flowers, microgreens, gold dust"
+            
+            user_inputs = {
+                "user_direction": st.session_state.user_direction,
+                "exclusions": f"{st.session_state.exclusions}, {fancy_exclusions}".strip(", "),
+                "serving_size": st.session_state.serving_size,
+                "difficulty": st.session_state.selected_difficulty
             }
-        }
-        
-        st.session_state.selected_difficulty = st.radio(
-            "Choose a difficulty level for your recipe.",
-            [
-                list(difficulty_dictionary.keys())[0],
-                list(difficulty_dictionary.keys())[1],
-                list(difficulty_dictionary.keys())[2]
-            ],
-            captions=[
-                difficulty_dictionary["Quick & Easy"]["description"],
-                difficulty_dictionary["Intermediate"]["description"],
-                difficulty_dictionary["Professional"]["description"]
-            ],
-            index=list(difficulty_dictionary).index(st.session_state.selected_difficulty)
-        )
-        
-        st.session_state.exclusions = st.text_area(
-            "Any ingredients you want to exclude?",
-            value=st.session_state.exclusions,
-            placeholder="gluten, dairy, nuts, cilantro",
-        )
-        
-        fancy_exclusions = ""
-        
-        if st.session_state.selected_difficulty == "Professional":
-            exclude_fancy = st.checkbox(
-                "Exclude cliche professional ingredients? (gold leaf, truffle, edible flowers, microgreens)",
-                value=True)
-            if exclude_fancy:
-                fancy_exclusions = "gold leaf, truffle, edible flowers, microgreens, gold dust"
-        
-        user_inputs = {
-            "user_direction": st.session_state.user_direction,
-            "exclusions": f"{st.session_state.exclusions}, {fancy_exclusions}".strip(", "),
-            "serving_size": st.session_state.serving_size,
-            "difficulty": st.session_state.selected_difficulty
-        }
-        
-        button_cols_submit = st.columns([1, 1, 4])
-        with button_cols_submit[0]:
-            st.button(label='Submit', on_click=generate_recipe, kwargs=dict(user_inputs=user_inputs), type="primary",
-                      use_container_width=True)
-        with button_cols_submit[1]:
-            st.button(label='Reset', on_click=clear_inputs, type="secondary", use_container_width=True)
-        with button_cols_submit[2]:
-            st.empty()
-
-        def create_safe_filename(recipe_name):
-            # format and generate random URL-safe text string
-            safe_name = recipe_name.lower()
-            safe_name = safe_name.replace(" ", "_")
-            safe_name = re.sub(r"[^a-zA-Z0-9_]", "", safe_name)
-            safe_name = (safe_name[:50]) if len(safe_name) > 50 else safe_name
-            unique_token = secrets.token_hex(8)
-            safe_filename = f"{unique_token}_{safe_name}"
-            return safe_filename
-
-        def save_recipe():
-            with st.spinner('WAIT SAVING YOUR DISH...'):
-                filename = create_safe_filename(recipe["name"])
-                os.makedirs('data', exist_ok=True)
-                with open(f'./data/{filename}.pkl', 'wb') as f:
-                    pickle.dump(recipe, f)
-                st.session_state.recipe_saved = True
-        
-        if st.session_state.recipe is not None:
-            st.divider()
-            print(st.session_state.recipe)
-            recipe = json.loads(st.session_state.recipe)
-            show_recipe(recipe)
-            recipe['timestamp'] = str(datetime.datetime.now())
-            if st.session_state.recipe_saved == True:
-                disable_button = True
-            else:
-                disable_button = False
-            button_cols_save = st.columns([1, 1, 4])
-            with button_cols_save[0]:
-                st.button("Save Recipe", on_click=save_recipe, disabled=disable_button, type="primary")
-            with button_cols_save[1]:
-                st.empty()
-            with button_cols_save[2]:
+            
+            button_cols_submit = st.columns([1, 1, 4])
+            with button_cols_submit[0]:
+                st.button(label='Submit', on_click=generate_recipe, kwargs=dict(user_inputs=user_inputs), type="primary",
+                          use_container_width=True)
+            with button_cols_submit[1]:
+                st.button(label='Reset', on_click=clear_inputs, type="secondary", use_container_width=True)
+            with button_cols_submit[2]:
                 st.empty()
-            if st.session_state.recipe_saved == True:
-                st.success("Recipe Saved!")
-    with saved:
-        st.title("Saved Recipes")
-        with st.spinner('LOADING YOUR RECIPE...'):
-            def load_saved_recipes_from_pickle(directory_path):
-                os.makedirs('data', exist_ok=True)
-                recipes = []
-                # Iterate through all files in the directory
-                for filename in os.listdir(directory_path):
-                    if filename.endswith('.pkl'):
-                        file_path = os.path.join(directory_path, filename)
-                        with open(file_path, 'rb') as file:
-                            recipe = pickle.load(file)
-                            recipes.append(recipe)
-                return recipes
-    
     
-            # get all saved files
-            directory_path = 'data'
-            recipes = load_saved_recipes_from_pickle(directory_path)
-            print(recipes)
+            def create_safe_filename(recipe_name):
+                # format and generate random URL-safe text string
+                safe_name = recipe_name.lower()
+                safe_name = safe_name.replace(" ", "_")
+                safe_name = re.sub(r"[^a-zA-Z0-9_]", "", safe_name)
+                safe_name = (safe_name[:50]) if len(safe_name) > 50 else safe_name
+                unique_token = secrets.token_hex(8)
+                safe_filename = f"{unique_token}_{safe_name}"
+                return safe_filename
     
-            cols = st.columns([4, 1])
-            with cols[1]:
-                user_sort = st.selectbox("Sort", ('Recent', 'Oldest', 'A-Z', 'Z-A', 'Random'))
-                if user_sort == 'Recent':
-                    recipes.sort(key=lambda x: x['timestamp'], reverse=True)
-                elif user_sort == 'Oldest':
-                    recipes.sort(key=lambda x: x['timestamp'])
-                elif user_sort == 'A-Z':
-                    recipes.sort(key=lambda x: x['name'])
-                elif user_sort == 'Z-A':
-                    recipes.sort(key=lambda x: x['name'], reverse=True)
-                elif user_sort == 'Random':
-                    recipes.sort(key=lambda x: x['file'])
-            with cols[0]:
-                user_search = st.selectbox("Search Recipes", [""] + [recipe['name'] for recipe in recipes])
-            st.write("")  # just some space
+            def save_recipe():
+                with st.spinner('WAIT SAVING YOUR DISH...'):
+                    filename = create_safe_filename(recipe["name"])
+                    os.makedirs('data', exist_ok=True)
+                    with open(f'./data/{filename}.pkl', 'wb') as f:
+                        pickle.dump(recipe, f)
+                    st.session_state.recipe_saved = True
             
-            if user_search!="":
+            if st.session_state.recipe is not None:
                 st.divider()
-                filtered_recipes = [recipe for recipe in recipes if recipe['name'] == user_search]
-                if filtered_recipes:
-                    show_recipe(filtered_recipes[0])
+                print(st.session_state.recipe)
+                recipe = json.loads(st.session_state.recipe)
+                if not st.session_state.recipe_saved:
+                    show_recipe(recipe)
+                recipe['timestamp'] = str(datetime.datetime.now())
+                if st.session_state.recipe_saved == True:
+                    disable_button = True
                 else:
-                    st.write("No recipe found.")
-    
+                    disable_button = False
+                button_cols_save = st.columns([1, 1, 4])
+                with button_cols_save[0]:
+                    st.button("Save Recipe", on_click=save_recipe, disabled=disable_button, type="primary")
+                with button_cols_save[1]:
+                    st.empty()
+                with button_cols_save[2]:
+                    st.empty()
+                if st.session_state.recipe_saved == True:
+                    st.success("Recipe Saved!")
+    if st.session_state.current_tab == 'SAVED':
+        with saved:
+            st.title("Saved Recipes")
+            with st.spinner('LOADING YOUR RECIPE...'):
+                def load_saved_recipes_from_pickle(directory_path):
+                    os.makedirs('data', exist_ok=True)
+                    recipes = []
+                    # Iterate through all files in the directory
+                    for filename in os.listdir(directory_path):
+                        if filename.endswith('.pkl'):
+                            file_path = os.path.join(directory_path, filename)
+                            with open(file_path, 'rb') as file:
+                                recipe = pickle.load(file)
+                                recipes.append(recipe)
+                    return recipes
+        
+        
+                # get all saved files
+                directory_path = 'data'
+                recipes = load_saved_recipes_from_pickle(directory_path)
+                print(recipes)
+        
+                cols = st.columns([4, 1])
+                with cols[1]:
+                    user_sort = st.selectbox("Sort", ('Recent', 'Oldest', 'A-Z', 'Z-A', 'Random'))
+                    if user_sort == 'Recent':
+                        recipes.sort(key=lambda x: x['timestamp'], reverse=True)
+                    elif user_sort == 'Oldest':
+                        recipes.sort(key=lambda x: x['timestamp'])
+                    elif user_sort == 'A-Z':
+                        recipes.sort(key=lambda x: x['name'])
+                    elif user_sort == 'Z-A':
+                        recipes.sort(key=lambda x: x['name'], reverse=True)
+                    elif user_sort == 'Random':
+                        recipes.sort(key=lambda x: x['file'])
+                with cols[0]:
+                    user_search = st.selectbox("Search Recipes", [""] + [recipe['name'] for recipe in recipes])
+                st.write("")  # just some space
+                
+                if user_search!="":
+                    st.divider()
+                    filtered_recipes = [recipe for recipe in recipes if recipe['name'] == user_search]
+                    if filtered_recipes:
+                        show_recipe(filtered_recipes[0])
+                    else:
+                        st.write("No recipe found.")