Update app.py

app.py

@@ -4,125 +4,114 @@ import string
 import time
 from collections import defaultdict
 
-# Global variables
-students = {}
-class_data = defaultdict(lambda: {"students": set(), "questions": []})
-
 def generate_student_id():
     return ''.join(random.choices(string.ascii_letters + string.digits, k=12))
 
-def update_student_state(student_id, class_id, color):
-    if student_id not in students:
-        students[student_id] = {"class_id": class_id, "color": "inactive"}
-    students[student_id]["color"] = color
-    class_data[class_id]["students"].add(student_id)
-    return f"Status updated to {color}"
+def update_student_state(students, class_id, student_id, color):
+    if class_id not in students:
+        students[class_id] = {}
+    students[class_id][student_id] = {"color": color}
+    return students, f"Status updated to {color}"
+
+def submit_question(questions, class_id, student_id, question):
+    if class_id not in questions:
+        questions[class_id] = []
+    questions[class_id].append({"student_id": student_id, "question": question})
+    return questions, "Question submitted successfully"
 
-def submit_question(student_id, class_id, question):
-    class_data[class_id]["questions"].append({"student_id": student_id, "question": question})
-    return "Question submitted successfully"
+def get_class_stats(students, questions, class_id):
+    if class_id not in students:
+        return "No data available for this class."
 
-def get_class_stats(class_id):
-    class_students = [s for s in students.values() if s["class_id"] == class_id]
+    class_students = students[class_id]
     total_students = len(class_students)
-    active_students = sum(1 for s in class_students if s["color"] != "inactive")
+    active_students = sum(1 for s in class_students.values() if s["color"] != "inactive")
     color_counts = {"green": 0, "yellow": 0, "red": 0}
-    for student in class_students:
+    
+    for student in class_students.values():
         if student["color"] in color_counts:
             color_counts[student["color"]] += 1
     
     color_fractions = {color: count / (active_students or 1) for color, count in color_counts.items()}
     
-    return {
-        "total_students": total_students,
-        "active_students": active_students,
-        "color_fractions": color_fractions,
-        "questions": class_data[class_id]["questions"]
-    }
-
-def create_student_interface(class_id):
-    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.")
+    color_chart = f"""
+    <div style="width: 100%; height: 60px; display: flex;">
+        <div style="width: {color_fractions['red']*100}%; background-color: #ff6b6b;"></div>
+        <div style="width: {color_fractions['yellow']*100}%; background-color: #feca57;"></div>
+        <div style="width: {color_fractions['green']*100}%; background-color: #5cd85c;"></div>
+    </div>
+    """
+    stats_text = f"""
+    <h3>Class Statistics</h3>
+    <p>Total Students: {total_students}</p>
+    <p>Active Students: {active_students}</p>
+    """
+    questions_text = "<h3>Student Questions</h3>"
+    for q in questions.get(class_id, []):
+        questions_text += f"<p><strong>Student {q['student_id'][:4]}...</strong>: {q['question']}</p>"
+    
+    return f"{color_chart}<br>{stats_text}<br>{questions_text}"
+
+def create_interface():
+    with gr.Blocks(theme=gr.themes.Soft()) as app:
+        students = gr.State({})
+        questions = gr.State({})
+        class_id = gr.State("")
+        student_id = gr.State("")
+
+        gr.Markdown("# Fastcups")
         
-        with gr.Row():
-            with gr.Column(scale=2):
+        with gr.Tab("Setup"):
+            class_input = gr.Textbox(label="Enter Class ID")
+            setup_button = gr.Button("Set Class ID")
+
+        with gr.Tab("Student"):
+            gr.Markdown("## Student Interface")
+            status = gr.Textbox(label="Current Status", interactive=False)
+            with gr.Row():
                 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)
+            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.Tab("Teacher"):
+            gr.Markdown("## Teacher Interface")
+            stats_html = gr.HTML()
+            refresh_button = gr.Button("Refresh Stats")
+
+        def setup_class(class_id_input):
+            return class_id_input, generate_student_id()
+
+        setup_button.click(setup_class, inputs=[class_input], outputs=[class_id, student_id])
 
         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
+                inputs=[students, class_id, student_id, gr.State(color)],
+                outputs=[students, status]
             )
         
         submit_button.click(
             submit_question,
-            inputs=[student_id, gr.State(class_id), question_input],
-            outputs=question_status
+            inputs=[questions, class_id, student_id, question_input],
+            outputs=[questions, question_status]
         )
 
-    return student_interface
-
-def create_teacher_interface(class_id):
-    def render_stats():
-        stats = get_class_stats(class_id)
-        color_chart = f"""
-        <div style="width: 100%; height: 60px; display: flex;">
-            <div style="width: {stats['color_fractions']['red']*100}%; background-color: #ff6b6b;"></div>
-            <div style="width: {stats['color_fractions']['yellow']*100}%; background-color: #feca57;"></div>
-            <div style="width: {stats['color_fractions']['green']*100}%; background-color: #5cd85c;"></div>
-        </div>
-        """
-        stats_text = f"""
-        <h3>Class Statistics</h3>
-        <p>Total Students: {stats['total_students']}</p>
-        <p>Active Students: {stats['active_students']}</p>
-        """
-        questions_text = "<h3>Student Questions</h3>"
-        for q in stats['questions']:
-            questions_text += f"<p><strong>Student {q['student_id'][:4]}...</strong>: {q['question']}</p>"
-        
-        return f"{color_chart}<br>{stats_text}<br>{questions_text}"
-
-    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(
+            get_class_stats,
+            inputs=[students, questions, class_id],
+            outputs=[stats_html]
+        )
 
         # Auto-refresh every 5 seconds
         gr.HTML('<script>setInterval(function(){ document.querySelector("button[aria-label=\'Refresh Stats\']").click(); }, 5000);</script>')
 
-    return teacher_interface
-
-def launch_app(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.launch(share=True)
+    return app
 
 if __name__ == "__main__":
-    class_id = "Google I/O Extended Kolkata"
-    launch_app(class_id)
+    app = create_interface()
+    app.launch()