Add student database with medical school field and password-protected admin panel

app.py

@@ -7,6 +7,8 @@ from PIL import Image
 import PyPDF2
 from pdf2image import convert_from_path
 import tempfile
+import sqlite3
+from datetime import datetime
 
 # Load environment variables
 load_dotenv()
@@ -15,6 +17,60 @@ SERPAPI_KEY = os.getenv("SERPAPI_KEY")
 HYPERBOLIC_API_KEY = os.getenv("HYPERBOLIC_API_KEY")
 ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY")
 
+# Admin password
+ADMIN_PASSWORD = "BT54iv!@"
+
+# Database setup
+DB_PATH = "students.db"
+
+def init_database():
+    """Initialize the SQLite database and create students table if it doesn't exist."""
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS students (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            name TEXT NOT NULL,
+            medical_school TEXT NOT NULL,
+            registration_date TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+        )
+    """)
+    conn.commit()
+    conn.close()
+
+def save_student(name, medical_school):
+    """Save student information to the database."""
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute(
+            "INSERT INTO students (name, medical_school) VALUES (?, ?)",
+            (name, medical_school)
+        )
+        conn.commit()
+        conn.close()
+        return True
+    except Exception as e:
+        print(f"Error saving student: {e}")
+        return False
+
+def get_all_students():
+    """Retrieve all students from the database."""
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute("SELECT id, name, medical_school, registration_date FROM students ORDER BY registration_date DESC")
+        students = cursor.fetchall()
+        conn.close()
+        return students
+    except Exception as e:
+        print(f"Error retrieving students: {e}")
+        return []
+
+# Initialize database on startup
+init_database()
+
+
 # Hyperbolic API configuration
 HYPERBOLIC_API_URL = "https://api.hyperbolic.xyz/v1/chat/completions"
 HYPERBOLIC_MODEL = "meta-llama/Llama-3.3-70B-Instruct"
@@ -665,7 +721,7 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
         )
         gr.Markdown("---")
         gr.Markdown("## 👨‍⚕️ Let's Get Started!")
-        gr.Markdown("Please enter your name to begin your personalized learning journey.")
+        gr.Markdown("Please enter your information to begin your personalized learning journey.")
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -677,6 +733,12 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                     lines=1,
                     scale=1
                 )
+                welcome_school_input = gr.Textbox(
+                    label="Medical School",
+                    placeholder="Enter your medical school (e.g., King Saud University)",
+                    lines=1,
+                    scale=1
+                )
                 welcome_submit_btn = gr.Button(
                     "🚀 Start Learning",
                     variant="primary",
@@ -697,23 +759,24 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
         
         # Display student name
         student_name_display = gr.Markdown("")
-    
-    with gr.Tabs(visible=False) as tabs:
-        # LEARNING MODE TAB
-        with gr.Tab("📚 Learning Mode"):
-            # Search and examples at the top
-            with gr.Row():
-                query_input = gr.Textbox(
-                    label="Ask an Anatomy Question",
-                    placeholder="e.g., Show me the Circle of Willis",
-                    lines=2
-                )
-            
-            # Examples
-            gr.Examples(
-                examples=[
-                    ["Show me the Circle of Willis"],
-                    ["Brachial plexus anatomy"],
+        
+        # Tabs are inside main_app so they're hidden initially
+        with gr.Tabs() as tabs:
+            # LEARNING MODE TAB
+            with gr.Tab("📚 Learning Mode"):
+                # Search and examples at the top
+                with gr.Row():
+                    query_input = gr.Textbox(
+                        label="Ask an Anatomy Question",
+                        placeholder="e.g., Show me the Circle of Willis",
+                        lines=2
+                    )
+                
+                # Examples
+                gr.Examples(
+                    examples=[
+                        ["Show me the Circle of Willis"],
+                        ["Brachial plexus anatomy"],
                     ["Carpal bones arrangement"],
                     ["Layers of the scalp"],
                     ["Anatomy of the heart chambers"],
@@ -867,15 +930,55 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
             outputs=[current_question_idx]
         )
         
+        # ADMIN PANEL TAB
+        with gr.Tab("🔐 Admin Panel") as admin_tab:
+            gr.Markdown("## Admin Panel - Student Database")
+            gr.Markdown("Enter the admin password to view registered students.")
+            
+            # Password input
+            with gr.Row():
+                admin_password_input = gr.Textbox(
+                    label="Admin Password",
+                    placeholder="Enter admin password",
+                    type="password",
+                    scale=2
+                )
+                admin_login_btn = gr.Button("🔓 Login", variant="primary", scale=1)
+            
+            admin_status = gr.Markdown("")
+            
+            # Admin content (hidden until authenticated)
+            with gr.Column(visible=False) as admin_content:
+                gr.Markdown("### 📊 Registered Students")
+                
+                admin_stats = gr.Markdown("")
+                
+                with gr.Row():
+                    refresh_btn = gr.Button("🔄 Refresh Data", variant="secondary")
+                    logout_btn = gr.Button("🚪 Logout", variant="secondary")
+                
+                students_table = gr.Dataframe(
+                    headers=["ID", "Name", "Medical School", "Registration Date"],
+                    label="Students Database",
+                    interactive=False,
+                    wrap=True
+                )
+        
 
     
     # Welcome Screen Handler
-    def handle_welcome_submit(name):
+    def handle_welcome_submit(name, medical_school):
         """Handle welcome screen submission."""
         if not name or not name.strip():
             return gr.update(), gr.update(), gr.update(), gr.update(), ""  # Don't proceed if name is empty
         
-        greeting = f"**Welcome, Doctor {name}!** 👋"
+        if not medical_school or not medical_school.strip():
+            return gr.update(), gr.update(), gr.update(), gr.update(), ""  # Don't proceed if medical school is empty
+        
+        # Save to database
+        save_student(name.strip(), medical_school.strip())
+        
+        greeting = f"**Welcome, Doctor {name}!** 👋 from {medical_school}"
         return (
             gr.update(visible=False),  # Hide welcome screen
             gr.update(visible=True),   # Show main app
@@ -886,14 +989,20 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     
     welcome_submit_btn.click(
         fn=handle_welcome_submit,
-        inputs=[welcome_name_input],
+        inputs=[welcome_name_input, welcome_school_input],
         outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
     )
     
     # Also allow Enter key to submit
     welcome_name_input.submit(
         fn=handle_welcome_submit,
-        inputs=[welcome_name_input],
+        inputs=[welcome_name_input, welcome_school_input],
+        outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
+    )
+    
+    welcome_school_input.submit(
+        fn=handle_welcome_submit,
+        inputs=[welcome_name_input, welcome_school_input],
         outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
     )
     
@@ -953,6 +1062,66 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
             question_audio  # Output audio for next question
         ]
     )
+    
+    # Admin Panel Handlers
+    def admin_login(password):
+        """Verify admin password and show admin content."""
+        if password == ADMIN_PASSWORD:
+            students = get_all_students()
+            total_students = len(students)
+            stats = f"**Total Registered Students:** {total_students}"
+            return (
+                gr.update(value="✅ Login successful!", visible=True),
+                gr.update(visible=True),  # Show admin content
+                stats,
+                students
+            )
+        else:
+            return (
+                gr.update(value="❌ Invalid password. Access denied.", visible=True),
+                gr.update(visible=False),  # Hide admin content
+                "",
+                []
+            )
+    
+    def admin_logout():
+        """Logout from admin panel."""
+        return (
+            gr.update(value=""),  # Clear password
+            gr.update(value=""),  # Clear status
+            gr.update(visible=False),  # Hide admin content
+            "",  # Clear stats
+            []  # Clear table
+        )
+    
+    def refresh_students():
+        """Refresh the students table."""
+        students = get_all_students()
+        total_students = len(students)
+        stats = f"**Total Registered Students:** {total_students}"
+        return stats, students
+    
+    admin_login_btn.click(
+        fn=admin_login,
+        inputs=[admin_password_input],
+        outputs=[admin_status, admin_content, admin_stats, students_table]
+    )
+    
+    admin_password_input.submit(
+        fn=admin_login,
+        inputs=[admin_password_input],
+        outputs=[admin_status, admin_content, admin_stats, students_table]
+    )
+    
+    logout_btn.click(
+        fn=admin_logout,
+        outputs=[admin_password_input, admin_status, admin_content, admin_stats, students_table]
+    )
+    
+    refresh_btn.click(
+        fn=refresh_students,
+        outputs=[admin_stats, students_table]
+    )
 
 if __name__ == "__main__":
     # Check if API keys are configured