Update app.py

app.py

@@ -168,7 +168,9 @@ def generate_test(total_marks, topic_name, level):
         - **Topic Code**
         - **Maximum Marks**
     7. Display the full question content exactly as it appears in 'QP Content'.
-    8. Use the provided **TOPICS LIST** to identify relevant topic codes.
+    8. **IMPORTANT**: Keep all LaTeX mathematical expressions in their original LaTeX format using $ for inline math (e.g., $x^2$) and $$ for display math (e.g., $$\\frac{{a}}{{b}}$$). Do NOT convert LaTeX to plain text.
+    9. Use proper markdown formatting with headers (##, ###), bullet points, and numbered lists for better readability.
+    10. Use the provided **TOPICS LIST** to identify relevant topic codes.
 
     **TOPICS LIST:**
     ---
@@ -181,6 +183,7 @@ def generate_test(total_marks, topic_name, level):
     ---
 
     Now, create the test based on the user's request, making sure to use the **latest available questions** first.
+    Ensure all mathematical notation remains in LaTeX format for proper rendering.
     """
 
     try:
@@ -189,6 +192,20 @@ def generate_test(total_marks, topic_name, level):
     except Exception as e:
         return f"An error occurred while communicating with the AI model: {e}"
 
+# --- Wrapper function for processing state ---
+def generate_test_with_processing(total_marks, topic_name, level):
+    """
+    Wrapper function that shows processing message before generating test.
+    """
+    # This will be shown while processing
+    yield "## 🔄 Processing...\n\nGenerating your test paper. Please wait..."
+    
+    # Generate the actual test
+    result = generate_test(total_marks, topic_name, level)
+    
+    # Return the final result
+    yield result
+
 # --- Gradio Interface ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 📝 IB Mathematics Test Generator (Latest Question Priority)")
@@ -212,10 +229,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             generate_button = gr.Button("Generate Test", variant="primary")
 
         with gr.Column(scale=3):
-            output_markdown = gr.Markdown(label="Generated Test Paper")
+            output_markdown = gr.Markdown(
+                label="Generated Test Paper",
+                latex_delimiters=[
+                    {"left": "$$", "right": "$$", "display": True},
+                    {"left": "$", "right": "$", "display": False}
+                ]
+            )
 
     generate_button.click(
-        fn=generate_test,
+        fn=generate_test_with_processing,
         inputs=[total_marks_input, topic_input, level_input],
         outputs=output_markdown
     )