Update app.py

app.py

@@ -1,11 +1,13 @@
+import random
 import streamlit as st
-import re
-import json
-import os,secrets
+import re, ast, secrets, os, json
 import pandas as pd
 from huggingface_hub import login, InferenceClient
-import pickle
+import pickle,datetime
+from functools import partial
 from sklearn.metrics.pairwise import cosine_similarity
+import datetime
+
 
 st.set_page_config(layout="wide")
 
@@ -41,12 +43,16 @@ def show_recipe(recipe):
     col01, col02 = st.columns(2)
     with col01:
         cont = st.container(border=True, height=500)
-        for j,i in enumerate(recipe["ingredients"]):
-            cont.selectbox(i['name'],options=items_dict.iloc[get_recommendations(i['name'],cv,vectors)]["PRODUCT_NAME"].values,key=f"selectbox_{j}_{i['name']}")
+        cont.write(ingredients)
+        for j, i in enumerate(recipe["ingredients"]):
+            cont.selectbox(i['name'],
+                           options=items_dict.iloc[get_recommendations(i['name'], cv, vectors)]["PRODUCT_NAME"].values,
+                           key=f"selectbox_{j}_{i['name']}{random.random()*100}")
     with col02:
         cont = st.container(border=True, height=500)
         cont.write(instructions)
 
+
 # Initialize the inference client for the Mixtral model
 if not mode:
     cook,saved=st.tabs(['COOK','SAVED'])
@@ -288,6 +294,7 @@ if not mode:
             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:
@@ -301,4 +308,48 @@ if not mode:
                 st.empty()
             if st.session_state.recipe_saved == True:
                 st.success("Recipe Saved!")
-            
+    with saved:
+        st.title("Saved Recipes")
+        def load_saved_recipes_from_pickle(directory_path):
+            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 = r"./data/"  # Update this path if needed
+        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!="":
+            with st.spinner('LOADING YOUR RECIPE...'):
+                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.")
+