Update app.py

app.py

@@ -1,142 +1,27 @@
 """
-ThutoAI - Modern AI Student Assistant with Student Accounts & Multi-Language
-Meaning: "Thuto" = Learning/Education (Setswana)
+ThutoAI - Complete School Assistant with Voice, Assignments & Class Groups
+Meaning: "Thuto" = Learning/Education (Setswana — used for branding only)
 
-✅ Student accounts: save chat history & files per user
-✅ Multi-language: English / Setswana toggle
+✅ Student Accounts (saved chat history & files)
+✅ 🎙️ Voice Input (Web Speech API)
+✅ 📅 Assignment & Deadline Tracker (Calendar UI)
+✅ 👥 Class Groups (Students join classes like "MATH10A")
 ✅ Modern UI with dark mode, animations, teacher dashboard
 ✅ Fully commented for easy understanding
 """
 
 import os
 import gradio as gr
-from datetime import datetime
+from datetime import datetime, timedelta
 from typing import List, Dict, Optional
 import time
 import json
-
-# ==================== LANGUAGE SUPPORT ====================
-
-class LanguageService:
-    """Handles UI text in multiple languages."""
-
-    def __init__(self):
-        self.current_lang = "en"  # Default: English
-        self.translations = {
-            "en": {
-                "app_title": "🎓 ThutoAI — Your Modern AI Learning Assistant",
-                "login_title": "🔐 Student Login",
-                "username_label": "Username",
-                "password_label": "Password",
-                "login_btn": "🔓 Login",
-                "register_btn": "📝 Register",
-                "logout_btn": "⬅️ Logout",
-                "welcome": "Welcome back, {name}!",
-                "register_success": "✅ Account created! Please log in.",
-                "invalid_login": "❌ Invalid username or password",
-                "ask_placeholder": "E.g., How do I prepare for the Math exam?",
-                "send_btn": "➤ Send",
-                "clear_chat": "🗑️ Clear Chat",
-                "upload_label": "Drag & drop or click to upload",
-                "upload_btn": "📤 Upload",
-                "file_uploaded": "✅ Uploaded: {filename}",
-                "no_file": "❌ No file selected",
-                "announcements_title": "📢 Announcements",
-                "filter_label": "Select Course",
-                "refresh_btn": "🔄 Refresh",
-                "no_announcements": "📭 No announcements for this course.",
-                "teacher_login_title": "👩‍🏫 Teacher Login",
-                "post_announcement": "✍️ Create New Announcement",
-                "announcement_title": "Title",
-                "announcement_content": "Details for students...",
-                "priority_label": "Priority",
-                "post_btn": "📬 Post Announcement",
-                "posted_success": "✅ Posted! 🎉 New announcement ID: {id}",
-                "login_first": "🔒 Please log in first.",
-                "title_required": "⚠️ Title is required.",
-                "content_required": "⚠️ Content is required.",
-                "ai_thinking": "🤔 ThutoAI is thinking...",
-                "ai_mock_tip": "💡 Pro tip: Add your OpenAI API key in HF Secrets for smarter answers!",
-                "stats_title": "📊 Dashboard Stats",
-                "total_announcements": "Total Announcements",
-                "total_files": "Total Files Uploaded",
-                "active_courses": "Active Courses",
-                "high_priority": "High Priority Posts",
-                "posted_by": "Posted by",
-                "course": "Course",
-                "date": "Date",
-                "priority": "Priority",
-                "dark_mode": "🌙 Toggle Dark Mode",
-                "language_toggle": "🌐 Switch to Setswana",
-                "back_to_english": "🔤 Back to English"
-            },
-            "tn": {  # Setswana
-                "app_title": "🎓 ThutoAI — Motshwareledi wa Gago wa AI",
-                "login_title": "🔐 Tshedimosetso ya Moithuti",
-                "username_label": "Leina la mo sebeletso",
-                "password_label": "Password",
-                "login_btn": "🔓 Tshedimosa",
-                "register_btn": "📝 Ngolola",
-                "logout_btn": "⬅️ Tsoha",
-                "welcome": "La le amogatswe, {name}!",
-                "register_success": "✅ Ak'awunti e entsiwe! Tshedimosetse ka gonne.",
-                "invalid_login": "❌ Leina le lebetsweng kapa password ga le sirele",
-                "ask_placeholder": "Mohlala: Ke itumelelang jang mo go tlwaelo ya Math?",
-                "send_btn": "➤ Romela",
-                "clear_chat": "🗑️ Phumolosa Puisano",
-                "upload_label": "Lemba & tsela kapa tobetsa go upload",
-                "upload_btn": "📤 Upload",
-                "file_uploaded": "✅ E rometswe: {filename}",
-                "no_file": "❌ Ga file e kgethilwe",
-                "announcements_title": "📢 Ditlhaloso",
-                "filter_label": "Kgetha Sebele",
-                "refresh_btn": "🔄 Nna gape",
-                "no_announcements": "��� Ga go na ditlhaloso bakeng sa sebele seo.",
-                "teacher_login_title": "👩‍🏫 Tshedimosetso ya Moruti",
-                "post_announcement": "✍️ Bopa Tlhaloso e Ntsha",
-                "announcement_title": "Thaetele",
-                "announcement_content": "Tshedimosetso ya moithutwana...",
-                "priority_label": "Boikanyo",
-                "post_btn": "📬 Romela Tlhaloso",
-                "posted_success": "✅ E rometswe! 🎉 Tlhaloso e ntsha ID: {id}",
-                "login_first": "🔒 Tshedimosetse pele.",
-                "title_required": "⚠️ Thaetele e tlhofosegile.",
-                "content_required": "⚠️ Tshedimosetso e tlhofosegile.",
-                "ai_thinking": "🤔 ThutoAI e nang le maikutlo...",
-                "ai_mock_tip": "💡 Tip: Naya OpenAI API key mo HF Secrets bakeng sa ditlhahiso tse di botlhokwa!",
-                "stats_title": "📊 Ditlhophiso tsa Dashboard",
-                "total_announcements": "Ditlhaloso tsohle",
-                "total_files": "Mafaele a rometsweng",
-                "active_courses": "Disebele tse di sebetsang",
-                "high_priority": "Diposo tse di ikanyeditsegile",
-                "posted_by": "E rometswe ke",
-                "course": "Sebele",
-                "date": "Letsatsi",
-                "priority": "Boikanyo",
-                "dark_mode": "🌙 Nna Mode e E Nnyane",
-                "language_toggle": "🌐 Tselang go ya Setswaneng",
-                "back_to_english": "🔤 Tselang go ya English"
-            }
-        }
-
-    def set_language(self, lang_code: str):
-        """Set current language."""
-        if lang_code in self.translations:
-            self.current_lang = lang_code
-
-    def t(self, key: str, **kwargs) -> str:
-        """Get translated text with optional formatting."""
-        text = self.translations[self.current_lang].get(key, self.translations["en"].get(key, f"[{key}]"))
-        return text.format(**kwargs) if kwargs else text
-
-
-# Initialize language service
-lang_service = LanguageService()
+import random
 
 # ==================== STUDENT ACCOUNTS SERVICE ====================
 
 class StudentService:
-    """Manages student accounts, chat history, and files."""
+    """Manages student accounts, chat history, files, assignments, and class groups."""
 
     def __init__(self):
         # In-memory storage — replace with SQLite in production
@@ -144,7 +29,29 @@ class StudentService:
             "student1": {"password": "pass123", "name": "John Doe"},
             "student2": {"password": "pass456", "name": "Jane Smith"}
         }
-        self.student_sessions = {}  # {username: {chat_history: [], files: []}}
+        # Structure: {username: {chat_history: [], files: [], assignments: [], groups: []}}
+        self.student_sessions = {
+            "student1": {
+                "chat_history": [],
+                "files": [],
+                "assignments": [
+                    {"title": "Math Quiz", "due_date": "2025-04-25", "course": "MATH10A", "status": "pending"},
+                    {"title": "Science Lab Report", "due_date": "2025-04-30", "course": "SCI11B", "status": "pending"}
+                ],
+                "groups": ["MATH10A", "SCI11B"]
+            },
+            "student2": {
+                "chat_history": [],
+                "files": [],
+                "assignments": [
+                    {"title": "History Essay", "due_date": "2025-04-22", "course": "HIST9A", "status": "overdue"},
+                    {"title": "English Reading", "due_date": "2025-04-28", "course": "ENG10A", "status": "pending"}
+                ],
+                "groups": ["HIST9A", "ENG10A"]
+            }
+        }
+        # Predefined valid class groups
+        self.valid_groups = ["MATH10A", "MATH10B", "SCI11A", "SCI11B", "ENG10A", "ENG10B", "HIST9A", "HIST9B"]
 
     def register_student(self, username: str, password: str, name: str) -> str:
         """Register new student."""
@@ -153,47 +60,88 @@ class StudentService:
         if username in self.students:
             return "⚠️ Username already exists."
         self.students[username] = {"password": password, "name": name}
-        self.student_sessions[username] = {"chat_history": [], "files": []}
-        return lang_service.t("register_success")
+        self.student_sessions[username] = {
+            "chat_history": [],
+            "files": [],
+            "assignments": [],
+            "groups": []
+        }
+        return "✅ Account created! Please log in."
 
     def authenticate_student(self, username: str, password: str) -> Optional[str]:
         """Authenticate student and return name if successful."""
         student = self.students.get(username)
         if student and student["password"] == password:
             if username not in self.student_sessions:
-                self.student_sessions[username] = {"chat_history": [], "files": []}
+                self.student_sessions[username] = {
+                    "chat_history": [],
+                    "files": [],
+                    "assignments": [],
+                    "groups": []
+                }
             return student["name"]
         return None
 
     def get_chat_history(self, username: str) -> List:
-        """Get chat history for student."""
         return self.student_sessions.get(username, {}).get("chat_history", [])
 
     def add_to_chat_history(self, username: str, user_msg: str, bot_reply: str):
-        """Add message pair to student's chat history."""
         if username in self.student_sessions:
             self.student_sessions[username]["chat_history"].append((user_msg, bot_reply))
 
     def get_files(self, username: str) -> List:
-        """Get uploaded files for student."""
         return self.student_sessions.get(username, {}).get("files", [])
 
     def add_file(self, username: str, filename: str):
-        """Add file to student's list."""
         if username in self.student_sessions:
             self.student_sessions[username]["files"].append({
                 "name": filename,
                 "uploaded_at": datetime.now().strftime("%Y-%m-%d %H:%M")
             })
 
+    def get_assignments(self, username: str) -> List:
+        """Get student's assignments sorted by due date."""
+        assignments = self.student_sessions.get(username, {}).get("assignments", [])
+        # Sort by due date
+        return sorted(assignments, key=lambda x: x["due_date"])
+
+    def add_assignment(self, username: str, title: str, due_date: str, course: str):
+        """Add assignment (used by teacher or self)."""
+        if username in self.student_sessions:
+            self.student_sessions[username]["assignments"].append({
+                "title": title,
+                "due_date": due_date,
+                "course": course,
+                "status": "pending"
+            })
+
+    def get_groups(self, username: str) -> List:
+        return self.student_sessions.get(username, {}).get("groups", [])
+
+    def join_group(self, username: str, group_code: str) -> str:
+        if group_code.upper() not in self.valid_groups:
+            return "❌ Invalid group code. Ask your teacher for the correct code."
+        if username in self.student_sessions:
+            if group_code.upper() not in self.student_sessions[username]["groups"]:
+                self.student_sessions[username]["groups"].append(group_code.upper())
+            return f"✅ Joined group: {group_code.upper()}"
+        return "❌ Login required."
+
+    def leave_group(self, username: str, group_code: str) -> str:
+        if username in self.student_sessions:
+            if group_code.upper() in self.student_sessions[username]["groups"]:
+                self.student_sessions[username]["groups"].remove(group_code.upper())
+                return f"✅ Left group: {group_code.upper()}"
+        return "❌ Group not found or not joined."
+
 
 # Initialize student service
 student_service = StudentService()
 
-# ==================== SCHOOL SERVICE (Enhanced) ====================
+# ==================== SCHOOL SERVICE ====================
 
 class SchoolService:
-    """Handles announcements, files, and AI context."""
+    """Handles announcements, AI context, and shared assignments."""
 
     def __init__(self):
         self.announcements = [
@@ -262,7 +210,6 @@ class SchoolService:
         }
 
 
-# Initialize school service
 school_service = SchoolService()
 
 # ==================== ADMIN SERVICE ====================
@@ -305,7 +252,7 @@ def ai_chat(message: str, history: List, username: str = "guest") -> tuple:
         return history, ""
 
     # Show "thinking" state
-    thinking_msg = lang_service.t("ai_thinking")
+    thinking_msg = "🤔 ThutoAI is thinking..."
     history.append((message, thinking_msg))
     yield history, ""
 
@@ -348,8 +295,8 @@ Guidelines:
         except Exception as e:
             reply = f"⚠️ Sorry, I had a glitch: {str(e)}"
     else:
-        time.sleep(1.5)  # Simulate thinking
-        reply = f"👋 Hi! I'm ThutoAI. You asked: '{message}'.\n{lang_service.t('ai_mock_tip')}"
+        time.sleep(1.5)
+        reply = f"👋 Hi! I'm ThutoAI. You asked: '{message}'.\n💡 *Pro tip: Add your OpenAI API key in HF Secrets for smarter answers!*"
 
     # Replace thinking message with real reply
     history[-1] = (message, reply)
@@ -361,18 +308,17 @@ Guidelines:
     yield history, ""
 
 
-# ==================== UI RENDERING ====================
+# ==================== UI RENDERING HELPERS ====================
 
-def render_announcements(course: str, lang: str = "en") -> str:
+def render_announcements(course: str) -> str:
     """Render announcements with modern cards."""
-    lang_service.set_language(lang)
     announcements = school_service.get_announcements(course)
     if not announcements:
-        return f"""
+        return """
         <div style='text-align: center; padding: 40px; color: #6c757d;'>
             <div style='font-size: 4em; margin-bottom: 16px;'>📭</div>
-            <h3>{lang_service.t("no_announcements")}</h3>
-            <p>{lang_service.t("refresh_btn")} {lang_service.t("filter_label").lower()}</p>
+            <h3>No announcements for this course.</h3>
+            <p>Check back later or select "All" to see everything.</p>
         </div>
         """
 
@@ -420,25 +366,124 @@ def render_announcements(course: str, lang: str = "en") -> str:
     return html
 
 
+def render_assignments(assignments: List[Dict]) -> str:
+    """Render assignments in a clean, prioritized list."""
+    if not assignments:
+        return """
+        <div style='text-align: center; padding: 40px; color: #6c757d;'>
+            <div style='font-size: 4em; margin-bottom: 16px;'>📅</div>
+            <h3>No upcoming assignments.</h3>
+            <p>Ask your teacher or join a class group to see assignments.</p>
+        </div>
+        """
+
+    html = "<div style='display: grid; gap: 16px;'>"
+    today = datetime.today().date()
+
+    for task in assignments:
+        due_date = datetime.strptime(task["due_date"], "%Y-%m-%d").date()
+        days_left = (due_date - today).days
+        is_overdue = days_left < 0
+        is_today = days_left == 0
+
+        if is_overdue:
+            badge = "🚨 OVERDUE"
+            color = "#dc3545"
+        elif is_today:
+            badge = "🎯 TODAY"
+            color = "#fd7e14"
+        elif days_left <= 2:
+            badge = f"⚠️ Due in {days_left} day{'s' if days_left != 1 else ''}"
+            color = "#ffc107"
+        else:
+            badge = f"✅ Due in {days_left} days"
+            color = "#28a745"
+
+        html += f"""
+        <div style='
+            background: white;
+            border-radius: 12px;
+            padding: 20px;
+            box-shadow: 0 2px 8px rgba(0,0,0,0.08);
+            border-left: 4px solid {color};
+            transition: transform 0.2s;
+        '>
+            <div style='display: flex; justify-content: space-between; align-items: flex-start;'>
+                <div>
+                    <h3 style='margin: 0 0 8px 0; color: #212529;'>{task['title']}</h3>
+                    <div style='color: #6c757d; margin-bottom: 8px;'>📚 {task['course']}</div>
+                    <div style='color: #495057;'>📅 Due: {task['due_date']}</div>
+                </div>
+                <span style='
+                    background: {color};
+                    color: white;
+                    padding: 6px 12px;
+                    border-radius: 20px;
+                    font-weight: bold;
+                    font-size: 0.85em;
+                    align-self: flex-start;
+                '>{badge}</span>
+            </div>
+        </div>
+        """
+
+    html += "</div>"
+    return html
+
+
+def render_groups(groups: List[str]) -> str:
+    """Render joined class groups."""
+    if not groups:
+        return """
+        <div style='text-align: center; padding: 40px; color: #6c757d;'>
+            <div style='font-size: 4em; margin-bottom: 16px;'>👥</div>
+            <h3>You haven't joined any class groups yet.</h3>
+            <p>Ask your teacher for a group code to join your class.</p>
+        </div>
+        """
+
+    html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 16px;'>"
+    for group in groups:
+        html += f"""
+        <div style='
+            background: #e3f2fd;
+            border-radius: 12px;
+            padding: 20px;
+            text-align: center;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+            border: 2px solid #2196f3;
+        '>
+            <div style='font-size: 1.5em; margin-bottom: 8px;'>🎒</div>
+            <h3 style='margin: 0; color: #1976d2;'>{group}</h3>
+            <p style='margin: 8px 0 0 0; color: #555; font-size: 0.9em;'>Class Group</p>
+        </div>
+        """
+    html += "</div>"
+    return html
+
+
 # ==================== STATE MANAGEMENT ====================
 
 CURRENT_USER = "guest"
-CURRENT_LANG = "en"
 
 def login_student(username: str, password: str) -> tuple:
     """Login student and load their data."""
-    global CURRENT_USER, CURRENT_LANG
+    global CURRENT_USER
     student_name = student_service.authenticate_student(username, password)
     if student_name:
         CURRENT_USER = username
         chat_history = student_service.get_chat_history(username)
         files = student_service.get_files(username)
-        welcome_msg = lang_service.t("welcome", name=student_name)
+        assignments = student_service.get_assignments(username)
+        groups = student_service.get_groups(username)
+        welcome_msg = f"Welcome back, {student_name}!"
         return (
             gr.update(visible=False),  # Hide login
             gr.update(visible=True),   # Show main UI
             chat_history,
             files,
+            render_assignments(assignments),
+            render_groups(groups),
             welcome_msg,
             gr.update(value=student_name, visible=True),
             gr.update(visible=True)    # Show logout
@@ -448,17 +493,17 @@ def login_student(username: str, password: str) -> tuple:
         gr.update(visible=False),
         [],
         [],
-        lang_service.t("invalid_login"),
+        "",
+        "",
+        "❌ Invalid username or password",
         gr.update(visible=False),
         gr.update(visible=False)
     )
 
 def register_student(username: str, password: str, name: str) -> str:
-    """Register new student."""
     return student_service.register_student(username, password, name)
 
 def logout_student() -> tuple:
-    """Logout student."""
     global CURRENT_USER
     CURRENT_USER = "guest"
     return (
@@ -467,56 +512,66 @@ def logout_student() -> tuple:
         [],
         [],
         "",
+        "",
+        "",
         gr.update(visible=False),
         gr.update(visible=False)
     )
 
-def switch_language(lang_choice: str) -> tuple:
-    """Switch UI language."""
-    global CURRENT_LANG
-    lang_code = "tn" if "Setswana" in lang_choice else "en"
-    CURRENT_LANG = lang_code
-    lang_service.set_language(lang_code)
-
-    # Return translated UI elements
-    new_title = lang_service.t("app_title")
-    new_login_title = lang_service.t("login_title")
-    new_lang_btn_text = lang_service.t("back_to_english") if lang_code == "tn" else lang_service.t("language_toggle")
-
-    return (
-        new_title,
-        new_login_title,
-        new_lang_btn_text,
-        lang_service.t("username_label"),
-        lang_service.t("password_label"),
-        lang_service.t("login_btn"),
-        lang_service.t("register_btn"),
-        lang_service.t("ask_placeholder"),
-        lang_service.t("send_btn"),
-        lang_service.t("clear_chat"),
-        lang_service.t("upload_label"),
-        lang_service.t("upload_btn"),
-        lang_service.t("announcements_title"),
-        lang_service.t("filter_label"),
-        lang_service.t("refresh_btn"),
-        lang_service.t("teacher_login_title"),
-        lang_service.t("post_announcement"),
-        lang_service.t("announcement_title"),
-        lang_service.t("announcement_content"),
-        lang_service.t("priority_label"),
-        lang_service.t("post_btn"),
-        lang_service.t("dark_mode")
-    )
-
-def upload_file_for_student(file, username: str = "guest") -> str:
-    """Handle file upload for student."""
+def join_group(group_code: str) -> tuple:
+    if CURRENT_USER == "guest":
+        return "❌ Please log in first.", ""
+    result = student_service.join_group(CURRENT_USER, group_code)
+    groups = student_service.get_groups(CURRENT_USER)
+    return result, render_groups(groups)
+
+def leave_group(group_code: str) -> tuple:
+    if CURRENT_USER == "guest":
+        return "❌ Please log in first.", ""
+    result = student_service.leave_group(CURRENT_USER, group_code)
+    groups = student_service.get_groups(CURRENT_USER)
+    return result, render_groups(groups)
+
+def upload_file_for_student(file) -> str:
     if not file:
-        return lang_service.t("no_file")
-    result = f"{lang_service.t('file_uploaded', filename=file.name)}"
-    if username != "guest":
-        student_service.add_file(username, file.name)
+        return "❌ No file selected"
+    result = f"✅ Uploaded: {file.name}"
+    if CURRENT_USER != "guest":
+        student_service.add_file(CURRENT_USER, file.name)
     return result
 
+# ==================== VOICE INPUT (Web Speech API) ====================
+
+VOICE_JS = """
+async function startVoiceInput() {
+    if (!('webkitSpeechRecognition' in window) && !('SpeechRecognition' in window)) {
+        alert('🎙️ Voice input is not supported in your browser. Try Chrome or Edge.');
+        return '';
+    }
+
+    const recognition = new (window.SpeechRecognition || window.webkitSpeechRecognition)();
+    recognition.lang = 'en-US';
+    recognition.interimResults = false;
+    recognition.maxAlternatives = 1;
+
+    return new Promise((resolve) => {
+        recognition.start();
+        recognition.onresult = (event) => {
+            const transcript = event.results[0][0].transcript;
+            resolve(transcript);
+        };
+        recognition.onerror = (event) => {
+            alert('🎙️ Error: ' + event.error);
+            resolve('');
+        };
+    });
+}
+"""
+
+def voice_input_handler() -> str:
+    """Placeholder function — actual voice handled by JS."""
+    return ""
+
 # ==================== TEACHER FUNCTIONS ====================
 
 IS_ADMIN = False
@@ -526,7 +581,7 @@ def admin_login(username: str, password: str) -> tuple:
     if admin_service.authenticate(username, password):
         IS_ADMIN = True
         stats = school_service.get_stats()
-        stats_text = f"{lang_service.t('stats_title')}: {stats['total_announcements']} {lang_service.t('total_announcements').lower()}, {stats['total_files']} {lang_service.t('total_files').lower()}"
+        stats_text = f"📊 Stats: {stats['total_announcements']} announcements, {stats['total_files']} files"
         return (
             gr.update(visible=False),
             gr.update(visible=True),
@@ -536,7 +591,7 @@ def admin_login(username: str, password: str) -> tuple:
     return (
         gr.update(visible=True),
         gr.update(visible=False),
-        lang_service.t("invalid_login"),
+        "❌ Invalid credentials",
         gr.update(visible=False)
     )
 
@@ -552,14 +607,14 @@ def admin_logout():
 
 def post_announcement(title: str, content: str, course: str, priority: str) -> str:
     if not IS_ADMIN:
-        return lang_service.t("login_first")
+        return "🔒 Please log in first."
     if not title.strip():
-        return lang_service.t("title_required")
+        return "⚠️ Title is required."
     if not content.strip():
-        return lang_service.t("content_required")
+        return "⚠️ Content is required."
 
     school_service.add_announcement(title, content, course, priority)
-    return lang_service.t("posted_success", id=len(school_service.announcements))
+    return f"✅ Posted! 🎉 New announcement ID: {len(school_service.announcements)}"
 
 # ==================== CUSTOM CSS ====================
 
@@ -597,34 +652,23 @@ CUSTOM_CSS = """
 # ==================== BUILD UI ====================
 
 with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as demo:
-    # State variables
-    username_state = gr.State("guest")
-    lang_state = gr.State("en")
-
-    # Header with language toggle
-    with gr.Row():
-        title_display = gr.Markdown(f"# {lang_service.t('app_title')}")
-        lang_choice = gr.Dropdown(
-            choices=["🌐 English", "🌐 Setswana"],
-            value="🌐 English",
-            label="Language",
-            scale=1
-        )
+    # Header
+    gr.Markdown("# 🎓 ThutoAI — Your AI School Assistant")
 
     # Dark mode button
-    dark_mode_btn = gr.Button(lang_service.t("dark_mode"), variant="secondary")
+    dark_mode_btn = gr.Button("🌙 Toggle Dark Mode", variant="secondary")
 
     # ========= STUDENT LOGIN/REGISTER =========
     with gr.Group() as login_group:
-        gr.Markdown(f"### {lang_service.t('login_title')}")
+        gr.Markdown("### 🔐 Student Login")
         with gr.Row():
-            login_username = gr.Textbox(label=lang_service.t("username_label"), scale=3)
-            login_password = gr.Textbox(label=lang_service.t("password_label"), type="password", scale=3)
-        login_btn = gr.Button(lang_service.t("login_btn"), variant="primary")
-        register_btn = gr.Button(lang_service.t("register_btn"), variant="secondary")
+            login_username = gr.Textbox(label="Username", scale=3)
+            login_password = gr.Textbox(label="Password", type="password", scale=3)
+        login_btn = gr.Button("🔓 Login", variant="primary")
+        register_btn = gr.Button("📝 Register", variant="secondary")
         login_status = gr.Textbox(label="Status", interactive=False)
 
-        # Register modal (simplified)
+        # Register modal
         with gr.Accordion("📝 Register New Account", open=False):
             reg_username = gr.Textbox(label="Username")
             reg_password = gr.Textbox(label="Password", type="password")
@@ -635,32 +679,20 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
     # ========= MAIN APP (hidden until login) =========
     with gr.Group(visible=False) as main_app:
         with gr.Tabs():
-            with gr.Tab(lang_service.t("announcements_title")):
-                gr.Markdown(f"### {lang_service.t('announcements_title')}")
+            with gr.Tab("📢 Announcements"):
+                gr.Markdown("### Filter by Course or Subject")
                 with gr.Row():
                     course_filter = gr.Dropdown(
                         choices=school_service.courses,
                         value="All",
-                        label=lang_service.t("filter_label"),
+                        label="Select Course",
                         scale=3
                     )
-                    refresh_btn = gr.Button(lang_service.t("refresh_btn"), variant="secondary", scale=1)
+                    refresh_btn = gr.Button("🔄 Refresh", variant="secondary", scale=1)
                 announcements_html = gr.HTML()
-                course_filter.change(
-                    fn=lambda c: render_announcements(c, CURRENT_LANG),
-                    inputs=course_filter,
-                    outputs=announcements_html
-                )
-                refresh_btn.click(
-                    fn=lambda c: render_announcements(c, CURRENT_LANG),
-                    inputs=course_filter,
-                    outputs=announcements_html
-                )
-                demo.load(
-                    fn=lambda c: render_announcements(c, CURRENT_LANG),
-                    inputs=course_filter,
-                    outputs=announcements_html
-                )
+                course_filter.change(fn=render_announcements, inputs=course_filter, outputs=announcements_html)
+                refresh_btn.click(fn=render_announcements, inputs=course_filter, outputs=announcements_html)
+                demo.load(fn=render_announcements, inputs=course_filter, outputs=announcements_html)
 
             with gr.Tab("💬 Ask ThutoAI"):
                 gr.Markdown("### 💡 Ask me anything — I'm here to help!")
@@ -671,15 +703,15 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
                 )
                 with gr.Row():
                     msg = gr.Textbox(
-                        label="Your Question",
-                        placeholder=lang_service.t("ask_placeholder"),
-                        scale=8
+                        label="Type your question",
+                        placeholder="E.g., How do I prepare for the Math exam?",
+                        scale=7
                     )
-                    submit_btn = gr.Button(lang_service.t("send_btn"), variant="primary", scale=1)
-                clear_btn = gr.Button(lang_service.t("clear_chat"), variant="secondary")
+                    voice_btn = gr.Button("🎙️ Speak", variant="secondary", scale=1)
+                    submit_btn = gr.Button("➤ Send", variant="primary", scale=1)
+                clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
 
                 def respond(message, chat_history):
-                    # Use generator for streaming-like effect
                     for updated_history, _ in ai_chat(message, chat_history, CURRENT_USER):
                         yield updated_history, ""
 
@@ -687,20 +719,60 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
                 submit_btn.click(respond, [msg, chatbot], [msg, chatbot])
                 clear_btn.click(lambda: [], None, chatbot)
 
+                # Voice input (JS-based)
+                voice_btn.click(
+                    fn=voice_input_handler,
+                    inputs=None,
+                    outputs=msg,
+                    js=VOICE_JS + "return startVoiceInput();"
+                )
+
+            with gr.Tab("📅 Assignments"):
+                gr.Markdown("### 📌 Your Upcoming Assignments & Exams")
+                assignments_display = gr.HTML()
+                demo.load(
+                    fn=lambda: render_assignments(student_service.get_assignments(CURRENT_USER)) if CURRENT_USER != "guest" else "",
+                    inputs=None,
+                    outputs=assignments_display
+                )
+
+            with gr.Tab("👥 Class Groups"):
+                gr.Markdown("### 🎒 Join Your Class Groups")
+                with gr.Row():
+                    group_code_input = gr.Textbox(label="Group Code (e.g., MATH10A)", scale=3)
+                    join_btn = gr.Button("➕ Join Group", variant="primary", scale=1)
+                    leave_btn = gr.Button("➖ Leave Group", variant="secondary", scale=1)
+                group_status = gr.Textbox(label="Status")
+                groups_display = gr.HTML()
+                join_btn.click(
+                    fn=join_group,
+                    inputs=group_code_input,
+                    outputs=[group_status, groups_display]
+                )
+                leave_btn.click(
+                    fn=leave_group,
+                    inputs=group_code_input,
+                    outputs=[group_status, groups_display]
+                )
+                demo.load(
+                    fn=lambda: render_groups(student_service.get_groups(CURRENT_USER)) if CURRENT_USER != "guest" else "",
+                    inputs=None,
+                    outputs=groups_display
+                )
+
             with gr.Tab("📂 My Files"):
-                gr.Markdown("### 📁 Your Uploaded Files")
+                gr.Markdown("### 📁 Upload & Manage Study Materials")
                 with gr.Row():
-                    file_input = gr.File(label=lang_service.t("upload_label"), elem_classes=["file-upload"])
-                    upload_btn = gr.Button(lang_service.t("upload_btn"), variant="primary")
+                    file_input = gr.File(label="Drag & drop or click to upload", elem_classes=["file-upload"])
+                    upload_btn = gr.Button("📤 Upload", variant="primary")
                 upload_status = gr.Textbox(label="Status")
                 file_list = gr.JSON(label="Your Files")
 
                 upload_btn.click(
-                    fn=lambda f: upload_file_for_student(f, CURRENT_USER),
+                    fn=upload_file_for_student,
                     inputs=file_input,
                     outputs=upload_status
                 )
-                # Update file list when tab is opened (simplified)
                 demo.load(
                     fn=lambda: student_service.get_files(CURRENT_USER) if CURRENT_USER != "guest" else [],
                     inputs=None,
@@ -708,25 +780,34 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
                 )
 
             with gr.Tab("🔐 Teacher Admin"):
-                gr.Markdown(f"### {lang_service.t('teacher_login_title')}")
+                gr.Markdown("### 👩‍🏫 Post Announcements & View Stats")
 
                 with gr.Group() as teacher_login_group:
-                    teacher_username = gr.Textbox(label=lang_service.t("username_label"))
-                    teacher_password = gr.Textbox(label=lang_service.t("password_label"), type="password")
-                    teacher_login_btn = gr.Button(lang_service.t("login_btn"), variant="primary")
+                    teacher_username = gr.Textbox(label="Username")
+                    teacher_password = gr.Textbox(label="Password", type="password")
+                    teacher_login_btn = gr.Button("🔓 Login", variant="primary")
                     teacher_status = gr.Textbox(label="Status")
 
                 with gr.Group(visible=False) as teacher_dashboard:
-                    gr.Markdown(f"### {lang_service.t('post_announcement')}")
+                    gr.Markdown("### 📊 Dashboard Stats")
+                    stats = school_service.get_stats()
+                    gr.Markdown(f"""
+                    - 📢 Total Announcements: **{stats['total_announcements']}**
+                    - 📁 Total Files Uploaded: **{stats['total_files']}**
+                    - 🎯 Active Courses: **{stats['active_courses']}**
+                    - 🚨 High Priority Posts: **{stats['high_priority']}**
+                    """)
+
+                    gr.Markdown("### ✍️ Create New Announcement")
                     with gr.Row():
-                        ann_title = gr.Textbox(label=lang_service.t("announcement_title"), scale=3)
-                        ann_course = gr.Dropdown(choices=school_service.courses[1:], label=lang_service.t("course"), value="General", scale=2)
-                    ann_content = gr.Textbox(label=lang_service.t("announcement_content"), lines=3)
-                    ann_priority = gr.Radio(["low", "normal", "high"], label=lang_service.t("priority_label"), value="normal")
-                    post_btn = gr.Button(lang_service.t("post_btn"), variant="primary")
+                        ann_title = gr.Textbox(label="Title", placeholder="e.g., Quiz Moved to Friday", scale=3)
+                        ann_course = gr.Dropdown(choices=school_service.courses[1:], label="Course", value="General", scale=2)
+                    ann_content = gr.Textbox(label="Content", placeholder="Details for students...", lines=3)
+                    ann_priority = gr.Radio(["low", "normal", "high"], label="Priority", value="normal", inline=True)
+                    post_btn = gr.Button("📬 Post Announcement", variant="primary")
                     post_result = gr.Textbox(label="Result")
 
-                    teacher_logout_btn = gr.Button(lang_service.t("logout_btn"), variant="secondary")
+                    teacher_logout_btn = gr.Button("⬅️ Logout", variant="secondary")
 
                 teacher_login_btn.click(
                     fn=admin_login,
@@ -747,12 +828,12 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
         # Logout button
         with gr.Row():
             user_display = gr.Textbox(label="Logged in as", interactive=False, visible=False)
-            logout_btn = gr.Button(lang_service.t("logout_btn"), variant="secondary", visible=False)
+            logout_btn = gr.Button("⬅️ Logout", variant="secondary", visible=False)
             logout_btn.click(
                 fn=logout_student,
                 inputs=None,
                 outputs=[
-                    login_group, main_app, chatbot, file_list,
+                    login_group, main_app, chatbot, gr.update(), gr.update(), gr.update(),
                     login_status, user_display, logout_btn
                 ]
             )
@@ -762,7 +843,7 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
         fn=login_student,
         inputs=[login_username, login_password],
         outputs=[
-            login_group, main_app, chatbot, file_list,
+            login_group, main_app, chatbot, gr.update(), assignments_display, groups_display,
             login_status, user_display, logout_btn
         ]
     )
@@ -773,26 +854,6 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft(primary_hue="indigo")) as de
         outputs=reg_status
     )
 
-    lang_choice.change(
-        fn=switch_language,
-        inputs=lang_choice,
-        outputs=[
-            title_display,
-            # Update all translatable elements
-            gr.update(),  # login_title (Markdown)
-            lang_choice,  # update button text
-            login_username, login_password, login_btn, register_btn,
-            msg, submit_btn, clear_btn,
-            file_input, upload_btn,
-            gr.update(),  # announcements_title (Tab)
-            course_filter, refresh_btn,
-            gr.update(),  # teacher_login_title (Markdown)
-            gr.update(),  # post_announcement (Markdown)
-            ann_title, ann_content, ann_priority, post_btn,
-            dark_mode_btn
-        ]
-    )
-
 # Launch app
 if __name__ == "__main__":
     demo.launch()