Update app.py

app.py

@@ -1,84 +1,40 @@
-# First, make sure to install the openpyxl package:
-# !pip install openpyxl
-
 import streamlit as st
 import pandas as pd
-from datetime import timedelta
 
-# Function to parse the uploaded file and return a DataFrame
+# Function to parse the uploaded roster file
 def parse_roster(file):
-    df = pd.read_excel(file)
+    df = pd.read_excel(file, engine='openpyxl')
+    st.write("File read successfully. Columns in the file:", df.columns)
     return df
 
-# Function to generate the next one-year roster
+# Function to generate the yearly roster
 def generate_yearly_roster(df):
-    days_in_year = 365
-    new_roster = []
     nurses = df['NAMES'].unique()
-
-    for nurse in nurses:
-        current_day = 0
-        nurse_schedule = []
-
-        while current_day < days_in_year:
-            if nurse == "ADOMAKO SIMON" or nurse == "MAHAMA RAHMAT":
-                shift_pattern = ['M', 'M', 'M', 'M', 'M', 'X', 'X']
-            else:
-                shift_pattern = ['M', 'A', 'N', 'N', 'N', 'N', 'N', 'X', 'X', 'M', 'A', 'A', 'M', 'X', 'X']
-
-            for shift in shift_pattern:
-                if current_day >= days_in_year:
-                    break
-                nurse_schedule.append(shift)
-                current_day += 1
-
-        new_roster.append(nurse_schedule)
-
-    return pd.DataFrame(new_roster, index=nurses).T
-
-# Streamlit app layout
+    roles = df['RANK'].unique()
+    
+    # Your logic to generate the yearly roster goes here
+    yearly_roster = df.copy()  # Placeholder for demonstration
+    
+    return yearly_roster
+
+# Streamlit app setup
 st.title("Nurse Duty Roster Generator")
 
-st.markdown("""
-This app generates a one-year duty roster for nurses based on the current month's schedule.
-* **Upload**: Upload the current month's roster in Excel format.
-* **Generate**: Generate the duty roster for the next one year.
-* **Modify**: Make modifications to the generated roster.
-""")
+uploaded_file = st.file_uploader("Upload the current roster file", type=["xlsx"])
 
-uploaded_file = st.file_uploader("Upload current month's roster", type=['xlsx'])
-
-if uploaded_file:
-    current_roster = parse_roster(uploaded_file)
-    st.write("### Current Month's Roster")
-    st.dataframe(current_roster)
-
-    generate_button = st.button("Generate One-Year Roster")
-    if generate_button:
+if uploaded_file is not None:
+    try:
+        current_roster = parse_roster(uploaded_file)
         one_year_roster = generate_yearly_roster(current_roster)
-        st.write("### One-Year Duty Roster")
+        
+        st.write("Generated Yearly Roster:")
         st.dataframe(one_year_roster)
-
-        # Option to download the generated roster
-        @st.cache
-        def convert_df(df):
-            return df.to_csv(index=False).encode('utf-8')
-
-        csv = convert_df(one_year_roster)
-
+        
+        # Allow download of the generated roster
         st.download_button(
-            label="Download roster as CSV",
-            data=csv,
-            file_name='one_year_roster.csv',
-            mime='text/csv',
+            label="Download Yearly Roster",
+            data=one_year_roster.to_excel(index=False),
+            file_name="yearly_roster.xlsx"
         )
-
-# Save changes and regenerate
-st.write("### Modify and Regenerate Roster")
-edit_roster = st.text_area("Edit the roster here (CSV format):", height=300)
-
-if st.button("Save Changes and Regenerate"):
-    edited_df = pd.read_csv(pd.compat.StringIO(edit_roster))
-    one_year_roster = generate_yearly_roster(edited_df)
-    st.write("### Updated One-Year Duty Roster")
-    st.dataframe(one_year_roster)
+    except Exception as e:
+        st.error(f"An error occurred: {e}")