Update app.py

app.py

@@ -70,7 +70,6 @@ with gr.Blocks() as demo:
     # Section for generating Text-based output (lesson, question, feedback)
     with gr.Row():
         with gr.Column(scale=2):
-            # Input fields for subject, difficulty, and student input for textual output
             subject = gr.Dropdown(
                 ["Math", "Science", "History", "Literature", "Code", "AI"], 
                 label="Subject", 
@@ -89,7 +88,6 @@ with gr.Blocks() as demo:
             submit_button_text = gr.Button("Generate Lesson & Question", variant="primary")
         
         with gr.Column(scale=3):
-            # Output fields for lesson, question, and feedback
             lesson_output = gr.Markdown(label="Lesson")
             question_output = gr.Markdown(label="Comprehension Question")
             feedback_output = gr.Markdown(label="Feedback")
@@ -97,7 +95,6 @@ with gr.Blocks() as demo:
     # Section for generating Visual output
     with gr.Row():
         with gr.Column(scale=2):
-            # Input fields for text and model selection for image generation
             model_selector = gr.Radio(
                 ["Model 1 (Turbo Realism)", "Model 2 (Face Projection)"],
                 label="Select Image Generation Model",
@@ -106,7 +103,6 @@ with gr.Blocks() as demo:
             submit_button_visual = gr.Button("Generate Visuals", variant="primary")
         
         with gr.Column(scale=3):
-            # Output fields for generated images
             output1 = gr.Image(label="Generated Image 1")
             output2 = gr.Image(label="Generated Image 2")
             output3 = gr.Image(label="Generated Image 3")
@@ -128,19 +124,17 @@ with gr.Blocks() as demo:
     
     def process_output_visual(text, selected_model):
         try:
-            images = generate_images(text, selected_model)  # Generate images
+            images = generate_images(text, selected_model)
             return images[0], images[1], images[2]
         except:
             return None, None, None
     
-    # Generate Text-based Output
     submit_button_text.click(
         fn=process_output_text,
         inputs=[subject, difficulty, student_input],
         outputs=[lesson_output, question_output, feedback_output]
     )
     
-    # Generate Visual Output
     submit_button_visual.click(
         fn=process_output_visual,
         inputs=[student_input, model_selector],