Update app.py

app.py

@@ -4,14 +4,13 @@ import random
 import pandas as pd
 import streamlit as st
 
+# Define session state variables
 if 'df' not in st.session_state:
     st.session_state.df = pd.DataFrame()
     st.session_state.df_edited = st.session_state.df.copy()
     st.session_state.df2 = pd.DataFrame()
 
-st.title('Get Recipes:')
-n_recipes = st.slider('Select number of recipes to get', 0, 20, 10)
-
+# Define functions
 def get_recipes():
     url = "https://api.notion.com/v1/databases/180f4b492f4d421c88028c54cfe077a5/query"
     payload = json.dumps({
@@ -33,8 +32,8 @@ def get_recipes():
     recipe_list = []
     for recipe in random.sample(recipes['results'], n_recipes):
         recipe_dict = {}
-        recipe_dict['name'] = recipe["properties"]["Name"]["title"][0]["plain_text"].strip()
-        recipe_dict['url'] = recipe["url"]
+        recipe_dict['Name'] = recipe["properties"]["Name"]["title"][0]["plain_text"].strip()
+        recipe_dict['Url'] = recipe["url"]
         recipe_list.append(recipe_dict)
     
     st.session_state.df = pd.DataFrame.from_dict(recipe_list)
@@ -48,22 +47,33 @@ def save_recipes():
 def clear_saved():
     st.session_state.df2 = pd.DataFrame()
 
-st.button(label='Get', on_click=get_recipes)
 
+# Define app
+st.title('Get Recipes:')
+n_recipes = st.slider('Select number of recipes to get', 0, 20, 10)
+st.button(label='Get', on_click=get_recipes)
 if len(st.session_state.df) > 0:
     st.session_state.df_edited = st.data_editor(
             st.session_state.df,
-            column_config={"url": st.column_config.LinkColumn()},
-            disabled=["name", "url"],
+            column_config={
+                "Name": st.column_config.Column(width="medium"),
+                "Save": st.column_config.Column(width="small"),
+                "Url": st.column_config.LinkColumn(width="large")
+            },
+            disabled=["Name", "Url"],
             hide_index=True,
         )
     st.button(label='Save Recipes', on_click=save_recipes)
-
 st.title('Saved Recipes:')
-
 if len(st.session_state.df2) == 0:
     st.write('Nothing saved, yet!')
 else:
-    st.dataframe(st.session_state.df2, hide_index=True)
-
+    st.dataframe(
+        st.session_state.df2,
+        column_config={
+                "Name": st.column_config.Column(width="medium"),
+                "Save": st.column_config.Column(width="small"),
+                "Url": st.column_config.LinkColumn(width="large")
+            },
+        hide_index=True)
 st.button(label='Clear Saved', on_click=clear_saved)