Update app.py

app.py

@@ -462,6 +462,9 @@ class StudentSession:
 
 session = StudentSession()
 
+# In-memory submissions store for student results
+submissions_storage = []
+
 # ---------- 7. Download Functions ----------
 def download_questions_file(questions_list, topic, grade_level, subject=None):
     """Download questions as text file"""
@@ -901,12 +904,50 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 if canvas_input is not None:
                     # Map the same canvas image to all question indices so each question gets graded
                     session.current_answers = {i: canvas_input for i in range(len(session.current_questions))}
-                    return f" Canvas submission received with {len(session.current_questions)} questions. Processing..."
+                    # Automatically grade and store results
+                    feedback = grade_student_answers(
+                        session.current_questions,
+                        session.current_answers,
+                        session.current_grade,
+                        session.current_topic,
+                        getattr(session, 'current_subject', None)
+                    )
+                    session.last_feedback = feedback
+                    session.correction_conversation = [{'role':'assistant', 'content': feedback}]
+                    # Save to submissions storage
+                    submissions_storage.append({
+                        'student': session.student_name,
+                        'timestamp': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                        'subject': getattr(session, 'current_subject', None),
+                        'grade_level': session.current_grade,
+                        'topic': session.current_topic,
+                        'answers': 'image',
+                        'feedback': feedback
+                    })
+                    return f"✅ Canvas submission received and graded. View feedback in 'AI Correction' tab."
                 
                 elif upload_input is not None:
                     # Map the uploaded image to all question indices so AI can analyse each part
                     session.current_answers = {i: upload_input for i in range(len(session.current_questions))}
-                    return f" Photo submission received with {len(session.current_questions)} questions. Processing..."
+                    feedback = grade_student_answers(
+                        session.current_questions,
+                        session.current_answers,
+                        session.current_grade,
+                        session.current_topic,
+                        getattr(session, 'current_subject', None)
+                    )
+                    session.last_feedback = feedback
+                    session.correction_conversation = [{'role':'assistant', 'content': feedback}]
+                    submissions_storage.append({
+                        'student': session.student_name,
+                        'timestamp': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                        'subject': getattr(session, 'current_subject', None),
+                        'grade_level': session.current_grade,
+                        'topic': session.current_topic,
+                        'answers': 'image',
+                        'feedback': feedback
+                    })
+                    return f"✅ Photo submission received and graded. View feedback in 'AI Correction' tab."
                 
                 elif typed_input and typed_input.strip():
                     # Parse typed answers with improved logic
@@ -969,7 +1010,26 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                     if session.current_answers:
                         answered = len(session.current_answers)
                         total = len(session.current_questions)
-                        return f"✅ Received {answered}/{total} answer(s). Processing for AI correction..."
+                        # Automatically grade textual answers as well
+                        feedback = grade_student_answers(
+                            session.current_questions,
+                            session.current_answers,
+                            session.current_grade,
+                            session.current_topic,
+                            getattr(session, 'current_subject', None)
+                        )
+                        session.last_feedback = feedback
+                        session.correction_conversation = [{'role':'assistant', 'content': feedback}]
+                        submissions_storage.append({
+                            'student': session.student_name,
+                            'timestamp': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                            'subject': getattr(session, 'current_subject', None),
+                            'grade_level': session.current_grade,
+                            'topic': session.current_topic,
+                            'answers': session.current_answers,
+                            'feedback': feedback
+                        })
+                        return f"✅ Received {answered}/{total} answer(s) and graded. View feedback in 'AI Correction' tab."
                     else:
                         return "⚠️ Could not find any answers. Please type your answers using one of the suggested formats."
                 else:
@@ -1059,6 +1119,28 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 outputs=download_feedback_btn
             )
 
+            # View student results from submissions_storage
+            results_output = gr.Markdown()
+            results_btn = gr.Button(" Show My Results", variant="secondary")
+
+            def view_student_results(name):
+                student = name if name else session.student_name
+                items = [s for s in submissions_storage if s.get('student') == student]
+                if not items:
+                    return f"No submissions found for {student}."
+                out = f"### 📋 Submissions for {student}\n"
+                for it in sorted(items, key=lambda x: x['timestamp'], reverse=True):
+                    subj = it.get('subject') or 'N/A'
+                    ts = it.get('timestamp')
+                    topic = it.get('topic') or 'N/A'
+                    fb = it.get('feedback')
+                    # show a short preview of feedback
+                    preview = fb[:300] + ("..." if len(str(fb)) > 300 else "")
+                    out += f"\n- {ts} | {subj} | {topic}\n\n  {preview}\n"
+                return out
+
+            results_btn.click(fn=view_student_results, inputs=[student_name_input], outputs=results_output)
+
             # Follow-up handler: student asks a question about the feedback
             def send_followup(question_text):
                 if not session.last_feedback: