Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,273 +1,179 @@
+import altair as alt
 import streamlit as st
 import pandas as pd
 import os
 import zipfile
 from PIL import Image
+import sympy as sp
 import time
-import io
-import atexit
-import shutil
+import tempfile
 
-# -------------------- PAGE CONFIG --------------------
-st.set_page_config(
-    page_title="Smart Math Teacher",
-    page_icon="🧮",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+# Page config
+st.set_page_config(page_title="🎓 Smart Math Teacher", layout="centered")
 
-# -------------------- CUSTOM THEME --------------------
+# Custom CSS
 st.markdown("""
     <style>
-    .main-title {
-        font-size: 3rem;
-        font-weight: 700;
-        color: #FF4B4B;
-        text-align: center;
-        margin-bottom: 1rem;
-        text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
-    }
-    .question-box {
-        background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
-        padding: 2rem;
-        border-radius: 20px;
-        box-shadow: 0 10px 20px rgba(0,0,0,0.1);
-        margin: 1rem 0;
-        font-size: 1.2rem;
-        border-left: 5px solid #FF4B4B;
-    }
-    .hint-box {
-        background-color: #fff3cd;
-        border-left: 5px solid #ffc107;
-        padding: 1rem;
+    .fun-title { font-size: 40px; color: #ff3399; text-align: center; font-family: 'Comic Sans MS', cursive; }
+    .question-box { border: 4px dotted #ffcc00; padding: 20px; border-radius: 20px; background-color: #fff7e6; font-size: 20px; }
+    .stButton > button { font-size: 18px; background-color: #00cc99; color: white; border-radius: 10px; padding: 10px; }
+    .stTextInput > div > input { font-size: 18px; }
+    
+    /* Guide styling */
+    .guide-box {
+        background-color: #e8f5e8;
+        padding: 15px;
         border-radius: 10px;
-        margin: 1rem 0;
-    }
-    .stButton > button {
-        transition: all 0.3s ease;
-    }
-    .stButton > button:hover {
-        transform: translateY(-2px);
-        box-shadow: 0 5px 15px rgba(0,0,0,0.2);
+        border: 2px solid #4CAF50;
+        margin: 10px 0;
     }
     </style>
 """, unsafe_allow_html=True)
 
-# -------------------- SESSION STATE INIT --------------------
-if 'score' not in st.session_state:
-    st.session_state.score = 0
-if 'total_attempted' not in st.session_state:
-    st.session_state.total_attempted = 0
-if 'hint_used' not in st.session_state:
-    st.session_state.hint_used = False
+# Welcome title
+st.markdown("<div class='fun-title'>🧠✨ Welcome to the Smart Math Teacher! ✨🧠</div>", unsafe_allow_html=True)
 
-# -------------------- DATA LOADING WITH CACHE --------------------
-@st.cache_data
-def load_age_groups():
-    return {
-        "4-6 Years (Beginner)": {
-            "dataset": "src/Datase_of_4-6_Age_Group.xlsx",
-            "zip_file": "src/Image_for_group_4-6.zip"
-        },
-        "7-9 Years (Intermediate)": {
-            "dataset": "src/Datase_of_7-9_Age_Group.xlsx",
-            "zip_file": "src/Image_for_group_7-9.zip"
-        },
-        "13-15 Years (Advanced)": {
-            "dataset": "src/Datase_of_13-15_Age_Group.xlsx",
-            "zip_file": "src/Image_for_group_13-15.zip"
-        },
-    }
-
-@st.cache_data
-def load_dataset(dataset_path):
-    return pd.read_excel(dataset_path)
-
-age_groups = load_age_groups()
-
-# -------------------- SIDEBAR --------------------
-with st.sidebar:
-    st.image("https://img.icons8.com/fluency/96/math.png", width=80)
-    st.title("🧮 Smart Math Teacher")
-    st.markdown("---")
+# USER GUIDE - Always visible at the top
+st.markdown("""
+<div class="guide-box">
+<h3>📖 How to Use This App:</h3>
+
+**1. Select Your Age Group** - Choose your learning level (4-6, 7-9, or 13-15 years)<br>
+**2. Choose Math Category** - Pick what type of math to practice<br>
+**3. Solve Questions** - Read each question and type your answer<br>
+**4. Get Help** - Use hints or skip if needed<br>
+**5. Learn** - See correct answers and step-by-step solutions
+</div>
+""", unsafe_allow_html=True)
 
-    # Age group selection
-    selected_age = st.selectbox(
-        "📚 Select Age Group",
-        options=list(age_groups.keys()),
-        index=0
-    )
+# Initialize session state for temp directory
+if 'temp_dir' not in st.session_state:
+    st.session_state.temp_dir = tempfile.mkdtemp()
 
-    # Load dataset for selected age group
-    group_info = age_groups[selected_age]
-    dataset_path = group_info["dataset"]
-    zip_path = group_info["zip_file"]
+# Age group setup 
+age_groups = {
+    "4-6 Age Group": {"dataset": "src/Datase_of_4-6_Age_Group.xlsx", "zip_file": "src/Image_for_group_4-6.zip", "image_folder": "Image_for_group_4-6"},
+    "7-9 Age Group": {"dataset": "src/Datase_of_7-9_Age_Group.xlsx", "zip_file": "src/Image_for_group_7-9.zip", "image_folder": "Image_for_group_7-9"},
+    "13-15 Age Group": {"dataset": "src/Datase_of_13-15_Age_Group.xlsx", "zip_file": "src/Image_for_group_13-15.zip", "image_folder": "Image_for_group_13-15"},
+}
 
-    if not os.path.exists(dataset_path):
-        st.error(f"Dataset not found: {dataset_path}")
-        st.stop()
+selected_age_group = st.selectbox("🧒 Select your Age Group:", list(age_groups.keys()))
 
-    df = load_dataset(dataset_path)
+# Initialize session
+if "session_initialized" not in st.session_state or st.session_state.age_group != selected_age_group:
+    st.session_state.age_group = selected_age_group
+    st.session_state.category = None
+    st.session_state.question_index = 0
+    st.session_state.show_answer = False
+    st.session_state.show_steps = False
+    st.session_state.session_initialized = True
+
+# Load dataset
+group_info = age_groups[selected_age_group]
+dataset_path = group_info["dataset"]
+zip_path = group_info["zip_file"]
+image_folder = group_info["image_folder"]
+
+# Create image folder in temp directory
+temp_image_folder = os.path.join(st.session_state.temp_dir, image_folder)
+os.makedirs(temp_image_folder, exist_ok=True)
+
+# Extract images if zip exists
+try:
+    if os.path.exists(zip_path):
+        with zipfile.ZipFile(zip_path, "r") as zip_ref:
+            zip_ref.extractall(temp_image_folder)
+except Exception as e:
+    st.warning(f"Error extracting images: {e}")
+
+if not os.path.exists(dataset_path):
+    st.error(f"Dataset not found: {dataset_path}")
+    st.stop()
+
+try:
+    df = pd.read_excel(dataset_path)
     df['category'] = df['category'].astype(str).str.strip()
-    categories = sorted(df['category'].dropna().unique())
-
-    selected_category = st.selectbox(
-        "📖 Choose Category",
-        options=categories,
-        key="category_select"
-    )
-
-    st.markdown("---")
-
-    # Stats
-    st.subheader("📊 Your Stats")
-    col1, col2 = st.columns(2)
-    with col1:
-        st.metric("Score", st.session_state.score)
-    with col2:
-        st.metric("Attempted", st.session_state.total_attempted)
+except Exception as e:
+    st.error(f"Error loading dataset: {e}")
+    st.stop()
 
-    if st.session_state.total_attempted > 0:
-        accuracy = (st.session_state.score / st.session_state.total_attempted) * 100
-        st.progress(accuracy/100, text=f"Accuracy: {accuracy:.1f}%")
+# Category selection
+categories = sorted(df['category'].dropna().unique())
+selected_category = st.selectbox("📚 Choose a Math Category:", options=categories)
 
-    st.markdown("---")
-    st.caption("Made with ❤️ by Your Math Teacher")
-
-# -------------------- MAIN CONTENT --------------------
-st.markdown("<h1 class='main-title'>✨ Smart Math Teacher ✨</h1>", unsafe_allow_html=True)
-
-# Guide section (collapsible)
-with st.expander("📘 How to Use This App", expanded=False):
-    st.markdown("""
-    1. **Choose your age group** from the sidebar.
-    2. **Pick a math category** you want to practice.
-    3. **Read the question** and type your answer.
-    4. **Submit** to check your answer.
-    5. Use **Hint** if you need help.
-    6. Track your progress in the sidebar stats!
-    """)
-
-# Filter questions by selected category
-subset_df = df[df['category'] == selected_category].reset_index(drop=True)
-
-# Initialize question index in session state if needed
-if 'question_index' not in st.session_state or st.session_state.get('last_category') != selected_category:
+# Update session if category changes
+if st.session_state.category != selected_category:
+    st.session_state.category = selected_category
     st.session_state.question_index = 0
-    st.session_state.last_category = selected_category
-    st.session_state.hint_used = False
+    st.session_state.show_answer = False
+    st.session_state.show_steps = False
+    st.rerun()
 
-# Function to load image from zip on demand
-@st.cache_data(show_spinner=False)
-def load_image_from_zip(zip_path, image_name):
-    try:
-        with zipfile.ZipFile(zip_path, 'r') as zf:
-            # Find the file in the zip (case-insensitive)
-            for filename in zf.namelist():
-                if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
-                    base = os.path.splitext(os.path.basename(filename))[0]
-                    if base.lower() == image_name.lower():
-                        with zf.open(filename) as f:
-                            img_data = f.read()
-                            img = Image.open(io.BytesIO(img_data))
-                            return img
-    except Exception as e:
-        st.warning(f"Error loading image: {e}")
-    return None
+# Filter questions by selected category only
+subset_df = df[df['category'] == selected_category].reset_index(drop=True)
 
 if not subset_df.empty and st.session_state.question_index < len(subset_df):
     question = subset_df.iloc[st.session_state.question_index]
+    progress = int((st.session_state.question_index / len(subset_df)) * 100)
+    st.progress(progress)
 
-    # Progress bar
-    progress = (st.session_state.question_index / len(subset_df))
-    st.progress(progress, text=f"Question {st.session_state.question_index + 1} of {len(subset_df)}")
-
-    # Question box
-    st.markdown(f"""
-    <div class='question-box'>
-        <h3>❓ Question {st.session_state.question_index + 1}</h3>
-        <p>{question['problem']}</p>
-    </div>
-    """, unsafe_allow_html=True)
+    st.markdown(f"<div class='question-box'>📘 <b>Question {st.session_state.question_index + 1}:</b><br><br>{question['problem']}</div>", unsafe_allow_html=True)
 
-    # Display image if exists
+    # Display image with error handling
     if pd.notna(question.get('image')):
         image_name = str(question['image']).strip()
-        img = load_image_from_zip(zip_path, image_name)
-        if img:
-            st.image(img, use_column_width=True, caption="Question Image")
-        else:
-            st.info("📷 No image for this question.")
-
-    # Answer input
-    user_answer = st.text_input("✏️ Your Answer:", key="answer_input", placeholder="Type your answer here...")
-
-    # Buttons layout
-    col1, col2, col3, col4 = st.columns([1,1,1,1])
-    with col1:
-        submitted = st.button("✅ Submit", use_container_width=True)
-    with col2:
-        hint_clicked = st.button("💡 Hint", use_container_width=True)
-    with col3:
-        skip_clicked = st.button("⏭️ Skip", use_container_width=True)
-    with col4:
-        if st.button("🔄 Reset Progress", use_container_width=True):
-            st.session_state.question_index = 0
-            st.session_state.score = 0
-            st.session_state.total_attempted = 0
-            st.rerun()
-
-    # Hint logic
-    if hint_clicked:
-        st.session_state.hint_used = True
-        # Check if dataset has hints column, else provide generic
-        if 'hint' in question and pd.notna(question['hint']):
-            hint_text = question['hint']
-        else:
-            hint_text = "Think step by step. Try to break the problem into smaller parts."
-        st.markdown(f"<div class='hint-box'>💡 <b>Hint:</b> {hint_text}</div>", unsafe_allow_html=True)
-
-    # Submission logic
-    if submitted:
-        st.session_state.total_attempted += 1
-        correct_answer = str(question['answer']).strip().lower()
-        user_ans = str(user_answer).strip().lower()
-
-        if user_ans == correct_answer:
-            st.success("🎉 Correct! Great job!")
+        image_found = False
+        for root, _, files in os.walk(temp_image_folder):
+            for file in files:
+                if file.lower().startswith(image_name.lower()) or os.path.splitext(file)[0].lower() == image_name.lower():
+                    try:
+                        st.image(Image.open(os.path.join(root, file)), use_column_width=True)
+                        image_found = True
+                        break
+                    except:
+                        pass
+        if not image_found:
+            st.warning("❌ Image not found.")
+
+    user_ans = st.text_input("📝 Your Answer:", key=f"ans_{st.session_state.question_index}")
+
+    if st.button("✅ Submit Answer"):
+        if str(user_ans).strip().lower() == str(question['answer']).strip().lower():
+            st.success("🎉 Correct! Well done!")
             st.balloons()
-            # Award points
-            points = 10 if not st.session_state.hint_used else 5
-            st.session_state.score += points
-            # Move to next question after a short delay
             time.sleep(2)
             st.session_state.question_index += 1
-            st.session_state.hint_used = False
+            st.session_state.show_answer = False
+            st.session_state.show_steps = False
             st.rerun()
         else:
-            st.error("❌ Not quite right. Try again or use a hint.")
-
-    if skip_clicked:
+            st.error("❌ Try again or view the correct answer below.")
+            st.session_state.show_answer = True
+            st.session_state.show_steps = False
+
+    if st.session_state.show_answer:
+        st.info(f"✅ Correct Answer: **{question['answer']}**")
+        if selected_age_group in ["7-9 Age Group", "13-15 Age Group"]:
+            if st.button("🔍 Show Steps"):
+                st.session_state.show_steps = True
+        if st.session_state.show_steps and pd.notna(question.get("steps", None)):
+            st.success(f"### 🪄 Steps:\n{question['steps']}")
+
+    if st.button("⏭️ Skip"):
         st.session_state.question_index += 1
-        st.session_state.hint_used = False
+        st.session_state.show_answer = False
+        st.session_state.show_steps = False
         st.rerun()
-
-    # Show answer and steps if user requests (optional)
-    with st.expander("🔍 Show Answer & Steps"):
-        st.markdown(f"**Correct Answer:** {question['answer']}")
-        if 'steps' in question and pd.notna(question['steps']):
-            st.markdown(f"**Step-by-Step:**\n{question['steps']}")
-        else:
-            st.info("No steps provided for this question.")
-
+elif subset_df.empty:
+    st.warning("⚠️ No questions available in this category. Try another one.")
 else:
-    # No more questions
-    st.success("🎯 You've completed all questions in this category!")
-    st.balloons()
-    if st.button("🔄 Start Over in This Category"):
-        st.session_state.question_index = 0
-        st.session_state.score = 0
-        st.session_state.total_attempted = 0
-        st.rerun()
+    st.success("🏁 You've completed all questions in this category!")
 
-# No cleanup needed because we never extracted files
+# Cleanup function
+def cleanup():
+    import shutil
+    if 'temp_dir' in st.session_state and os.path.exists(st.session_state.temp_dir):
+        shutil.rmtree(st.session_state.temp_dir)
+
+import atexit
+atexit.register(cleanup)