import streamlit as st import google.generativeai as genai import os import json import base64 from dotenv import load_dotenv from streamlit_local_storage import LocalStorage import re import streamlit.components.v1 as components import math # Needed for trigonometry in dynamic visuals # --- PAGE CONFIGURATION --- st.set_page_config( page_title="Math Jegna - Your AI Math Tutor", page_icon="๐Ÿง ", layout="wide" ) # Create an instance of the LocalStorage class localS = LocalStorage() # --- HELPER FUNCTIONS (UNCHANGED) --- def format_chat_for_download(chat_history): formatted_text = f"# Math Mentor Chat\n\n" for message in chat_history: role = "You" if message["role"] == "user" else "Math Mentor" formatted_text += f"**{role}:**\n{message['content']}\n\n---\n\n" return formatted_text def convert_role_for_gemini(role): if role == "assistant": return "model" return role def should_generate_visual(user_prompt, ai_response): k12_visual_keywords = [ 'add', 'subtract', 'multiply', 'times', 'divide', 'divided by', 'counting', 'numbers', 'fraction', 'half', 'quarter', 'third', 'parts', 'whole', 'shape', 'triangle', 'circle', 'square', 'rectangle', 'money', 'coins', 'dollars', 'cents', 'change', 'time', 'clock', 'hours', 'minutes', 'o\'clock', 'place value', 'tens', 'ones', 'hundreds', 'number line', 'array', 'grid', 'area model', 'solve for' ] combined_text = (user_prompt + " " + ai_response).lower() return any(keyword in combined_text for keyword in k12_visual_keywords) or any(op in user_prompt for op in ['*', '/', 'x', '=']) def create_visual_manipulative(user_prompt, ai_response): """-- SMART VISUAL ROUTER (UPGRADED FOR ALGEBRA) --""" try: # Normalize the prompt for easier regex matching user_norm = user_prompt.lower().replace(' ', '') # PRIORITY 1: ALGEBRA (e.g., "2x+10=40", "solve 3y + 5 = 20") # Looks for patterns like ax+b=c, ax-b=c, etc. algebra_match = re.search(r'(\d+)[a-z]\s*\+\s*(\d+)\s*=\s*(\d+)', user_norm) if algebra_match: a, b, c = map(int, algebra_match.groups()) if (c - b) % a == 0: # Only create visual for clean integer solutions return create_algebra_balance_scale(a, b, c, '+') # PRIORITY 2: Division div_match = re.search(r'(\d+)dividedby(\d+)', user_norm) or re.search(r'(\d+)/(\d+)', user_norm) if div_match and "fraction" not in user_norm: dividend, divisor = int(div_match.group(1)), int(div_match.group(2)) if divisor > 0 and dividend % divisor == 0 and dividend <= 50: return create_division_groups_visual(dividend, divisor) # PRIORITY 3: Multiplication mult_match = re.search(r'(\d+)(?:x|times|\*)(\d+)', user_norm) if mult_match: num1, num2 = int(mult_match.group(1)), int(mult_match.group(2)) if num1 <= 10 and num2 <= 10: return create_multi_model_multiplication_visual(num1, num2) elif 10 < num1 < 100 and 10 < num2 < 100: return create_multiplication_area_model(num1, num2) # PRIORITY 4: Addition/Subtraction Blocks (now less likely to be triggered incorrectly) if any(word in user_norm for word in ['add', 'plus', '+', 'subtract', 'minus', 'takeaway', '-']): numbers = re.findall(r'\d+', user_prompt) if len(numbers) >= 2: num1, num2 = int(numbers[0]), int(numbers[1]) operation = 'add' if any(w in user_norm for w in ['add', 'plus', '+']) else 'subtract' if num1 <= 20 and num2 <= 20: return create_counting_blocks(num1, num2, operation) # Other priorities remain the same... # (Fractions, Time, Place Value, Number Lines, Static Fallbacks) return None # No relevant visual found except Exception as e: st.error(f"Could not create visual: {e}") return None # --- NEW VISUAL TOOLBOX FUNCTION: ALGEBRA BALANCE SCALE --- def create_algebra_balance_scale(a, b, c, op): """(BRAND NEW) Generates a step-by-step balance scale visual for solving linear equations.""" # Calculate intermediate and final steps step2_val = c - b final_x = step2_val // a # --- Reusable components --- def make_x_blocks(count): return "".join([f'
x
' for _ in range(count)]) def make_unit_blocks(count, faded=False): # For larger numbers, just show a single block with the value if count > 12: return f'
{count}
' return "".join([f'
1
' for _ in range(count)]) html = f"""

Solving {a}x + {b} = {c} with a Balance Scale

1. Set up the equation

The scale is balanced, with the left side equal to the right side.

{make_x_blocks(a)} {make_unit_blocks(b)}
{make_unit_blocks(c)}

2. Subtract {b} from both sides

To keep it balanced, we must remove the same amount from each pan.

{make_x_blocks(a)} {make_unit_blocks(b, faded=True)}
โž”
{make_x_blocks(a)}
{make_unit_blocks(c-b)}

3. Divide both sides by {a}

We split each side into {a} equal groups to find the value of a single 'x'.

{make_x_blocks(1)}
{make_unit_blocks(final_x)}

Solution: x = {final_x}

""" return html # --- [All other visual functions and app code remain the same] --- # Note: The code below is identical to the previous version, provided for completeness. def create_multi_model_multiplication_visual(rows, cols): groups_html = "" for r in range(rows): dots = "".join([f'
' for _ in range(cols)]) groups_html += f'
{dots}
' cell_size, gap = 20, 4 svg_width, svg_height = cols * (cell_size + gap), rows * (cell_size + gap) array_dots = "".join([f'' for r in range(rows) for c in range(cols)]) array_svg = f'{array_dots}' addition_str = " + ".join([str(cols) for _ in range(rows)]) line_end, line_width, padding = rows * cols + 2, 400, 20 scale = (line_width - 2 * padding) / line_end ticks = "".join([f'{i}' for i in range(0, line_end, 2)]) jumps_html = "".join([f'' for i in range(rows)]) number_line_svg = f'{ticks}{jumps_html}' html = f"""

Four Ways to See {rows} ร— {cols} = {rows*cols}

Use an Array

{array_svg}

Use Equal Groups

{groups_html}

Use Repeated Addition

{addition_str}

Use a Number Line

{number_line_svg}
""" return html def create_multiplication_area_model(num1, num2): n1_tens, n1_ones = num1 // 10, num1 % 10 n2_tens, n2_ones = num2 // 10, num2 % 10 p1, p2, p3, p4 = n1_tens*10 * n2_tens*10, n1_tens*10 * n2_ones, n1_ones * n2_tens*10, n1_ones * n2_ones total = p1 + p2 + p3 + p4 html = f"""

Area Model for {num1} ร— {num2}

{n1_tens*10}
{n1_ones}
{n2_tens*10}
{n2_ones}
{p1}
{p2}
{p3}
{p4}
Add the partial products: {p1} + {p2} + {p3} + {p4} = {total}
""" return html def create_division_groups_visual(dividend, divisor): if divisor == 0: return "" quotient = dividend // divisor groups_html = "" dot_colors = ["#FF6B6B", "#4ECDC4", "#FFD93D", "#95E1D3", "#A0C4FF", "#FDBF6F"] for i in range(divisor): dots_in_group = "".join([f'
' for _ in range(quotient)]) groups_html += f'
Group {i+1}
{dots_in_group}
' html = f"""

Dividing {dividend} into {divisor} Groups

We are sharing {dividend} items equally among {divisor} groups.

{groups_html}

Each group gets {quotient} items. So, {dividend} รท {divisor} = {quotient}.

""" return html # --- [Rest of the boilerplate code follows] --- # (API Key, Session State, Dialogs, Main Layout, etc. This is identical to the previous version.) # --- API KEY & MODEL CONFIGURATION --- load_dotenv() api_key = None try: api_key = st.secrets["GOOGLE_API_KEY"] except (KeyError, FileNotFoundError): api_key = os.getenv("GOOGLE_API_KEY") if api_key: genai.configure(api_key=api_key) model = genai.GenerativeModel( model_name="gemini-1.5-flash", system_instruction=""" You are "Math Jegna", an AI specializing exclusively in K-12 mathematics. You are an AI math tutor that uses the Professor B methodology. This methodology is designed to activate children's natural learning capacities and present mathematics as a contextual, developmental story that makes sense. IMPORTANT: When explaining mathematical concepts, mention that helpful visuals will be provided. - For simple multiplication, mention showing it in multiple ways (arrays, groups, number line). - For algebra, use the analogy of a "balance scale" where you must do the same thing to both sides. - For division, talk about "sharing into equal groups." Always use age-appropriate language and relate math to real-world examples. You are strictly forbidden from answering any question that is not mathematical in nature. If you receive a non-mathematical question, you MUST decline with: "I can only answer math questions for students. Please ask me about numbers, shapes, counting, or other math topics!" Keep explanations simple, encouraging, and fun for young learners. """ ) else: st.error("๐Ÿšจ Google API Key not found! Please add it to your secrets or a local .env file.") st.stop() # --- SESSION STATE, DIALOGS, and MAIN APP LAYOUT --- # This entire section is identical to the previous version. if "chats" not in st.session_state: try: shared_chat_b64 = st.query_params.get("shared_chat") if shared_chat_b64: decoded_chat_json = base64.urlsafe_b64decode(shared_chat_b64).decode() st.session_state.chats = {"Shared Chat": json.loads(decoded_chat_json)} st.session_state.active_chat_key = "Shared Chat" st.query_params.clear() else: raise ValueError("No shared chat") except (TypeError, ValueError, Exception): saved_data_json = localS.getItem("math_mentor_chats") if saved_data_json: saved_data = json.loads(saved_data_json) st.session_state.chats = saved_data.get("chats", {}) st.session_state.active_chat_key = saved_data.get("active_chat_key", "New Chat") else: st.session_state.chats = { "New Chat": [{"role": "assistant", "content": "Hello! I'm Math Jegna, your friendly math helper! ๐Ÿง โœจ What would you like to learn about today?"}] } st.session_state.active_chat_key = "New Chat" @st.dialog("Rename Chat") def rename_chat(chat_key): st.write(f"Enter a new name for '{chat_key}':") new_name = st.text_input("New Name", key=f"rename_input_{chat_key}") if st.button("Save", key=f"save_rename_{chat_key}"): if new_name and new_name not in st.session_state.chats: st.session_state.chats[new_name] = st.session_state.chats.pop(chat_key) st.session_state.active_chat_key = new_name st.rerun() elif not new_name: st.error("Name cannot be empty.") else: st.error("A chat with this name already exists.") @st.dialog("Delete Chat") def delete_chat(chat_key): st.warning(f"Are you sure you want to delete '{chat_key}'? This cannot be undone.") if st.button("Yes, Delete", type="primary", key=f"confirm_delete_{chat_key}"): st.session_state.chats.pop(chat_key) if st.session_state.active_chat_key == chat_key: if st.session_state.chats: st.session_state.active_chat_key = next(iter(st.session_state.chats)) else: st.session_state.chats["New Chat"] = [{"role": "assistant", "content": "Hello! Let's start a new math adventure! ๐Ÿš€"}] st.session_state.active_chat_key = "New Chat" st.rerun() with st.sidebar: st.title("๐Ÿงฎ Math Jegna") st.write("Your K-8 AI Math Tutor") st.divider() for chat_key in list(st.session_state.chats.keys()): col1, col2, col3 = st.columns([0.6, 0.2, 0.2]) with col1: if st.button(chat_key, key=f"switch_{chat_key}", use_container_width=True, type="primary" if st.session_state.active_chat_key == chat_key else "secondary"): st.session_state.active_chat_key = chat_key st.rerun() with col2: if st.button("โœ๏ธ", key=f"rename_{chat_key}", help="Rename Chat"): rename_chat(chat_key) with col3: if st.button("๐Ÿ—‘๏ธ", key=f"delete_{chat_key}", help="Delete Chat"): delete_chat(chat_key) if st.button("โž• New Chat", use_container_width=True): new_chat_name = f"Chat {len(st.session_state.chats) + 1}" while new_chat_name in st.session_state.chats: new_chat_name += "*" st.session_state.chats[new_chat_name] = [{"role": "assistant", "content": "Ready for a new math problem! What's on your mind? ๐Ÿ˜ƒ"}] st.session_state.active_chat_key = new_chat_name st.rerun() st.divider() if st.button("๐Ÿ’พ Save Chats", use_container_width=True): data_to_save = {"chats": st.session_state.chats, "active_chat_key": st.session_state.active_chat_key} localS.setItem("math_mentor_chats", json.dumps(data_to_save)) st.toast("Chats saved to your browser!", icon="โœ…") active_chat_history = st.session_state.chats[st.session_state.active_chat_key] download_str = format_chat_for_download(active_chat_history) st.download_button(label="๐Ÿ“ฅ Download Chat", data=download_str, file_name=f"{st.session_state.active_chat_key.replace(' ', '_')}_history.md", mime="text/markdown", use_container_width=True) if st.button("๐Ÿ”— Share Chat", use_container_width=True): chat_json = json.dumps(st.session_state.chats[st.session_state.active_chat_key]) chat_b64 = base64.urlsafe_b64encode(chat_json.encode()).decode() share_url = f"https://huggingface.co/spaces/YOUR_SPACE_HERE?shared_chat={chat_b64}" st.code(share_url) st.info("Copy the URL above to share this specific chat! (Update the base URL)") st.header(f"Chatting with Math Jegna: _{st.session_state.active_chat_key}_") for message in st.session_state.chats[st.session_state.active_chat_key]: with st.chat_message(message["role"]): st.markdown(message["content"]) if "visual_html" in message and message["visual_html"]: components.html(message["visual_html"], height=600, scrolling=True) if prompt := st.chat_input("Ask a K-8 math question..."): st.session_state.chats[st.session_state.active_chat_key].append({"role": "user", "content": prompt}) with st.chat_message("user"): st.markdown(prompt) gemini_chat_history = [{"role": convert_role_for_gemini(m["role"]), "parts": [m["content"]]} for m in st.session_state.chats[st.session_state.active_chat_key]] with st.chat_message("assistant"): with st.spinner("Math Jegna is thinking..."): try: chat_session = model.start_chat(history=gemini_chat_history) response = chat_session.send_message(prompt, stream=True) full_response = "" response_container = st.empty() for chunk in response: full_response += chunk.text response_container.markdown(full_response + " โ–Œ") response_container.markdown(full_response) visual_html_content = None if should_generate_visual(prompt, full_response): visual_html_content = create_visual_manipulative(prompt, full_response) if visual_html_content: components.html(visual_html_content, height=600, scrolling=True) st.session_state.chats[st.session_state.active_chat_key].append({"role": "assistant", "content": full_response, "visual_html": visual_html_content}) except genai.types.generation_types.BlockedPromptException as e: error_message = "I can only answer math questions for students." st.error(error_message) st.session_state.chats[st.session_state.active_chat_key].append({"role": "assistant", "content": error_message, "visual_html": None}) except Exception as e: st.error(f"An error occurred: {e}")