Update app.py

app.py

@@ -1,40 +1,81 @@
 import streamlit as st
 import pandas as pd
+from datetime import timedelta
 
-# Function to parse the uploaded roster file
+# Function to parse the uploaded file and return a DataFrame
 def parse_roster(file):
-    df = pd.read_excel(file, engine='openpyxl')
-    st.write("File read successfully. Columns in the file:", df.columns)
+    df = pd.read_excel(file)
     return df
 
-# Function to generate the yearly roster
+# Function to generate the next one-year roster
 def generate_yearly_roster(df):
+    days_in_year = 365
+    new_roster = []
     nurses = df['NAMES'].unique()
-    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
+
+    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
 st.title("Nurse Duty Roster Generator")
 
-uploaded_file = st.file_uploader("Upload the current roster file", type=["xlsx"])
+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 current month's roster", type=['xlsx'])
 
-if uploaded_file is not None:
-    try:
-        current_roster = parse_roster(uploaded_file)
+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:
         one_year_roster = generate_yearly_roster(current_roster)
-        
-        st.write("Generated Yearly Roster:")
+        st.write("### One-Year Duty Roster")
         st.dataframe(one_year_roster)
-        
-        # Allow download of the generated 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)
+
         st.download_button(
-            label="Download Yearly Roster",
-            data=one_year_roster.to_excel(index=False),
-            file_name="yearly_roster.xlsx"
+            label="Download roster as CSV",
+            data=csv,
+            file_name='one_year_roster.csv',
+            mime='text/csv',
         )
-    except Exception as e:
-        st.error(f"An error occurred: {e}")
+
+# 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)