Update app.py

app.py

@@ -14,14 +14,15 @@ def write_sidebar():
     st.sidebar.title("Instructions")
     st.sidebar.write("1. Select the student's grade level.")
     st.sidebar.write("2. Select the student's qualifying condition(s).")
-    st.sidebar.write("3. Choose a prompt based to communicate with the AI how you want your data analyzed.")
-    st.sidebar.write("4. Enter your student data.")
-    st.sidebar.write("5. Click the 'Analyze Student Data' button to generate a summary of your data.")
+    st.sidebar.write("3. Choose a category to analyze.")
+    st.sidebar.write("4. Select whether you want to generate a PLAAFP statement or an effective IEP goal.")
+    st.sidebar.write("5. Enter your student data.")
+    st.sidebar.write("6. Click the 'Generate' button to generate the selected output.")
 
     st.sidebar.write("")
     st.sidebar.write("")
 
-    st.sidebar.write("Note: This app uses OpenAI's GPT-3 API to generate the PLAAFP statement. Please enter data that is relevant and appropriate for generating the statement.")
+    st.sidebar.write("Note: This app uses OpenAI's GPT-3 API to generate the PLAAFP statement or IEP goal. Please enter data that is relevant and appropriate for generating the output.")
 
 def write_iep_assist():
     st.title("IEP Assist Premium")
@@ -34,67 +35,78 @@ def write_iep_assist():
     st.write("Select the student's qualifying condition(s):")
     qualifying_condition = st.multiselect("Qualifying Condition(s):", ["Specific Learning Disability", "Emotional Disturbance", "Autism", "Intellectual Disability", "Speech/Language Impairment", "Other Health Impairment", "Orthopedic Impairment", "Auditory Impairment", "Traumatic Brain Injury", "Deafness", "Blindness", "Developmental Delay"], key="qualifying-condition")
 
-    # Choose a prompt
-    st.write("Choose a prompt:(This tells the AI how you want your data analyzed)")
-    prompts = [
-        "Analyze the data provided on the student and provide a summary of their strengths and areas of need in regards to their academic performance.",
-        "Provide a summary of the student's behavior data and suggest possible interventions to try based on their areas of need.",
-        "Summarize the data provided on the student's academic performance, highlighting their strengths and areas of need, and suggesting possible interventions to try.",
-        "Please provide a summary of the student's academic performance, highlighting their strengths and areas of need.",
-        "What is the student's biggest strength and area of need in regards to their academic performance?",
-        "Analyze the data provided on the student and provide a summary of their progress in regards to their IEP goals.",
-        "Based on the student's academic performance data, what recommendations do you have for adjusting their instructional strategies?",
-        "How can the student's strengths be leveraged to help them improve in areas of need?",
-        "What barriers to academic success does the student face, and how can they be addressed?",
-        "Analyze the student's behavior data to identify trends and suggest possible interventions.",
-        "Provide a summary of the student's progress towards their academic and behavioral goals."
-    ]
+    # Define prompts by category
+    prompt_categories = {
+        "Behavior": [
+            "Provide a summary of the student's behavior data and suggest possible interventions to try based on their areas of need.",
+            "Analyze the student's behavior data to identify trends and suggest possible interventions.",
+            "Provide a summary of the student's progress towards their behavioral goals."
+        ],
+        "Academic": [
+            "Analyze the data provided on the student and provide a summary of their strengths and areas of need in regards to their academic performance.",
+            "Summarize the data provided on the student's academic performance, highlighting their strengths and areas of need, and suggesting possible interventions to try.",
+            "Please provide a summary of the student's academic performance, highlighting their strengths and areas of need.",
+            "What is the student's biggest strength and area of need in regards to their academic performance?",
+            "Based on the student's academic performance data, what recommendations do you have for adjusting their instructional strategies?",
+            "How can the student's strengths be leveraged to help them improve in areas of need?",
+            "What barriers to academic success does the student face, and how can they be addressed?"
+        ],
+        "IEP Goals": [
+            "Analyze the data provided on the student and provide a summary of their progress in regards to their IEP goals.",
+            "Provide a summary of the student's progress towards their academic and behavioral goals."
+        ]
+    }
+
+    # Choose a category
+    st.write("Choose a category to analyze:")
+    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.write("Choose a prompt:")
+    prompts = prompt_categories[selected_category]
     selected_prompt = st.selectbox("Prompt:", options=prompts, key="prompt")
 
+    # Enter student data to be analyzed
     st.write("Enter student data to be analyzed:")
     student_data = st.text_area("Paste student data here", height=250, key="student-data")
 
-    # Add a button to generate the PLAAFP statement
-    if st.button("Analyze Student Data", key="analyze-button"):
-        # Call the OpenAI API and generate a response
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=f"{selected_prompt} {student_data} {grade_level} {qualifying_condition}",
-            max_tokens=2000,
-            n=1,
-            stop=None,
-            temperature=0.9,
-        )
-
-        # Extract the generated statement from the API response
-        statement = response["choices"][0]["text"]
-
-        # Show the generated statement
-        st.write("Summary of Data entered:", statement)
-
-def write_iep_goal_generator():
-    st.title("IEP Goal Compass")
-
-    grade_level = st.selectbox("Select the student's grade level:", ["Pre-K", "K", "1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12"], key="grade_level")
-    qualifying_condition = st.multiselect("Select the qualifying condition(s):", ["Autism", "Emotional disturbance", "Intellectual disability", "Specific learning disability", "Other health impairment", "Speech or language impairment", "Deaf-blindness", "Deafness", "Blindness", "Multiple disabilities", "Orthopedic impairment", "Traumatic brain injury"], key="qualifying_condition")
-    teacher_input = st.text_area("Enter student data:", key="teacher_input")
-
-    # Generate IEP goal
-    if st.button("Generate Goal", key="generate_goal"):
-        # Call the OpenAI API and generate a goal
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=f"Generate an achievable and measurableable IEP goal for a student who is in grade {grade_level} and has qualifying conditions of {', '.join(qualifying_condition)} based on teacher reported information: {teacher_input}",
-            max_tokens=2000,
-            n=1,
-            stop=None,
-            temperature=0.85,
-        )
-        goal = response["choices"][0]["text"]
-        # Show the generated goal
-        st.write("IEP Goal:", goal)
+    # Add a button to generate the PLAAFP statement or effective IEP goal
+    if st.button("Generate", key="generate-button"):
+        if generate_goal:
+            # 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"
+
+            # Call the OpenAI API and generate an effective IEP goal
+            response = openai.Completion.create(
+                engine="text-davinci-003",
+                prompt=f"Generate an effective and measurable IEP goal for a student who is in grade {grade_level} and has qualifying conditions of {', '.join(qualifying_condition)}. The goal should be based on the analysis of their data and meet the following characteristics: {effective_goal_characteristics}. Data Analysis: {selected_prompt} {student_data}",
+                max_tokens=2000,
+                n=1,
+                stop=None,
+                temperature=0.85,
+            )
+            goal = response["choices"][0]["text"]
+
+            # Show the generated effective IEP goal
+            st.write("Effective IEP Goal:", goal)
+        else:
+            # Call the OpenAI API and generate a PLAAFP statement
+            response = openai.Completion.create(
+                engine="text-davinci-003",
+                prompt=f"{selected_prompt} {student_data} {grade_level} {qualifying_condition}",
+                max_tokens=2000,
+                n=1,
+                stop=None,
+                temperature=0.9,
+            )
+            statement = response["choices"][0]["text"]
+
+            # Show the generated PLAAFP statement
+            st.write("Summary of Data entered:", statement)
 
 if __name__ == "__main__":
     write_sidebar()
     write_iep_assist()
-    write_iep_goal_generator()