Update app.py

app.py

@@ -1,5 +1,5 @@
-import streamlit as st
 import pandas as pd
+import streamlit as st
 from datasets import load_dataset
 
 
@@ -10,12 +10,39 @@ def load_data():
     df = pd.DataFrame(dataset)
     return df
 
+
+# Function to process user input (date) and return description and reference
+def process_date(day, month, year=None):
+    # Filter data based on selected date
+    if year is None or year == "":
+        filtered_data = df[(df["month"] == month) & (df["day"] == int(day))]
+    else:
+        filtered_data = df[
+            (df["year"] == int(year)) & (df["month"] == month) & (df["day"] == int(day))
+        ]
+
+    if not filtered_data.empty:
+        # Prepare empty lists to store descriptions and references
+        descriptions = []
+        references = []
+
+        # Loop through filtered data and append descriptions and references
+        for index, row in filtered_data.iterrows():
+            descriptions.append(row["event_description"])
+            references.append(row["reference"])
+
+        # Return lists of descriptions and references
+        return descriptions
+    else:
+        return [f"No data found for selected date {year} {month} {day}"]
+
+
 def main():
     df = load_data()
     # Page title and header
     st.title("Day in History")
     with st.form("my_form"):
-        st.header("Select a date to view results")
+        st.header("Select a date")
         col1, col2, col3 = st.columns(3)
         # Datepicker
         with col1:
@@ -63,7 +90,7 @@ def main():
 
             for index, row in filtered_data.iterrows():
                 container = st.container(border=True)
-                container.title(f"{row['year']} {row['month']} {row['day']}")
+                container.subheader(f"{row['year']} {row['month']} {row['day']}")
                 container.markdown(
                     f"{row['event_description']}", unsafe_allow_html=True
                 )