Update app.py

app.py

@@ -11,13 +11,13 @@ def generate_random_color():
     return f'rgb({r}, {g}, {b})'
 
 st.set_page_config(layout="wide")
-
 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())
+end_date = st.date_input("Project End Date:", value=datetime.today() + timedelta(days=30))  # Default 30 days ahead
 
-num_activities = st.number_input("Number of Activities (Minimum 1):", min_value=1, value=5)  # No maximum limit
+num_activities = st.number_input("Number of Activities (Minimum 1):", min_value=1, value=5)
 
 activities =
 for i in range(num_activities):
@@ -25,9 +25,30 @@ for i in range(num_activities):
     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:", min_value=1, key=f"duration_{i}")
-        duration_unit = st.selectbox(f"Duration Unit for Activity {i+1}", options=["Days", "Weeks", "Months"], key=f"unit_{i}")
-    activities.append({"Task": activity_name, "Duration": activity_duration, "Unit": duration_unit})
+        activity_duration = st.number_input(f"Activity {i+1} Duration (days):", min_value=1, key=f"duration_{i}")
+    activities.append({"Task": activity_name, "Duration": activity_duration})
+
+if st.button("Create Gantt Chart"):
+    if not project_name:
+        st.error("Please enter a project name.")
+    elif any(not activity["Task"] for activity in activities):
+        st.error("Please enter names for all activities.")
+    elif start_date > end_date:
+        st.error("Start date cannot be after end date.")
+    else:
+        df = pd.DataFrame(activities)
+        df['Start'] = pd.to_datetime(start_date)
+        df['Finish'] = df['Start']
+
+        for i in range(len(df)):
+            if i > 0:
+                df.loc[i, 'Start'] = df.loc[i - 1, 'Finish']
+            df.loc[i, 'Finish'] = df.loc[i, 'Start'] + pd.to_timedelta(df.loc[i, 'Duration'], unit='D')
+
+        df['Start_str'] = df['Start'].dt.strftime('%Y-%m-%d')
+        df['Finish_str'] = df['Finish'].dt.strftime('%Y-%m-%d')
+        
 
-if st.button("Create Gantt Chart"): #Button is now correctly placed
-    if
+        data =
+        for index, row in df.iterrows():
+            data.append(dict(Task=row['Task'], Start=row['Start