Update app.py

app.py

@@ -5,33 +5,33 @@ import tempfile
 # Initialize the Hugging Face Inference Client
 client = InferenceClient()
 
-# Function to generate dynamic lessons, examples, or code
+# Function to generate content dynamically
 def generate_content(selected_topic, subtopic, complexity, input_text, examples_count, output_type):
     """
-    Generate content dynamically based on user input, with support for LaTeX rendering and file downloads.
+    Generate content dynamically based on user input with support for LaTeX and file downloads.
 
     Args:
-        selected_topic (str): Topic selected by the user.
-        subtopic (str): Specific subtopic for generation.
-        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 (Plain Text, LaTeX, or Downloadable).
+        selected_topic (str): The selected topic (e.g., Math, STEM, Code Generation).
+        subtopic (str): A specific subtopic for content generation.
+        complexity (str): Expertise level (Beginner, Intermediate, Advanced).
+        input_text (str): Additional context or problem to solve.
+        examples_count (int): Number of examples or outputs to generate.
+        output_type (str): Desired output format (Plain Text, LaTeX, Downloadable).
 
     Returns:
-        dict or str: Generated content in the desired format.
+        tuple: Generated content and file path (if applicable).
     """
-    # Build the prompt dynamically
+    # Create the prompt dynamically
     prompt = (
         f"Generate {examples_count} {complexity.lower()}-level {selected_topic.lower()} examples, lessons, "
         f"or problems related to {subtopic}. Context: {input_text}" if input_text.strip()
         else f"Generate {examples_count} {complexity.lower()}-level {selected_topic.lower()} lessons "
              f"or problems focused on {subtopic}."
     )
-    messages = [{"role": "user", "content": prompt}]
 
     try:
-        # Generate the content using the model
+        # Generate content using the model
+        messages = [{"role": "user", "content": prompt}]
         response = client.chat.completions.create(
             model="Qwen/Qwen2.5-Coder-32B-Instruct",
             messages=messages,
@@ -39,27 +39,32 @@ def generate_content(selected_topic, subtopic, complexity, input_text, examples_
             max_tokens=1024,
             top_p=0.7
         )
-        content = response.choices[0].message.content
+        # Extract content from the response
+        content = response.choices[0].message.content if response.choices else "No content generated."
 
-        # Adjust the output based on the selected type
+        # Handle output formatting
         if output_type == "LaTeX":
-            # Wrap content in LaTeX delimiters for rendering
-            latex_content = f"$$\n{content}\n$$"
-            return {"content": latex_content, "latex": True}
+            # Ensure LaTeX content is properly wrapped
+            latex_content = f"$$\n{content.strip()}\n$$"
+            return latex_content, None
         elif output_type == "Downloadable":
+            # Save content to a temporary file
             temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".txt")
             with open(temp_file.name, "w") as file:
                 file.write(content)
-            return {"file": temp_file.name}
+            return "File generated successfully. Use the download button.", temp_file.name
         else:
-            return content
+            # Default to plain text
+            return content, None
     except Exception as e:
-        return f"Error: {e}"
+        # Catch and return any errors
+        return f"Error during content generation: {e}", None
+
 
 # Create the Gradio interface
 with gr.Blocks() as app:
-    # App Title and Instructions
-    gr.Markdown("## 🌟 Advanced STEM and Code Generator with LaTeX Rendering and Downloads")
+    # App Title and Description
+    gr.Markdown("## 🌟 Advanced STEM and Code Generator with LaTeX and File Downloads")
 
     with gr.Row():
         # Input Section
@@ -104,14 +109,10 @@ with gr.Blocks() as app:
             download_button = gr.File(label="Download File (if applicable)")
 
     # 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
+    def update_output(result, file_path):
+        if file_path:
+            return result, file_path
+        return result, None
 
     generate_button.click(
         fn=generate_content,