Update app.py

app.py

@@ -10,19 +10,22 @@ def generate_random_color():
     b = random.randint(0, 255)
     return f'rgb({r}, {g}, {b})'
 
+st.set_page_config(layout="wide") # Make the layout wider
+
 st.title("Project Schedule/Gantt Chart Generator")
 
 project_name = st.text_input("Project Name:")
 start_date = st.date_input("Project Start Date:", value=datetime.today())
+
 num_activities = st.number_input("Number of Activities (5-20):", min_value=5, max_value=20, value=5)
 
-activities = []  # Initialize an empty list to store activity data
+activities =
 for i in range(num_activities):
     col1, col2 = st.columns(2)
     with col1:
         activity_name = st.text_input(f"Activity {i+1} Name:", key=f"name_{i}")
     with col2:
-        activity_duration = st.number_input(f"Activity {i+1} Duration (weeks):", min_value=1, key=f"duration_{i}")  # Duration in weeks
+        activity_duration = st.number_input(f"Activity {i+1} Duration (weeks):", min_value=1, key=f"duration_{i}")
     activities.append({"Task": activity_name, "Duration": activity_duration})
 
 if st.button("Create Gantt Chart"):
@@ -36,4 +39,20 @@ if st.button("Create Gantt Chart"):
         df['Finish'] = df['Start']
         for i in range(len(df)):
             if i > 0:
-                df.loc[i, 'Start'] = df.loc
+                df.loc[i, 'Start'] = df.loc[i-1, 'Finish']
+            df.loc[i, 'Finish'] = df.loc[i, 'Start'] + pd.to_timedelta(df.loc[i, 'Duration'] * 7, unit='D')
+
+        df['Start_str'] = df['Start'].dt.strftime('%Y-%m-%d')
+        df['Finish_str'] = df['Finish'].dt.strftime('%Y-%m-%d')
+        df['Start_week'] = df['Start'].dt.to_period('W-SUN').dt.start_time.dt.strftime('%Y-%m-%d')
+        df['Finish_week'] = df['Finish'].dt.to_period('W-SUN').dt.end_time.dt.strftime('%Y-%m-%d')
+        
+        project_end_date = df['Finish'].max().strftime('%Y-%m-%d') #Calculate project end date
+        st.write(f"Project End Date: {project_end_date}") #Display project end date
+
+        data =
+        for index, row in df.iterrows():
+            data.append(dict(Task=row['Task'], Start=row['Start_week'], Finish=row['Finish_week']))
+
+        fig = ff.create_gantt(data, colors=[generate_random_color() for _ in range(len(df))],
+                              index_col='