Update app.py

updated the streamlit for changes made for lesson plan

app.py

@@ -1,7 +1,14 @@
 import streamlit as st
 import json
+import os
+from dotenv import load_dotenv
+from typing import Dict, Any
 
-from openai_llm import LessonPlanGenerator
+# Import the LLM Handler directly
+from llms_handler import LLMHandler
+
+# Load environment variables for OpenAI API key
+load_dotenv()
 
 # Page configuration
 st.set_page_config(
@@ -93,13 +100,14 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-# Initialize generator
-#@st.cache_resource
-#def get_generator():
-    #return LessonPlanGenerator()
+# Initialize the LLM Handler (using Streamlit caching)
+@st.cache_resource
+def get_llm_handler():
+    """Create and cache the LLM Handler."""
+    return LLMHandler()
 
 
-generator = LessonPlanGenerator()
+llm_handler = get_llm_handler()
 
 # Header
 st.title("WizLab Lesson Plan Generator")
@@ -112,7 +120,8 @@ st.header("Input Your Requirements")
 col1, col2 = st.columns(2)
 
 with col1:
-    topic = st.text_input("Topic", placeholder="e.g., Lesson Plan to teach Spanish numbers as third language to English Speaking students ")
+    topic = st.text_input("Topic",
+                          placeholder="e.g., Lesson Plan to teach Spanish numbers as third language to English Speaking students ")
 
     age_group_options = ["Elementary (6-11)", "Middle School (12-14)",
                          "High School (15-18)", "Adult Learners", "Other"]
@@ -123,32 +132,23 @@ with col1:
         age_group = age_group_selection
 
     language_options = [
-        "English", "日本語 (Japanese)", "中文 (Mandarin Chinese)",
-        "Español (Spanish)", "हिंदी (Hindi)", "한국어 (Korean)",
-        "Deutsch (German)", "Français (French)", "Português (Portuguese)",
-        "العربية (Arabic)", "Русский (Russian)", "Other"
+        "English", "中文 (Mandarin Chinese)"
+
     ]
 
     language_selection = st.selectbox("Select Output Language", language_options)
     LANGUAGE_MAPPING = {
-        "日本語 (Japanese)": "Japanese",
-        "中文 (Mandarin Chinese)": "Mandarin Chinese",
-        "Español (Spanish)": "Spanish",
-        "हिंदी (Hindi)": "Hindi",
-        "한국어 (Korean)": "Korean",
-        "Deutsch (German)": "German",
-        "Français (French)": "French",
-        "Português (Portuguese)": "Portuguese",
-        "العربية (Arabic)": "Arabic",
-        "Русский (Russian)": "Russian"
+
+        "中文 (Mandarin Chinese)": "Mandarin Chinese"
+
     }
 
     # Then in your language selection code:
-    if language_selection == "Other":
-        language = st.text_input("Specify Language")
-    else:
+   # if language_selection == "Other":
+    #    language = st.text_input("Specify Language")
+    #else:
         # Use mapping if available, otherwise use the base language name
-        language = LANGUAGE_MAPPING.get(language_selection, language_selection)
+    language = LANGUAGE_MAPPING.get(language_selection, language_selection)
 
     skills_section = ["Reading", "Writing", "Listening", "Speaking", "Other"]
 
@@ -160,22 +160,10 @@ with col1:
         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:
-        # Extract the base language name without the native script
-       # language = language_selection.split(" (")[0] if " (" in language_selection else language_selection
-
 with col2:
-    #duration_options = ["30 minutes", "45 minutes", "60 minutes", "90 minutes", "Other"]
-    #duration_selection = st.selectbox("Lesson Duration", duration_options)
-    #if duration_selection == "Other":
-    #   duration = st.text_input("Specify Duration")
-    #else:
-    #    duration = duration_selection
-    duration = st.text_input("Duration [in minutes]", placeholder = "e.g., 25 minutes")
-    proficiency_options = ["Beginner [A1]", "Elementary [A2]", "Intermediate[B1]", "Upper-Intermediate [B2]", "Advanced[C1]", "Proficient [C2]", "Other"]
+    duration = st.text_input("Duration [in minutes]", placeholder="e.g., 25 minutes")
+    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")
@@ -186,8 +174,7 @@ with col2:
                     "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")
-
+    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")
@@ -253,13 +240,12 @@ def format_content(data):
     return html_content
 
 
-# Generate button
 # Generate button
 if st.button("Generate Lesson Plan", type="primary"):
     if not topic:
         st.error("Please enter a topic for the lesson plan.")
     else:
-        with st.spinner("Generating your lesson plan..."):
+        with st.spinner("Generating your lesson plan... This may take a minute or two."):
             # Set default values if not provided
             if not duration:
                 duration = "30 minutes"
@@ -273,50 +259,69 @@ if st.button("Generate Lesson Plan", type="primary"):
                 language = "English"
 
             detailed_prompt = f"""
-                        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.
-                        """
+                Create a lesson plan for teaching '{topic}' to {age_group} students, for a 
+                Duration: {duration}, 
+                Number of students in the class: {num_students},
+                Skills to focus in generating lesson plan: {', '.join(skill_focus) if skill_focus else 'All language skills'},
+                Proficiency Level [CEFR]: {proficiency}, 
+                Technology Requirements: {', '.join(tech_usage) if tech_usage else 'None'}, 
+                Generate response in {language} Language.
+                """
+
+            # Add debug output to see what's happening
+            progress_placeholder = st.empty()
 
             try:
-                # Pass both the prompt and language
-                result = generator.generate_lesson_plan(detailed_prompt, language)
-
-                # Consistent parsing regardless of language
-                if isinstance(result, dict):
-                    if "lessonPlan" in result:
-                        lesson_content = result["lessonPlan"]
-                    else:
-                        lesson_content = result
-
-                    st.success("Lesson plan generated successfully!")
-
-                    # Create tabs for different views
-                    tab1, tab2 = st.tabs(["Formatted View", "Raw JSON"])
-
-                    with tab1:
-                        st.markdown(format_content(lesson_content), unsafe_allow_html=True)
-
-                    with tab2:
-                        st.json(lesson_content)
-
-                    # Download button
-                    st.download_button(
-                        label="Download Lesson Plan",
-                        data=json.dumps(lesson_content, indent=2, ensure_ascii=False),
-                        # ensure_ascii=False for proper Unicode handling
-                        file_name=f"lesson_plan_{language}.json",
-                        mime="application/json"
-                    )
-                else:
-                    st.error("Invalid response format")
+                # Direct integration with LLM Handler
+
+                # Step 1: Extract and validate input parameters
+                progress_placeholder.text("Step 1/4: Extracting and validating inputs...")
+                input_data = llm_handler.input_extraction_validation(detailed_prompt)
+                progress_placeholder.text("Step 1/4: Input extraction completed")
+
+                # Step 2: Determine lesson plan structure
+                progress_placeholder.text("Step 2/4: Determining lesson structure...")
+                structure_info = llm_handler.lesson_plan_structure(input_data)
+                progress_placeholder.text("Step 2/4: Structure determination completed")
+
+                # Step 3: Select activity templates
+                progress_placeholder.text("Step 3/4: Selecting activity templates...")
+                activity_info = llm_handler.activity_template_selection(input_data, structure_info)
+                progress_placeholder.text("Step 3/4: Activity template selection completed")
+
+                # Step 4: Generate lesson plan
+                progress_placeholder.text("Step 4/4: Generating final lesson plan...")
+                lesson_plan = llm_handler.lesson_plan_generation(input_data, structure_info, activity_info)
+                progress_placeholder.text("Step 4/4: Lesson plan generation completed")
+
+                # Clear the progress placeholder
+                progress_placeholder.empty()
+
+                # Notify success
+                st.success("Lesson plan generated successfully!")
+
+                # Create tabs for different views
+                tab1, tab2 = st.tabs(["Formatted View", "Raw JSON"])
+
+                with tab1:
+                    st.markdown(format_content(lesson_plan), unsafe_allow_html=True)
+
+                with tab2:
+                    st.json(lesson_plan)
+
+                # Download button
+                st.download_button(
+                    label="Download Lesson Plan",
+                    data=json.dumps(lesson_plan, indent=2, ensure_ascii=False),
+                    # ensure_ascii=False for proper Unicode handling
+                    file_name=f"lesson_plan_{language}.json",
+                    mime="application/json"
+                )
 
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
+                st.error("Please check if your OpenAI API key is valid and has sufficient credits.")
 
 # Footer
 st.markdown("---")
+st.markdown("© WizLab Lesson Plan Generator | powered by AI")