Update app.py

app.py

@@ -34,7 +34,7 @@ def format_prompt(topic, description, difficulty):
 def clean_and_format_output(output):
     """
     Cleans, validates, and attempts to auto-repair JSON output from the AI model.
-    Handles truncation, malformed JSON, and missing delimiters.
+    Handles truncated, malformed JSON and missing delimiters.
     """
     try:
         # Step 1: Clean the raw output
@@ -45,12 +45,13 @@ def clean_and_format_output(output):
         cleaned_output = re.sub(r'\s+', ' ', cleaned_output).strip()  # Normalize whitespace
         cleaned_output = re.sub(r',\s*(\}|\])', r'\1', cleaned_output)  # Remove trailing commas
 
-        # Step 2: Check for truncation and attempt repair
+        # Step 2: Attempt to auto-repair JSON
         if not cleaned_output.endswith("]}"):
+            # Close the sections array and root object if needed
             if cleaned_output.endswith("]"):
-                cleaned_output += "}"  # Close the JSON object
+                cleaned_output += "}"  # Close the root object
             elif cleaned_output.endswith("}"):
-                cleaned_output = cleaned_output[:-1] + "]}"  # Close the sections array and JSON object
+                cleaned_output = cleaned_output[:-1] + "]}"  # Close the array and root object
             else:
                 cleaned_output += "]}"  # Append both array and object closures
 
@@ -72,12 +73,14 @@ def clean_and_format_output(output):
         return json_output
 
     except ValueError as e:
+        # Return detailed error information
         return {
             "error": "Failed to parse or validate output as JSON",
             "details": str(e),
             "output": cleaned_output
         }
 
+
 # Function to generate learning content
 def generate_learning_content(topic, description, difficulty, temperature=0.9, max_new_tokens=2000, top_p=0.95, repetition_penalty=1.2):
     """