Update app.py

app.py

@@ -17,6 +17,8 @@ os.environ["TRAINER_TELEMETRY"]= "0"
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 BASETEN_API = os.environ.get("BASETEN_API", None)
 BASETEN_KEY = os.environ.get("BASETEN_KEY", None)
+NOTION_KEY = os.environ.get("NOTION_API_KEY", None)
+NOTION_DB_ID = os.environ.get("NOTION_DB_ID", None)
 
 st.set_page_config(layout="wide")
 # Load custom CSS to integrate Bootstrap, Font Awesome, and Google Fonts
@@ -184,6 +186,64 @@ with right:
 
     # End of Box 4 and fourth Carousel Item
 
+left, middle, right = st.columns([1,3,1])
+
+with middle:
+    # Streamlit form for data input
+    with st.form(key='data_entry_form'):
+        name = st.text_input("Name")
+        age = st.number_input("Age", min_value=0)
+        location = st.text_input("Location")
+        submit_button = st.form_submit_button(label='Submit')
+    # Function to add data to Notion
+    def add_to_notion(name, age, location):
+        new_page = {
+            "Name": {
+                "title": [
+                    {
+                        "text": {
+                            "content": name
+                        }
+                    }
+                ]
+            },
+            "Age": {
+                "number": age
+            },
+            "Location": {
+                "rich_text": [
+                    {
+                        "text": {
+                            "content": location
+                        }
+                    }
+                ]
+            }
+        }    
+        notion.pages.create(parent={"database_id": database_id}, properties=new_page)
+    
+    # Add data to Notion when form is submitted
+    if submit_button:
+        add_to_notion(name, age, location)
+        st.success("Data submitted to Notion!")
+    
+    # Function to retrieve data from Notion
+    def retrieve_data_from_notion():
+        query_result = notion.databases.query(database_id=database_id)
+        data = []
+        for result in query_result["results"]:
+            name = result["properties"]["Name"]["title"][0]["text"]["content"]
+            age = result["properties"]["Age"]["number"]
+            location = result["properties"]["Location"]["rich_text"][0]["text"]["content"]
+            data.append({"Name": name, "Age": age, "Location": location})
+        
+        return pd.DataFrame(data)
+    
+    # Display the data in a table
+    st.subheader("Stored Data")
+    data_df = retrieve_data_from_notion()
+    st.table(data_df)
+
 hide_default_format = """
        <style>
        #MainMenu {visibility: hidden; }