Update app.py

app.py

@@ -41,43 +41,40 @@ def get_class_stats(class_id):
     }
 
 def create_student_interface(class_id):
-    student_id = gr.State(generate_student_id())
-    
-    with gr.Row():
-        with gr.Column(scale=2):
-            color_buttons = [
-                gr.Button("🟢 I'm following along", variant="primary"),
-                gr.Button("🟡 I need clarification", variant="secondary"),
-                gr.Button("🔴 I'm lost, please stop", variant="stop")
-            ]
-        with gr.Column(scale=1):
-            status = gr.Textbox(label="Current Status", interactive=False)
-    
-    question_input = gr.Textbox(label="Ask a question")
-    submit_button = gr.Button("Submit Question")
-    question_status = gr.Textbox(label="Question Status", interactive=False)
+    with gr.Blocks() as student_interface:
+        student_id = gr.State(generate_student_id())
+        
+        gr.Markdown(f"## Student Interface - Class {class_id}")
+        gr.Markdown("Use the buttons to indicate your understanding and ask questions.")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
+                color_buttons = [
+                    gr.Button("🟢 I'm following along", variant="primary"),
+                    gr.Button("🟡 I need clarification", variant="secondary"),
+                    gr.Button("🔴 I'm lost, please stop", variant="stop")
+                ]
+            with gr.Column(scale=1):
+                status = gr.Textbox(label="Current Status", interactive=False)
+        
+        question_input = gr.Textbox(label="Ask a question")
+        submit_button = gr.Button("Submit Question")
+        question_status = gr.Textbox(label="Question Status", interactive=False)
 
-    for button, color in zip(color_buttons, ["green", "yellow", "red"]):
-        button.click(
-            update_student_state,
-            inputs=[student_id, gr.State(class_id), gr.State(color)],
-            outputs=status
+        for button, color in zip(color_buttons, ["green", "yellow", "red"]):
+            button.click(
+                update_student_state,
+                inputs=[student_id, gr.State(class_id), gr.State(color)],
+                outputs=status
+            )
+        
+        submit_button.click(
+            submit_question,
+            inputs=[student_id, gr.State(class_id), question_input],
+            outputs=question_status
         )
-    
-    submit_button.click(
-        submit_question,
-        inputs=[student_id, gr.State(class_id), question_input],
-        outputs=question_status
-    )
 
-    return gr.Interface(
-        fn=lambda: None,
-        inputs=[],
-        outputs=[],
-        title=f"Student Interface - Class {class_id}",
-        description="Use the buttons to indicate your understanding and ask questions.",
-        theme="default"
-    )
+    return student_interface
 
 def create_teacher_interface(class_id):
     def render_stats():
@@ -100,30 +97,27 @@ def create_teacher_interface(class_id):
         
         return f"{color_chart}<br>{stats_text}<br>{questions_text}"
 
-    stats_html = gr.HTML()
-    refresh_button = gr.Button("Refresh Stats")
+    with gr.Blocks() as teacher_interface:
+        gr.Markdown(f"## Teacher Interface - Class {class_id}")
+        gr.Markdown("Monitor student understanding and view questions.")
+        
+        stats_html = gr.HTML()
+        refresh_button = gr.Button("Refresh Stats")
 
-    refresh_button.click(render_stats, inputs=[], outputs=[stats_html])
+        refresh_button.click(render_stats, inputs=[], outputs=[stats_html])
 
-    return gr.Interface(
-        fn=lambda: None,
-        inputs=[],
-        outputs=[],
-        title=f"Teacher Interface - Class {class_id}",
-        description="Monitor student understanding and view questions.",
-        theme="default"
-    )
+    return teacher_interface
 
 def launch_app(class_id):
-    student_interface = create_student_interface(class_id)
-    teacher_interface = create_teacher_interface(class_id)
+    with gr.Blocks(theme=gr.themes.Soft()) as app:
+        gr.Markdown(f"# Fastcups - Class {class_id}")
+        
+        with gr.Tabs():
+            with gr.TabItem("Student"):
+                create_student_interface(class_id)
+            with gr.TabItem("Teacher"):
+                create_teacher_interface(class_id)
     
-    app = gr.TabbedInterface(
-        [student_interface, teacher_interface],
-        ["Student", "Teacher"],
-        title=f"Fastcups - Class {class_id}",
-        theme=gr.themes.Soft()
-    )
     app.launch()
 
 if __name__ == "__main__":