Update app.py

app.py

@@ -1,12 +1,13 @@
 import pandas as pd
 import streamlit as st
 from datasets import load_dataset
+from datetime import datetime
 
 
 # Load data
 @st.cache_data  # Cache data for performance improvement
-def load_data():
-    dataset = load_dataset("santhosh/day_in_history", "en", split="train")
+def load_data(language="en"):
+    dataset = load_dataset("santhosh/day_in_history", language, split="train")
     df = pd.DataFrame(dataset)
     return df
 
@@ -36,35 +37,49 @@ def process_date(day, month, year=None):
     else:
         return [f"No data found for selected date {year} {month} {day}"]
 
+supported_languages = {
+    "en": "English",
+    "ml": "Malayalam",
+}
+MONTHS = [
+    "January",
+    "February",
+    "March",
+    "April",
+    "May",
+    "June",
+    "July",
+    "August",
+    "September",
+    "October",
+    "November",
+    "December",
+]
+
 
 def main():
-    df = load_data()
     # Page title and header
     st.title("Day in History")
     with st.form("my_form"):
         st.header("Select a date")
+        selected_language = st.selectbox(
+            "Select language",
+            supported_languages.keys(),
+            format_func=lambda x: supported_languages[x],
+            index=0,
+        )
         col1, col2, col3 = st.columns(3)
         # Datepicker
         with col1:
-            selected_day = st.selectbox("Select Day", range(1, 32), index=14)
+            selected_day = st.selectbox(
+                "Select Day", range(1, 32), index=datetime.now().day - 1
+            )
         with col2:
             selected_month = st.selectbox(
                 "Select Month",
-                [
-                    "January",
-                    "February",
-                    "March",
-                    "April",
-                    "May",
-                    "June",
-                    "July",
-                    "August",
-                    "September",
-                    "October",
-                    "November",
-                    "December",
-                ],
-                index=7,
+                range(0, 12),
+                format_func=lambda x: MONTHS[x],
+                index=datetime.now().month - 1,
             )
         with col3:
             selected_year = st.number_input(
@@ -72,15 +87,17 @@ def main():
             )
         submitted = st.form_submit_button("Submit")
     if submitted:
+        df = load_data(selected_language)
         # Process data based on selected date
         if selected_year is None or selected_year == "":
             filtered_data = df[
-                (df["month"] == selected_month) & (df["day"] == int(selected_day))
+                (df["month"] == int(selected_month + 1))
+                & (df["day"] == int(selected_day))
             ]
         else:
             filtered_data = df[
                 (df["year"] == int(selected_year))
-                & (df["month"] == selected_month)
+                & (df["month"] == selected_month + 1)
                 & (df["day"] == int(selected_day))
             ]
 
@@ -90,7 +107,9 @@ def main():
 
             for index, row in filtered_data.iterrows():
                 container = st.container(border=True)
-                container.subheader(f"{row['year']} {row['month']} {row['day']}")
+                container.subheader(
+                    f"{row['year']} {MONTHS[row['month']-1]} {row['day']}"
+                )
                 container.markdown(
                     f"{row['event_description']}", unsafe_allow_html=True
                 )