Update app.py

app.py

@@ -525,7 +525,18 @@ class QuizApp:
         self.quiz_generator = QuizGenerator(api_key)
         self.certificate_generator = CertificateGenerator()
         self.current_questions: List[Question] = []
+        self.learning_content: Optional[str] = None  # Store learning content
+        self.logo_path = "atgc_logo.png"  # Fixed path to ATGC logo
 
+    def set_learning_content(self, text: str) -> None:
+        """Set the learning content if not already set"""
+        if not self.learning_content:
+            self.learning_content = text.strip()    
+
+    def clear_learning_content(self) -> None:
+        """Clear the stored learning content"""
+        self.learning_content = None            
+    
     def generate_questions(self, text: str, num_questions: int) -> Tuple[bool, List[Question]]:
         """
         Generate quiz questions using the QuizGenerator
@@ -728,13 +739,19 @@ def create_quiz_interface():
                     step=1,
                     label="Number of Questions"
                 )
-                
+                clear_btn = gr.Button("Clear Learning Content", variant="secondary", size="sm")
                 with gr.Row():
-                    company_logo = gr.Image(label="Company Logo (Optional)", type="filepath")
                     participant_photo = gr.Image(label="Your Photo (Optional)", type="filepath")
                 
                 generate_btn = gr.Button("Generate Assessment", variant="primary", size="lg")
+        def clear_content():
+            quiz_app.clear_learning_content()
+            return gr.update(value="")
             
+        clear_btn.click(
+            fn=clear_content,
+            outputs=[text_input]
+        )            
             # Assessment Tab
             with gr.Tab(id=2, label="📝 Step 2: Take Assessment"):
                 with gr.Column() as main_container:
@@ -819,7 +836,14 @@ def create_quiz_interface():
                     gr.update(visible=False)
                 ]
             
-            success, questions = quiz_app.generate_questions(text, num_questions)
+            # Store content if not already stored
+            if not quiz_app.learning_content:
+                quiz_app.learning_content = text.strip()
+            
+            # Use stored content instead of new text if available
+            content_to_use = quiz_app.learning_content or text
+            
+            success, questions = quiz_app.generate_questions(content_to_use, num_questions)
             if not success or not questions:
                 return [
                     "",
@@ -836,7 +860,7 @@ def create_quiz_interface():
             
             question = questions[0]
             question_md = f"""### Question 1
-{question.question}"""
+        {question.question}"""
             
             return [
                 question_md,
@@ -972,7 +996,7 @@ def create_quiz_interface():
             
         # Event Handlers
         generate_btn.click(
-            fn=on_generate_questions,
+            fn=on_generate_questions,  # Using the same function name
             inputs=[text_input, num_questions],
             outputs=[
                 question_display,