Xpolymer123 commited on
Commit
bf51e2c
·
verified ·
1 Parent(s): f2aa6f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -19,7 +19,7 @@ end_date = st.date_input("Project End Date:", value=datetime.today() + timedelta
19
 
20
  num_activities = st.number_input("Number of Activities (Minimum 1):", min_value=1, value=5)
21
 
22
- activities = []
23
  for i in range(num_activities):
24
  col1, col2 = st.columns(2)
25
  with col1:
@@ -28,9 +28,19 @@ for i in range(num_activities):
28
  activity_duration = st.number_input(f"Activity {i+1} Duration (days):", min_value=1, key=f"duration_{i}")
29
  activities.append({"Task": activity_name, "Duration": activity_duration})
30
 
31
- if st.button("Create Gantt Chart"):
32
  if not project_name:
33
  st.error("Please enter a project name.")
34
  elif any(not activity["Task"] for activity in activities):
35
  st.error("Please enter names for all activities.")
36
- # Corrected line
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  num_activities = st.number_input("Number of Activities (Minimum 1):", min_value=1, value=5)
21
 
22
+ activities =
23
  for i in range(num_activities):
24
  col1, col2 = st.columns(2)
25
  with col1:
 
28
  activity_duration = st.number_input(f"Activity {i+1} Duration (days):", min_value=1, key=f"duration_{i}")
29
  activities.append({"Task": activity_name, "Duration": activity_duration})
30
 
31
+ if st.button("Create Gantt Chart"): # The button is now correctly placed
32
  if not project_name:
33
  st.error("Please enter a project name.")
34
  elif any(not activity["Task"] for activity in activities):
35
  st.error("Please enter names for all activities.")
36
+ elif start_date > end_date:
37
+ st.error("Start date cannot be after end date.")
38
+ else:
39
+ try:
40
+ df = pd.DataFrame(activities)
41
+ df['Start'] = pd.to_datetime(start_date)
42
+ df['Finish'] = df['Start']
43
+
44
+ for i in range(len(df)):
45
+ if i > 0:
46
+ df.loc[i, 'Start'] = df.