Update Home.py

Home.py

@@ -20,6 +20,16 @@ if 'recipe_saved' not in st.session_state:
 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 = 0
+
+if 'exclusions' not in st.session_state:
+    st.session_state.exclusions = None
+
+
 # functions
 def create_detailed_prompt(user_direction, exclusions, serving_size, difficulty):
     if difficulty == "Quick & Easy":
@@ -130,6 +140,9 @@ def save_recipe():
 
 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"
     
 # app
 st.title("Let's get cooking")
@@ -139,7 +152,7 @@ st.session_state.user_direction = st.text_area(
     placeholder="quick snack, asian style bowl with either noodles or rice, something italian",
     )
 
-serving_size = st.number_input(
+st.session_state.serving_size = st.number_input(
     "How many servings would you like to cook?",
     min_value=1,
     max_value=100,
@@ -159,7 +172,7 @@ difficulty_dictionary = {
     }
 }
 
-selected_difficulty = st.radio(
+st.session_state.selected_difficulty = st.radio(
     "Choose a difficulty level for your recipe.",
     [
         list(difficulty_dictionary.keys())[0], 
@@ -170,11 +183,13 @@ selected_difficulty = st.radio(
         difficulty_dictionary["Quick & Easy"]["description"], 
         difficulty_dictionary["Intermediate"]["description"],
         difficulty_dictionary["Professional"]["description"]
-    ]
+    ],
+    index=list(difficulty_dictionary).index(st.session_state.selected_difficulty)
 )
 
-exclusions = st.text_area(
+st.session_state.exclusions = st.text_area(
     "Any ingredients you want to exclude?",
+    value = st.session_state.exclusions,
     placeholder="gluten, dairy, nuts, cilantro",
     )
 
@@ -189,13 +204,16 @@ if selected_difficulty == "Professional":
 
 user_inputs = {
     "user_direction" : st.session_state.user_direction,
-    "exclusions": f"{exclusions}, {fancy_exclusions}",
-    "serving_size": serving_size,
-    "difficulty": selected_difficulty
+    "exclusions": f"{st.session_state.exclusions}, {fancy_exclusions}",
+    "serving_size": st.session_state.serving_size,
+    "difficulty": st.session_state.selected_difficulty
 }
 
-st.button(label='Submit', on_click=generate_recipe, kwargs=dict(user_inputs=user_inputs), type="primary")
-st.button(label='Reset', on_click=clear_inputs, type="secondary")
+col1, col2 = st.columns(2)
+with col1:
+    st.button(label='Submit', on_click=generate_recipe, kwargs=dict(user_inputs=user_inputs), type="primary")
+with col2:
+    st.button(label='Reset', on_click=clear_inputs, type="secondary")
 
 if st.session_state.gpt_response is not None:
     st.divider()