Update app.py

app.py

@@ -8,7 +8,7 @@ client = InferenceClient()
 # Function to generate dynamic lessons, examples, or code
 def generate_content(selected_topic, subtopic, complexity, input_text, examples_count, output_type):
     """
-    Generate content dynamically based on user input.
+    Generate content dynamically based on user input, with support for LaTeX rendering and file downloads.
 
     Args:
         selected_topic (str): Topic selected by the user.
@@ -16,10 +16,10 @@ def generate_content(selected_topic, subtopic, complexity, input_text, examples_
         complexity (str): User expertise level.
         input_text (str): Additional input context.
         examples_count (int): Number of examples to generate.
-        output_type (str): Desired output format.
+        output_type (str): Desired output format (Plain Text, LaTeX, or Downloadable).
 
     Returns:
-        str or dict: Generated content in the selected format.
+        dict or str: Generated content in the desired format.
     """
     # Build the prompt dynamically
     prompt = (
@@ -43,7 +43,9 @@ def generate_content(selected_topic, subtopic, complexity, input_text, examples_
 
         # Adjust the output based on the selected type
         if output_type == "LaTeX":
-            return {"content": content, "latex": True}
+            # Wrap content in LaTeX delimiters for rendering
+            latex_content = f"$$\n{content}\n$$"
+            return {"content": latex_content, "latex": True}
         elif output_type == "Downloadable":
             temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt")
             with open(temp_file.name, "w") as file:
@@ -56,11 +58,11 @@ def generate_content(selected_topic, subtopic, complexity, input_text, examples_
 
 # Create the Gradio interface
 with gr.Blocks() as app:
-    # App Title
-    gr.Markdown("## 🌟 Advanced STEM and Code Generator with Interactive Features")
+    # App Title and Instructions
+    gr.Markdown("## 🌟 Advanced STEM and Code Generator with LaTeX Rendering and Downloads")
 
     with gr.Row():
-        # Input Panel
+        # Input Section
         with gr.Column():
             selected_topic = gr.Radio(
                 choices=["Math", "STEM", "Code Generation"],
@@ -95,24 +97,31 @@ with gr.Blocks() as app:
             )
             generate_button = gr.Button("Generate Content")
 
-        # Output Panel
+        # Output Section
         with gr.Column():
-            gr.Markdown("### 📝 Output")
-            output = gr.Textbox(
-                label="Generated Output",
-                lines=15,
-                interactive=False
-            )
+            gr.Markdown("### 📝 Generated Output (Supports LaTeX)")
+            output_text = gr.Markdown(label="Generated Content")
             download_button = gr.File(label="Download File (if applicable)")
 
-    # Link the generation function to the button
+    # Connect the generate function to the button
+    def update_output(result):
+        if isinstance(result, dict):
+            if "latex" in result:
+                return result["content"], None
+            elif "file" in result:
+                return "File ready for download.", result["file"]
+        else:
+            return result, None
+
     generate_button.click(
         fn=generate_content,
         inputs=[selected_topic, subtopic, complexity, input_text, examples_count, output_type],
-        outputs=[output, download_button]
+        outputs=[output_text, download_button],
+        preprocess=False,
+        postprocess=update_output
     )
 
-    # Feedback Mechanism
+    # Feedback Section
     feedback_label = gr.Label(value="Was this content helpful?")
     feedback_rating = gr.Radio(
         choices=["Yes", "No"],