def main():
  st.title("Gantt Chart Generator")

  project_name = st.text_input("Project Name")
  project_duration = st.number_input("Project Duration (days)", min_value=1)

  activities = []
  num_activities = st.number_input("Number of Activities", min_value=1)

  for i in range(num_activities):
    activity_name = st.text_input(f"Activity {i+1} Name")
    start_date = st.number_input(f"Activity {i+1} Start Date (day)", min_value=0, max_value=project_duration)
    end_date = st.number_input(f"Activity {i+1} End Date (day)", min_value=start_date, max_value=project_duration)
    resource = st.text_input(f"Activity {i+1} Resource") 
    activities.append({
        'Task': activity_name,
        'Start': start_date,
        'Finish': end_date,
        'Resource': resource
    })

  if st.button("Generate Gantt Chart"):
    fig = create_gantt_chart(project_name, project_duration, activities)
    st.plotly_chart(fig)

if __name__ == "__main__":
  main()