Update app.py

app.py

@@ -1,16 +1,6 @@
 """
-SmartMate - AI Student Assistant (Upgraded for Hugging Face Spaces)
-Author: ngwakomadikwe
-Enhanced with: course filtering, admin panel, mobile UI, smarter AI context.
-
-Features:
-- Filter announcements by course/subject
-- Upload files (PDFs, notes)
-- AI chatbot with school context (OpenAI)
-- Admin login to post announcements
-- Responsive, beautiful UI
-
-Deployed on Hugging Face Spaces → https://huggingface.co/spaces/ngwakomadikwe/SmartMate
+SmartMate - AI Student Assistant (Gradio Version)
+For Hugging Face Spaces
 """
 
 import os
@@ -18,13 +8,10 @@ import gradio as gr
 from datetime import datetime
 from typing import List, Dict
 
-# ==================== MOCK DATA SERVICES (Replace with DB in future) ====================
+# ==================== MOCK SERVICES ====================
 
 class SchoolService:
-    """Manages announcements, files, and school context for AI."""
-
     def __init__(self):
-        # Sample announcements — you can replace this with database later
         self.announcements = [
             {
                 "id": 1,
@@ -52,16 +39,14 @@ class SchoolService:
             },
         ]
         self.courses = ["All", "Mathematics", "Science", "English", "History", "General"]
-        self.uploaded_files = []
+        self.files = []
 
     def get_announcements(self, course_filter: str = "All") -> List[Dict]:
-        """Return announcements filtered by course."""
         if course_filter == "All":
             return self.announcements
         return [a for a in self.announcements if a["course"] == course_filter]
 
     def add_announcement(self, title: str, content: str, course: str, priority: str):
-        """Add new announcement (used by admin)."""
         new_id = max([a["id"] for a in self.announcements], default=0) + 1
         self.announcements.append({
             "id": new_id,
@@ -73,83 +58,53 @@ class SchoolService:
         })
 
     def upload_file(self, file) -> str:
-        """Simulate file upload."""
         if not file:
             return "❌ No file selected"
-        self.uploaded_files.append({
-            "name": file.name,
-            "uploaded_at": datetime.now().strftime("%Y-%m-%d %H:%M")
-        })
+        self.files.append({"name": file.name, "uploaded_at": datetime.now().strftime("%Y-%m-%d %H:%M")})
         return f"✅ Uploaded: {file.name}"
 
     def get_school_context_for_ai(self) -> str:
-        """Return formatted context for AI to use in responses."""
-        context = "Current School Announcements (for context):\n"
-        for ann in self.announcements[:5]:  # Limit to 5 most recent
+        context = "Current School Announcements:\n"
+        for ann in self.announcements[:5]:
             context += f"- [{ann['course']}] {ann['title']}: {ann['content'][:60]}... (Priority: {ann['priority']})\n"
         return context
 
 
 class AdminService:
-    """Handles admin authentication."""
-
     def __init__(self):
-        # In production, store hashed passwords in a database!
-        self.admin_credentials = {
-            "teacher@smartmate.edu": "password123",
-            "admin@school.org": "letmein"
-        }
+        self.admins = {"teacher@smartmate.edu": "password123"}
 
     def authenticate(self, username: str, password: str) -> bool:
-        """Check if username/password is valid."""
-        return self.admin_credentials.get(username) == password
+        return self.admins.get(username) == password
 
 
 # Initialize services
 school_service = SchoolService()
 admin_service = AdminService()
 
-# Check if OpenAI key is set
+# OpenAI setup
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 USE_OPENAI = bool(OPENAI_API_KEY)
 
 if USE_OPENAI:
     try:
         from openai import OpenAI
-        openai_client = OpenAI(api_key=OPENAI_API_KEY)
-        print("✅ OpenAI client ready")
+        client = OpenAI(api_key=OPENAI_API_KEY)
     except Exception as e:
-        print(f"⚠️ Error initializing OpenAI: {e}")
+        print(f"⚠️ OpenAI error: {e}")
         USE_OPENAI = False
 else:
-    print("⚠️ OPENAI_API_KEY not set. AI will use mock responses.")
+    print("⚠️ OPENAI_API_KEY not set. Using mock responses.")
 
-# ==================== AI CHAT FUNCTION ====================
-
-def smart_chat(message: str, history: List) -> str:
-    """
-    Generate AI response using OpenAI (or mock if no key).
-    Injects school announcements as context for better, relevant answers.
-    """
+def ai_chat(message: str, history: List) -> str:
     if USE_OPENAI:
         try:
-            # Get school context to make AI responses relevant
-            context = school_service.get_school_context_for_ai()
-
-            system_prompt = f"""You are SmartMate, a friendly AI assistant for students.
-Use the school context below to give accurate, helpful answers.
+            system_prompt = f"""You are SmartMate, an AI assistant for students.
+Use this context to answer accurately:
+{school_service.get_school_context_for_ai()}
 
-{context}
-
-Guidelines:
-- Be encouraging and supportive.
-- If asked about exams, assignments, or announcements, refer to the context.
-- Offer study tips if appropriate.
-- If unsure, say so — don't guess.
-
-Now respond to the student's question:"""
-
-            response = openai_client.chat.completions.create(
+Be helpful, encouraging, and accurate. If unsure, say so."""
+            response = client.chat.completions.create(
                 model="gpt-3.5-turbo",
                 messages=[
                     {"role": "system", "content": system_prompt},
@@ -160,235 +115,136 @@ Now respond to the student's question:"""
             )
             reply = response.choices[0].message.content.strip()
 
-            # If question relates to school, append matching announcements
-            keywords = ["exam", "test", "due", "assignment", "announcement", "when", "what", "deadline"]
+            # Add related announcements
+            keywords = ["exam", "test", "due", "assignment"]
             if any(kw in message.lower() for kw in keywords):
                 matches = school_service.get_announcements()
                 relevant = [a for a in matches if any(kw in a["title"].lower() or kw in a["content"].lower() for kw in keywords)]
                 if relevant:
                     reply += "\n\n📌 **Relevant Announcements:**"
-                    for ann in relevant[:2]:  # Show top 2
-                        reply += f"\n- **{ann['title']}** ({ann['course']}) → {ann['content'][:70]}..."
+                    for ann in relevant[:2]:
+                        reply += f"\n- {ann['title']} ({ann['course']}) → {ann['content'][:70]}..."
 
             return reply
-
         except Exception as e:
-            return f"⚠️ Sorry, I had trouble thinking. Error: {str(e)}"
+            return f"⚠️ AI Error: {str(e)}"
     else:
-        # Mock response if no API key
-        return f"👋 Hi! I’m SmartMate. You asked: '{message}'.\n*(Tip: Set OPENAI_API_KEY in HF Secrets for real AI answers.)*"
+        return f"👋 Hi! I'm SmartMate. You asked: '{message}'. *(Set OPENAI_API_KEY in HF Secrets for real AI.)*"
 
 
-# ==================== UI RENDERING HELPERS ====================
+# ==================== GRADIO INTERFACE ====================
 
 def render_announcements(course: str) -> str:
-    """Generate HTML to display announcements with styling."""
     announcements = school_service.get_announcements(course)
     if not announcements:
-        return "<p style='color: #6c757d; text-align: center;'>No announcements for this course.</p>"
-
+        return "<p style='color: #6c757d; text-align: center;'>No announcements.</p>"
     html = ""
-    priority_colors = {"high": "#dc3545", "normal": "#ffc107", "low": "#6c757d"}
-
     for ann in announcements:
-        color = priority_colors.get(ann["priority"], "#6c757d")
+        color = "red" if ann["priority"] == "high" else "orange" if ann["priority"] == "normal" else "gray"
         html += f"""
         <div style="
-            border: 1px solid #e9ecef;
+            border: 1px solid #e0e0e0;
             border-radius: 8px;
-            padding: 16px;
-            margin-bottom: 12px;
-            background: #f8f9fa;
+            padding: 12px;
+            margin: 8px 0;
+            background: white;
         ">
-            <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 8px;">
-                <strong style="font-size: 1.1em; color: #212529;">{ann['title']}</strong>
-                <span style="
-                    background: {color};
-                    color: white;
-                    padding: 3px 8px;
-                    border-radius: 4px;
-                    font-size: 0.8em;
-                    font-weight: bold;
-                ">{ann['priority'].upper()}</span>
-            </div>
-            <div style="margin-bottom: 8px;">{ann['content']}</div>
-            <div style="font-size: 0.85em; color: #6c757d;">
-                📚 {ann['course']} &nbsp; | &nbsp; 📅 {ann['date']}
+            <div style="display: flex; justify-content: space-between;">
+                <strong>{ann['title']}</strong>
+                <span style="background: {color}; color: white; padding: 2px 6px; border-radius: 4px; font-size: 0.7em;">{ann['priority'].upper()}</span>
             </div>
+            <div style="margin-top: 4px; color: #555;">{ann['content']}</div>
+            <div style="font-size: 0.8em; color: #999;">{ann['course']} • {ann['date']}</div>
         </div>
         """
     return html
 
 
-# ==================== ADMIN STATE & FUNCTIONS ====================
-
-# Track if user is logged in as admin (simple state — not secure for production)
+# State for admin login
 IS_ADMIN = False
 
 def admin_login(username: str, password: str) -> tuple:
-    """Attempt to log in admin. Returns UI update instructions."""
     global IS_ADMIN
     if admin_service.authenticate(username, password):
         IS_ADMIN = True
         return (
-            gr.update(visible=False),  # Hide login form
-            gr.update(visible=True),   # Show dashboard
-            "✅ Login successful! You can now post announcements.",
-            gr.update(visible=True)    # Show post form
-        )
-    else:
-        return (
-            gr.update(visible=True),
             gr.update(visible=False),
-            "❌ Invalid username or password. Try again.",
-            gr.update(visible=False)
+            gr.update(visible=True),
+            "✅ Login successful!",
+            gr.update(visible=True)
         )
+    return (
+        gr.update(visible=True),
+        gr.update(visible=False),
+        "❌ Invalid credentials",
+        gr.update(visible=False)
+    )
 
-def admin_logout() -> tuple:
-    """Log out admin."""
+def admin_logout():
     global IS_ADMIN
     IS_ADMIN = False
     return (
-        gr.update(visible=True),   # Show login
-        gr.update(visible=False),  # Hide dashboard
+        gr.update(visible=True),
+        gr.update(visible=False),
         "",
-        gr.update(visible=False)   # Hide post form
+        gr.update(visible=False)
     )
 
-def admin_post_announcement(title: str, content: str, course: str, priority: str) -> str:
-    """Post new announcement if admin."""
+def post_announcement(title: str, content: str, course: str, priority: str) -> str:
     if not IS_ADMIN:
-        return "🔒 You must be logged in as admin to post announcements."
-    if not title.strip() or not content.strip():
-        return "⚠️ Title and content cannot be empty."
+        return "🔒 Only admins can post."
+    if not title or not content:
+        return "⚠️ Title and content required."
     school_service.add_announcement(title, content, course, priority)
-    return "✅ Announcement posted successfully! Refresh announcements to see it."
+    return "✅ Posted!"
 
-# ==================== CUSTOM CSS FOR MOBILE & BEAUTY ====================
+# ==================== BUILD UI ====================
 
-CUSTOM_CSS = """
+with gr.Blocks(css="""
 .gradio-container {
-    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif;
-}
-.announcement-box {
-    border: 1px solid #e0e0e0;
-    border-radius: 10px;
-    padding: 16px;
-    margin: 10px 0;
-    background: white;
-    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+    font-family: 'Inter', sans-serif;
 }
-.priority-badge {
-    font-size: 0.75em;
-    padding: 3px 8px;
-    border-radius: 4px;
-    font-weight: bold;
-    color: white;
-}
-.high { background: #e74c3c; }
-.normal { background: #f39c12; }
-.low { background: #7f8c8d; }
-"""
+""") as demo:
+    gr.Markdown("# 🎓 SmartMate - AI Student Assistant")
 
-# ==================== BUILD GRADIO INTERFACE ====================
-
-with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🎓 SmartMate — Your AI School Assistant")
-
-    # ========= TAB: Announcements =========
     with gr.Tab("📢 Announcements"):
-        gr.Markdown("### Filter by Course/Subject")
-        course_dropdown = gr.Dropdown(
-            choices=school_service.courses,
-            value="All",
-            label="Select Course",
-            interactive=True
-        )
+        course_filter = gr.Dropdown(choices=school_service.courses, value="All", label="Filter by Course")
         announcements_html = gr.HTML()
-        course_dropdown.change(
-            fn=render_announcements,
-            inputs=course_dropdown,
-            outputs=announcements_html
-        )
-        # Load initial announcements
-        demo.load(
-            fn=render_announcements,
-            inputs=course_dropdown,
-            outputs=announcements_html
-        )
-
-    # ========= TAB: AI Chat =========
-    with gr.Tab("💬 Ask SmartMate"):
-        gr.Markdown("### Ask me anything — homework, deadlines, school info, or just for motivation!")
-        chatbot = gr.Chatbot(height=450, bubble_full_width=False)
-        msg = gr.Textbox(label="Type your question", placeholder="E.g., What's due this week in Math?")
-        clear_btn = gr.Button("🗑️ Clear Chat")
-
-        def respond(message, chat_history):
-            bot_reply = smart_chat(message, chat_history)
-            chat_history.append((message, bot_reply))
-            return "", chat_history
-
-        msg.submit(respond, [msg, chatbot], [msg, chatbot])
-        clear_btn.click(lambda: None, None, chatbot, queue=False)
-
-    # ========= TAB: Upload Files =========
-    with gr.Tab("📂 Upload Notes/PDFs"):
-        gr.Markdown("### Upload study materials for personal reference (not yet stored long-term)")
-        file_uploader = gr.File(label="Choose a file (PDF, DOC, TXT, etc.)")
-        upload_btn = gr.Button("📤 Upload")
-        upload_status = gr.Textbox(label="Status", interactive=False)
-        upload_btn.click(
-            fn=school_service.upload_file,
-            inputs=file_uploader,
-            outputs=upload_status
-        )
-
-    # ========= TAB: Admin Panel =========
-    with gr.Tab("🔐 Admin (Teachers)"):
-        gr.Markdown("### Post Announcements to Students")
-
-        # Login Form
+        course_filter.change(fn=render_announcements, inputs=course_filter, outputs=announcements_html)
+        demo.load(fn=render_announcements, inputs=course_filter, outputs=announcements_html)
+
+    with gr.Tab("💬 Chat"):
+        chatbot = gr.Chatbot(height=400)
+        msg = gr.Textbox(label="Ask me anything...")
+        clear = gr.Button("Clear")
+        msg.submit(fn=ai_chat, inputs=[msg, chatbot], outputs=[msg, chatbot])
+        clear.click(lambda: None, None, chatbot)
+
+    with gr.Tab("📂 Upload Files"):
+        file_input = gr.File(label="Upload PDF/Notes")
+        upload_btn = gr.Button("Upload")
+        status = gr.Textbox(label="Status")
+        upload_btn.click(fn=school_service.upload_file, inputs=file_input, outputs=status)
+
+    with gr.Tab("🔐 Admin"):
         with gr.Group() as login_group:
-            gr.Markdown("#### Login to Post Announcements")
-            admin_username = gr.Textbox(label="Username", placeholder="e.g., teacher@smartmate.edu")
-            admin_password = gr.Textbox(label="Password", type="password", placeholder="••••••••")
-            login_btn = gr.Button("🔓 Login")
-            login_status = gr.Textbox(label="Login Status", interactive=False)
-
-        # Admin Dashboard (hidden until login)
-        with gr.Group(visible=False) as admin_dashboard:
-            gr.Markdown("#### ✍️ Create New Announcement")
-            with gr.Row():
-                ann_title = gr.Textbox(label="Title", placeholder="e.g., Science Quiz Moved")
-                ann_course = gr.Dropdown(choices=school_service.courses[1:], label="Course", value="General")
-            ann_content = gr.Textbox(label="Content", placeholder="Details for students...", lines=3)
-            ann_priority = gr.Radio(["low", "normal", "high"], label="Priority", value="normal")
-            post_btn = gr.Button("📬 Post Announcement")
-            post_status = gr.Textbox(label="Result", interactive=False)
-
-            # Post logic
-            post_btn.click(
-                fn=admin_post_announcement,
-                inputs=[ann_title, ann_content, ann_course, ann_priority],
-                outputs=post_status
-            )
-
-            # Logout button
-            logout_btn = gr.Button("⬅️ Logout")
-            logout_btn.click(
-                fn=admin_logout,
-                inputs=None,
-                outputs=[login_group, admin_dashboard, login_status, post_status]
-            )
+            username = gr.Textbox(label="Username")
+            password = gr.Textbox(label="Password", type="password")
+            login_btn = gr.Button("Login")
+            login_status = gr.Textbox(label="Status")
+        with gr.Group(visible=False) as dashboard:
+            title = gr.Textbox(label="Title")
+            content = gr.Textbox(label="Content", lines=3)
+            course = gr.Dropdown(choices=school_service.courses[1:], label="Course")
+            priority = gr.Radio(["low", "normal", "high"], label="Priority")
+            post_btn = gr.Button("Post")
+            post_result = gr.Textbox(label="Result")
+            logout_btn = gr.Button("Logout")
+
+        login_btn.click(fn=admin_login, inputs=[username, password], outputs=[login_group, dashboard, login_status, post_btn])
+        logout_btn.click(fn=admin_logout, inputs=None, outputs=[login_group, dashboard, login_status, post_result])
+        post_btn.click(fn=post_announcement, inputs=[title, content, course, priority], outputs=post_result)
 
-        # Login logic
-        login_btn.click(
-            fn=admin_login,
-            inputs=[admin_username, admin_password],
-            outputs=[login_group, admin_dashboard, login_status, post_status]
-        )
 
-# Launch app
 if __name__ == "__main__":
     demo.launch()