Update app.py

app.py

@@ -10,15 +10,24 @@ 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")
 
 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)
+num_activities = st.number_input("Number of Activities (Minimum 1):", min_value=1, value=5)  # No maximum limit
 
-activities = []
+activities =
 for i in range(num_activities):
-    col1, col2 = st.columns(2)
+    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:", 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})
+
+if st.button("Create Gantt Chart"): #Button is now correctly placed
+    if