Update app.py

app.py

@@ -95,12 +95,12 @@ st.markdown("""
 
 # Initialize generator
 #@st.cache_resource
-def get_generator():
-    return LessonPlanGenerator()
+#def get_generator():
+    #return LessonPlanGenerator()
 
 
-#generator = LessonPlanGenerator()
-generator = get_generator()
+generator = LessonPlanGenerator()
+
 # Header
 st.title("WizLab Lesson Plan Generator")
 st.markdown("We help you generate lesson plans tailored to your needs.")
@@ -150,6 +150,17 @@ with col1:
         # Use mapping if available, otherwise use the base language name
         language = LANGUAGE_MAPPING.get(language_selection, language_selection)
 
+    skills_section = ["Reading", "Writing", "Listening", "Speaking", "Other"]
+
+    skill_focus = st.multiselect("Skills to focus", skills_section)
+
+    if "Other" in skill_focus:
+        other_skills = st.text_input("Specify Skills to focus")
+        skill_focus.remove("Other")  # Remove 'Other' placeholder from the list
+        if other_skills:
+            skill_focus.append(other_skills)  # Add user-specified skills
+
+    #st.write("Selected Skills:", skill_focus)
     #if language_selection == "Other":
        # language = st.text_input("Specify Language")
   #  else:
@@ -164,16 +175,19 @@ with col2:
     #else:
     #    duration = duration_selection
     duration = st.text_input("Duration [in minutes]", placeholder = "e.g., 25 minutes")
-    proficiency_options = ["Beginner [A1]", "Elementary [A2]", "Intermediate[B1, B2]", "Advanced[C1]", "Proficient [C2]", "Other"]
+    proficiency_options = ["Beginner [A1]", "Elementary [A2]", "Intermediate[B1]", "Upper-Intermediate [B2]", "Advanced[C1]", "Proficient [C2]", "Other"]
     proficiency_selection = st.selectbox("Proficiency Level [CEFR]", proficiency_options)
     if proficiency_selection == "Other":
         proficiency = st.text_input("Specify Proficiency Level")
     else:
         proficiency = proficiency_selection
 
-tech_options = ["Interactive Whiteboard", "Computers/Laptops",
-                "Mobile Devices", "Internet Access", "None", "Other"]
-tech_selection = st.multiselect("Accessible Tech Resources", tech_options)
+    tech_options = ["Interactive Whiteboard", "Computers/Laptops",
+                    "Mobile Devices", "Internet Access", "None", "Other"]
+    tech_selection = st.multiselect("Accessible Tech Resources", tech_options)
+
+    num_students = st.text_input("Number of students", placeholder= "e.g., 30 students")
+
 
 if "Other" in tech_selection:
     other_tech = st.text_input("Specify Other Technology Requirements")
@@ -259,8 +273,10 @@ if st.button("Generate Lesson Plan", type="primary"):
                 language = "English"
 
             detailed_prompt = f"""
-                        Create a lesson plan for teaching '{topic}' to {age_group} students.
-                        Duration: {duration} minutes,
+                        Create a lesson plan for teaching '{topic}' to {age_group} students, for a 
+                        Duration: {duration} minutes, 
+                        Number of students in the class: {num_students},
+                        SKills to focus in generating lesson plan: {skill_focus}
                         Proficiency Level [CEFR]: {proficiency}, 
                         Technology Requirements: {', '.join(tech_usage) if tech_usage else 'None'}, 
                         Generate response in {language} Language.