Update app.py

app.py

@@ -1,15 +1,11 @@
-import os
-import re
-import ast
-import json
-import pickle
-import secrets
 import random
-import datetime
+import streamlit as st
+import re, secrets, os, json
 import pandas as pd
 from huggingface_hub import login, InferenceClient
+import pickle
 from sklearn.metrics.pairwise import cosine_similarity
-import streamlit as st
+import datetime
 import wordcloud
 import matplotlib.pyplot as plt
 from pygwalker.api.streamlit import StreamlitRenderer
@@ -31,7 +27,7 @@ with open('items_dict.pkl', 'rb') as file:
     items_dict = pd.DataFrame.from_dict(pickle.load(file))
 
 
-mode = st.toggle(label="MART")
+mode = st.toggle(label="MARTMAN")
 
 if 'ind' not in st.session_state:
     st.session_state.ind = []
@@ -71,7 +67,6 @@ def show_recipe(recipe):
 
 
 # Initialize the inference client for the Mixtral model
-
 if not mode:
     cook, saved = st.tabs(['COOK', 'SAVED'])
     with cook:
@@ -96,13 +91,6 @@ if not mode:
             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 = (
@@ -127,7 +115,6 @@ if not mode:
                 )
             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'],
@@ -201,7 +188,6 @@ if not mode:
                 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
@@ -209,7 +195,6 @@ if not mode:
             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:
@@ -278,14 +263,14 @@ if not mode:
 
         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",
+            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()
@@ -296,7 +281,6 @@ if not mode:
             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"])
@@ -305,13 +289,11 @@ if not mode:
                     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)
-            if not st.session_state.recipe_saved:
-                show_recipe(recipe)
+            show_recipe(recipe)
             recipe['timestamp'] = str(datetime.datetime.now())
             if st.session_state.recipe_saved == True:
                 disable_button = True
@@ -341,7 +323,6 @@ if not mode:
                             recipes.append(recipe)
                 return recipes
 
-
             # get all saved files
             directory_path = 'data'
             recipes = load_saved_recipes_from_pickle(directory_path)
@@ -428,4 +409,4 @@ else:
                 return StreamlitRenderer(df)
             # st.markdown("## :desktop_computer: DO YOUR OWN RESEARCH...")
             with st.expander("## :desktop_computer: LOOKING FOR SOMETHING ELSE..."):
-                pygwalerapp(df).explorer()
+                pygwalerapp(df).explorer()