Update app.py

app.py

@@ -525,7 +525,15 @@ class QuizApp:
         self.quiz_generator = QuizGenerator(api_key)
         self.certificate_generator = CertificateGenerator()
         self.current_questions: List[Question] = []
-        self.logo_path = "atgc_logo.png"  # Fixed ATGC logo path
+        self.logo_path = "atgc_logo.png"
+        self.selected_level = "Basic"  # Default level
+        
+        # Map difficulty levels to number of questions
+        self.difficulty_levels = {
+            "Basic": 5,
+            "Intermediate": 10,
+            "Advanced": 20
+        }
         
         # Add fixed content here
         self.fixed_content = """ATGC Transport and General Contracting LLC's Human Resources Policy and Procedures Manual serves as the definitive guide for personnel management within the organization. This comprehensive document defines policies and clarifies responsibilities to ensure consistent application of HR practices across all levels of the company. The manual applies to both permanent and temporary employees and has received approval from the Managing Director and Board of Directors. As a confidential document, unauthorized disclosure is strictly prohibited. The manual undergoes regular revisions to accommodate changes in company needs and UAE labor law, maintaining its relevance and flexibility.
@@ -568,6 +576,9 @@ Training and development initiatives require department head initiation, HR revi
 Implementation and Compliance
 The manual emphasizes the importance of proper implementation and compliance with all policies. Regular reviews and updates ensure continued relevance and effectiveness of policies. The HR Department maintains responsibility for policy interpretation and implementation, with ultimate authority resting with the Managing Director and Board of Directors."""
 
+    def get_certificate_title(self, base_title: str) -> str:
+        """Get certificate title with difficulty level"""
+        return f"{base_title} - {self.selected_level} Level"    
 
     def generate_questions(self, text: str, num_questions: int) -> Tuple[bool, List[Question]]:
         """
@@ -765,14 +776,11 @@ def create_quiz_interface():
                     lines=10
                 )
                 
-                
-                num_questions = gr.Slider(
-                    minimum=1,
-                    maximum=20,
-                    value=10,
-                    step=1,
-                    label="Number of Questions"
-                )
+                difficulty_level = gr.Radio(
+                    choices=["Basic", "Intermediate", "Advanced"],
+                    value="Basic",
+                    label="Select Difficulty Level",
+                    info="Basic: 5 questions | Intermediate: 10 questions | Advanced: 20 questions"
                 
                 # Add clear button
                 clear_btn = gr.Button("Clear Learning Content", variant="secondary", size="sm")
@@ -833,14 +841,17 @@ def create_quiz_interface():
                             )                    
 
             # Certification Tab (Hidden by default)
+
             with gr.Tab(id=3, label="🎓 Step 3: Get Certified", visible=False) as cert_tab:
                 score_display = gr.Number(label="Your Score", visible=False)
                 course_name = gr.Textbox(
                     label="Certification Title",
-                    value="Professional Assessment Certification",
-                    interactive=False  # Make it non-editable
+                    value=lambda: f"Professional Assessment Certification - {quiz_app.selected_level} Level",  # Dynamic title with level
+                    interactive=False
                 )
-                certificate_display = gr.Image(label="Your Certificate")
+                certificate_display = gr.Image(label="Your Certificate")        
+                    
+        
         def clear_content():
             quiz_app.clear_learning_content()
             return gr.update(value="")
@@ -853,12 +864,18 @@ def create_quiz_interface():
         def show_certificate_tab():
             return [
                 gr.update(visible=True),  # Make cert_tab visible
+                gr.update(
+                    value=f"Professional Assessment Certification - {quiz_app.selected_level} Level"  # Update title with level
+                ),
                 gr.update(selected=3)     # Switch to cert_tab
             ]
 
 
         # Helper Functions
-        def on_generate_questions(text, num_questions):
+        def on_generate_questions(text, level):
+            quiz_app.selected_level = level  # Store selected level
+            num_questions = quiz_app.difficulty_levels[level]  # Get number of questions for level
+            
             success, questions = quiz_app.generate_questions(quiz_app.fixed_content, num_questions)
             if not success or not questions:
                 return [
@@ -868,7 +885,7 @@ def create_quiz_interface():
                     "",
                     [],
                     0,
-                    [None] * 5,
+                    [None] * num_questions,  # Updated to use dynamic size
                     gr.update(selected=1),
                     gr.update(visible=False),
                     gr.update(visible=False)
@@ -895,7 +912,6 @@ def create_quiz_interface():
                 gr.update(visible=False)
             ]
 
-
         def navigate(direction, current_idx, questions, answers, current_answer):
             if not questions:
                 return [0, answers, "", gr.update(choices=[], visible=False), "", gr.update(visible=False)]
@@ -1013,7 +1029,7 @@ def create_quiz_interface():
         # Event Handlers
         generate_btn.click(
             fn=on_generate_questions,
-            inputs=[text_input, num_questions],
+            inputs=[text_input, difficulty_level],  # Replace num_questions with difficulty_level
             outputs=[
                 question_display,
                 question_box,
@@ -1057,7 +1073,7 @@ def create_quiz_interface():
         
         reset_btn.click(
             fn=on_generate_questions,
-            inputs=[text_input, num_questions],
+            inputs=[text_input, difficulty_level],  # Replace num_questions with difficulty_level
             outputs=[
                 question_display,
                 question_box,
@@ -1071,10 +1087,10 @@ def create_quiz_interface():
                 view_cert_btn
             ]
         )
-        
+
         view_cert_btn.click(
             fn=show_certificate_tab,
-            outputs=[cert_tab, tabs]
+            outputs=[cert_tab, course_name, tabs]  # Add course_name to outputs
         )
         
         current_options.change(
@@ -1084,8 +1100,12 @@ def create_quiz_interface():
         )
         
         score_display.change(
-            fn=lambda s, n, c, p: quiz_app.certificate_generator.generate(s, n, c, quiz_app.logo_path, p) or gr.update(value=None),
-            inputs=[score_display, name, course_name, participant_photo],  # company_logo removed
+            fn=lambda s, n, c, p: quiz_app.certificate_generator.generate(
+                s, n, 
+                quiz_app.get_certificate_title(c),  # Add difficulty level to title
+                quiz_app.logo_path, p
+            ) or gr.update(value=None),
+            inputs=[score_display, name, course_name, participant_photo],
             outputs=certificate_display
         )