Update app.py

app.py

@@ -227,47 +227,131 @@ class CertificateGenerator:
         certificate.save(temp_file.name, 'PNG', quality=95)
         return temp_file.name
 
+    
 class QuizApp:
     def __init__(self, api_key: str):
         self.quiz_generator = QuizGenerator(api_key)
         self.certificate_generator = CertificateGenerator()
         self.current_questions: List[Question] = []
     
-    def generate_questions(self, text: str, num_questions: int) -> Tuple[bool, List[Question]]:
-        try:
-            self.current_questions = self.quiz_generator.generate_questions(text, num_questions)
-            return True, self.current_questions
-        except Exception as e:
-            print(f"Error generating questions: {e}")
-            return False, []
+    def update_questions(self, text: str, num_questions: int) -> Tuple[gr.update, gr.update, List[gr.update], List[Question], gr.update]:
+        """
+        Event handler for generating new questions
+        """
+        if not text.strip():
+            return (
+                gr.update(value=""),
+                gr.update(value="⚠️ Please enter some text content to generate questions."),
+                *[gr.update(visible=False, choices=[]) for _ in range(5)],
+                [],
+                gr.update(selected=1)
+            )
+        
+        success, questions = self.generate_questions(text, num_questions)
+        
+        if not success or not questions:
+            return (
+                gr.update(value=""),
+                gr.update(value="❌ Failed to generate questions. Please try again."),
+                *[gr.update(visible=False, choices=[]) for _ in range(5)],
+                [],
+                gr.update(selected=1)
+            )
+        
+        # Create question display
+        questions_html = "# 📝 Assessment Questions\n\n"
+        questions_html += "> Please select one answer for each question.\n\n"
+        
+        # Update radio buttons
+        updates = []
+        for i, q in enumerate(questions):
+            questions_html += f"### Question {i+1}\n{q.question}\n\n"
+            updates.append(gr.update(
+                visible=True,
+                choices=q.options,
+                value=None,
+                label=f"Select your answer:"
+            ))
+        
+        # Hide unused radio buttons
+        for i in range(len(questions), 5):
+            updates.append(gr.update(visible=False, choices=[]))
+        
+        return (
+            gr.update(value=questions_html),
+            gr.update(value=""),
+            *updates,
+            questions,
+            gr.update(selected=1)
+        )
     
-    def calculate_score(self, answers: List[Optional[str]]) -> Tuple[float, bool, List[QuizFeedback]]:
-        if not answers or not self.current_questions:
-            return 0, False, []
+    def submit_quiz(self, q1: Optional[str], q2: Optional[str], q3: Optional[str], 
+                   q4: Optional[str], q5: Optional[str], questions: List[Question]
+    ) -> Tuple[gr.update, List[gr.update], float, str, gr.update]:
+        """
+        Event handler for quiz submission
+        """
+        answers = [q1, q2, q3, q4, q5][:len(questions)]
         
-        feedback = []
-        correct = 0
+        if not all(a is not None for a in answers):
+            return (
+                gr.update(value="⚠️ Please answer all questions before submitting."),
+                *[gr.update() for _ in range(5)],
+                0,
+                "",
+                gr.update(selected=1)
+            )
         
-        for question, answer in zip(self.current_questions, answers):
-            if answer is None:
-                feedback.append(QuizFeedback(False, None, question.options[question.correct_answer]))
-                continue
-                
-            try:
-                selected_index = question.options.index(answer)
-                is_correct = selected_index == question.correct_answer
-                if is_correct:
-                    correct += 1
-                feedback.append(QuizFeedback(
-                    is_correct,
-                    answer,
-                    question.options[question.correct_answer]
-                ))
-            except ValueError:
-                feedback.append(QuizFeedback(False, answer, question.options[question.correct_answer]))
+        score, passed, feedback = self.calculate_score(answers)
+        
+        # Create feedback HTML
+        feedback_html = "# Assessment Results\n\n"
+        for i, (q, f) in enumerate(zip(self.current_questions, feedback)):
+            color = "green" if f.is_correct else "red"
+            symbol = "✅" if f.is_correct else "❌"
+            feedback_html += f"""
+            ### Question {i+1}
+            {q.question}
+            
+            <div style="color: {color}; padding: 10px; margin: 5px 0; border-left: 3px solid {color};">
+                {symbol} Your answer: {f.selected}
+                {'' if f.is_correct else f'<br>Correct answer: {f.correct_answer}'}
+            </div>
+            """
+        
+        # Add result message
+        if passed:
+            feedback_html += self._create_success_message(score)
+            result_msg = f"🎉 Congratulations! You passed with {score:.1f}%"
+        else:
+            feedback_html += self._create_failure_message(score)
+            result_msg = f"Score: {score:.1f}%. You need 80% to pass."
         
-        score = (correct / len(self.current_questions)) * 100
-        return score, score >= 80, feedback
+        return (
+            gr.update(value=feedback_html),
+            *[gr.update(visible=False) for _ in range(5)],
+            score,
+            result_msg,
+            gr.update(selected=2)
+        )
+    
+    def _create_success_message(self, score: float) -> str:
+        return f"""
+        <div style="background-color: #e6ffe6; padding: 20px; margin-top: 20px; border-radius: 10px;">
+            <h3 style="color: #008000;">🎉 Congratulations!</h3>
+            <p>You passed the assessment with a score of {score:.1f}%</p>
+            <p>Your certificate has been generated.</p>
+        </div>
+        """
+    
+    def _create_failure_message(self, score: float) -> str:
+        return f"""
+        <div style="background-color: #ffe6e6; padding: 20px; margin-top: 20px; border-radius: 10px;">
+            <h3 style="color: #cc0000;">Please Try Again</h3>
+            <p>Your score: {score:.1f}%</p>
+            <p>You need 80% or higher to pass and receive a certificate.</p>
+        </div>
+        """
 
 def create_quiz_interface():
     if not os.getenv("GROQ_API_KEY"):
@@ -276,6 +360,7 @@ def create_quiz_interface():
     quiz_app = QuizApp(os.getenv("GROQ_API_KEY"))
     
     with gr.Blocks(title="CertifyMe AI", theme=gr.themes.Soft()) as demo:
+
         # State management
         current_questions = gr.State([])
         
@@ -333,6 +418,7 @@ def create_quiz_interface():
                 certificate_display = gr.Image(label="Your Certificate")
         
         # Event handlers
+
         generate_btn.click(
             fn=quiz_app.update_questions,
             inputs=[text_input, num_questions],
@@ -367,4 +453,4 @@ def create_quiz_interface():
 
 if __name__ == "__main__":
     demo = create_quiz_interface()
-    demo.launch()
+    demo.launch()