Update app.py

app.py

@@ -14,10 +14,9 @@ def write_sidebar():
     st.sidebar.title("Instructions")
     st.sidebar.markdown("**1. Select the student's grade level.**")
     st.sidebar.markdown("**2. Select the student's qualifying condition(s).**")
-    st.sidebar.markdown("**3. Choose a category to analyze.**")
-    st.sidebar.markdown("**4. Select whether you want to generate a PLAAFP statement or an effective IEP goal.**")
-    st.sidebar.markdown("**5. Enter your student data.**")
-    st.sidebar.markdown("**6. Click the 'Generate' button to generate the selected output.**")
+    st.sidebar.markdown("**3. Choose a prompt category and prompt to analyze.**")
+    st.sidebar.markdown("**4. Enter your student data.**")
+    st.sidebar.markdown("**5. Click the 'Generate' button to generate the selected output.**")
 
     st.sidebar.write("")
     st.sidebar.write("")
@@ -57,15 +56,9 @@ def write_iep_assist():
         ]
     }
 
-    # Choose a category
-    st.markdown("<h3>Choose a category to analyze:</h3>", unsafe_allow_html=True)
-    selected_category = st.selectbox("Category:", options=list(prompt_categories.keys()), key="category")
-
-    # Choose the output type
-    generate_goal = st.checkbox("Generate Effective IEP Goal")
-
-    # Choose a prompt from the selected category
-    st.markdown("<h3>Choose a prompt:</h3>", unsafe_allow_html=True)
+    # Choose a category and prompt
+    st.markdown("<h3>Choose a prompt category and prompt:</h3>", unsafe_allow_html=True)
+    selected_category = st.selectbox("Category:", options=["Behavior", "Academic", "IEP Goals"], key="category")
     prompts = prompt_categories[selected_category]
     selected_prompt = st.selectbox("Prompt:", options=prompts, key="prompt")
 
@@ -74,8 +67,10 @@ def write_iep_assist():
     student_data = st.text_area("Paste student data here", height=250, key="student-data")
 
     # Add a button to generate the PLAAFP statement or effective IEP goal
-    if st.button("Generate", key="generate-button"):
-        if generate_goal:
+    generate_button = st.button("Generate", key="generate-button", help="Click here to generate the selected output.")
+
+    if generate_button:
+        if selected_category == "IEP Goals":
             # Define the characteristics of an effective IEP goal
             effective_goal_characteristics = "Measurable, Specific and Clear, Attainable and Realistic, Relevant and Meaningful, Time-Bound, Individualized, Action-Oriented, Aligned with Standards and Curriculum, Collaboratively Developed, Monitored and Adjusted"