Update app.py

app.py

@@ -10,7 +10,7 @@ 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.set_page_config(layout="wide")  # Make the layout wider
 
 st.title("Project Schedule/Gantt Chart Generator")
 
@@ -19,7 +19,7 @@ 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 =
+activities = []
 for i in range(num_activities):
     col1, col2 = st.columns(2)
     with col1:
@@ -46,13 +46,9 @@ if st.button("Create Gantt Chart"):
         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']))
+        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
 
-        fig = ff.create_gantt(data, colors=[generate_random_color() for _ in range(len(df))],
-                              index_col='
+        data = []
+        for index, row in