Update app.py

app.py

@@ -31,10 +31,12 @@ def format_prompt(topic, description, difficulty):
     return prompt
 
 # Function to clean and format the AI output
+import json
+import re
+
 def clean_and_format_learning_content(output):
     """
-    Cleans, validates, and attempts to auto-repair JSON output for learning content.
-    Handles truncated and malformed JSON gracefully.
+    Cleans, validates, and repairs JSON output for learning content.
     """
     try:
         # Step 1: Clean the raw output
@@ -48,12 +50,17 @@ def clean_and_format_learning_content(output):
 
         # Step 2: Detect truncation and auto-repair
         if not cleaned_output.endswith("]}"):
-            if cleaned_output.endswith("]"):
-                cleaned_output += "}"  # Close the root object
-            elif cleaned_output.endswith("}"):
-                cleaned_output = cleaned_output[:-1] + "]}"  # Close the sections array and root object
-            else:
-                cleaned_output += "]}"  # Append both array and object closures
+            # Count unmatched braces and brackets
+            open_braces = cleaned_output.count('{')
+            close_braces = cleaned_output.count('}')
+            open_brackets = cleaned_output.count('[')
+            close_brackets = cleaned_output.count(']')
+
+            # Add missing closing braces or brackets
+            if open_braces > close_braces:
+                cleaned_output += '}' * (open_braces - close_braces)
+            if open_brackets > close_brackets:
+                cleaned_output += ']' * (open_brackets - close_brackets)
 
         # Step 3: Attempt to parse the cleaned JSON
         json_output = json.loads(cleaned_output)
@@ -71,7 +78,7 @@ def clean_and_format_learning_content(output):
         return json_output
 
     except (json.JSONDecodeError, ValueError) as e:
-        # Return detailed error information
+        # Return detailed error information for debugging
         return {
             "error": "Failed to parse or validate output as JSON",
             "details": str(e),