Add Year/Level dropdown with MBBS year options

app.py

@@ -32,20 +32,21 @@ def init_database():
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             name TEXT NOT NULL,
             medical_school TEXT NOT NULL,
+            year TEXT NOT NULL,
             registration_date TIMESTAMP DEFAULT CURRENT_TIMESTAMP
         )
     """)
     conn.commit()
     conn.close()
 
-def save_student(name, medical_school):
+def save_student(name, medical_school, year):
     """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)
+            "INSERT INTO students (name, medical_school, year) VALUES (?, ?, ?)",
+            (name, medical_school, year)
         )
         conn.commit()
         conn.close()
@@ -59,7 +60,7 @@ def get_all_students():
     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")
+        cursor.execute("SELECT id, name, medical_school, year, registration_date FROM students ORDER BY registration_date DESC")
         students = cursor.fetchall()
         conn.close()
         return students
@@ -739,6 +740,12 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                     lines=1,
                     scale=1
                 )
+                welcome_year_input = gr.Dropdown(
+                    label="Year/Level",
+                    choices=["MBBS 1st Year", "MBBS 2nd Year", "MBBS 3rd Year", "MBBS Final Year", "Intern"],
+                    value=None,
+                    interactive=True
+                )
                 welcome_submit_btn = gr.Button(
                     "🚀 Start Learning",
                     variant="primary",
@@ -958,7 +965,7 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
                     logout_btn = gr.Button("🚪 Logout", variant="secondary")
                 
                 students_table = gr.Dataframe(
-                    headers=["ID", "Name", "Medical School", "Registration Date"],
+                    headers=["ID", "Name", "Medical School", "Year", "Registration Date"],
                     label="Students Database",
                     interactive=False,
                     wrap=True
@@ -967,7 +974,7 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
 
     
     # Welcome Screen Handler
-    def handle_welcome_submit(name, medical_school):
+    def handle_welcome_submit(name, medical_school, year):
         """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
@@ -975,10 +982,13 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
         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
         
+        if not year:
+            return gr.update(), gr.update(), gr.update(), gr.update(), ""  # Don't proceed if year is not selected
+        
         # Save to database
-        save_student(name.strip(), medical_school.strip())
+        save_student(name.strip(), medical_school.strip(), year)
         
-        greeting = f"**Welcome, Doctor {name}!** 👋 from {medical_school}"
+        greeting = f"**Welcome, Doctor {name}!** 👋 from {medical_school} ({year})"
         return (
             gr.update(visible=False),  # Hide welcome screen
             gr.update(visible=True),   # Show main app
@@ -989,20 +999,20 @@ with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     
     welcome_submit_btn.click(
         fn=handle_welcome_submit,
-        inputs=[welcome_name_input, welcome_school_input],
+        inputs=[welcome_name_input, welcome_school_input, welcome_year_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, welcome_school_input],
+        inputs=[welcome_name_input, welcome_school_input, welcome_year_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],
+        inputs=[welcome_name_input, welcome_school_input, welcome_year_input],
         outputs=[welcome_screen, main_app, tabs, student_name_display, student_name_state]
     )