Update app.py

app.py

@@ -31,12 +31,9 @@ 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 repairs JSON output for learning content.
+    Cleans, validates, and attempts to repair JSON output for learning content.
     """
     try:
         # Step 1: Clean the raw output
@@ -48,24 +45,26 @@ def clean_and_format_learning_content(output):
         cleaned_output = cleaned_output.replace('\\"', '"')  # Fix improperly escaped quotes
         cleaned_output = re.sub(r',\s*(\}|\])', r'\1', cleaned_output)  # Remove trailing commas
 
-        # Step 2: Detect truncation and auto-repair
-        if not cleaned_output.endswith("]}"):
-            # Count unmatched braces and brackets
+        # Step 2: Attempt to parse and repair malformed JSON
+        try:
+            json_output = json.loads(cleaned_output)  # First attempt to parse as-is
+        except json.JSONDecodeError:
+            # Attempt repair by handling common issues
+            # Fix missing commas between sections
+            cleaned_output = re.sub(r'(\})(\s*{)', r'\1,\2', cleaned_output)
+            # Fix missing closing brackets or braces
             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)
+            # Retry parsing
+            json_output = json.loads(cleaned_output)
 
-        # Step 3: Attempt to parse the cleaned JSON
-        json_output = json.loads(cleaned_output)
-
-        # Step 4: Validate the required structure
+        # Step 3: Validate the required structure
         required_keys = ["title", "sections"]
         if "title" not in json_output or "sections" not in json_output:
             raise ValueError("Missing required keys: 'title' or 'sections'.")
@@ -78,7 +77,7 @@ def clean_and_format_learning_content(output):
         return json_output
 
     except (json.JSONDecodeError, ValueError) as e:
-        # Return detailed error information for debugging
+        # Return detailed error information
         return {
             "error": "Failed to parse or validate output as JSON",
             "details": str(e),