Update app.py

app.py

@@ -31,10 +31,9 @@ def format_prompt(topic, description, difficulty):
     return prompt
 
 # Function to clean and format the AI output
-def clean_and_format_output(output):
+def clean_and_format_learning_content(output):
     """
-    Cleans, validates, and attempts to auto-repair JSON output from the AI model.
-    Handles truncated, malformed JSON and missing delimiters.
+    Cleans, validates, and parses JSON output for learning content.
     """
     try:
         # Step 1: Clean the raw output
@@ -43,27 +42,19 @@ def clean_and_format_output(output):
         cleaned_output = re.sub(r'^[^{]*', '', cleaned_output)  # Remove text before the first '{'
         cleaned_output = re.sub(r'[^}]*$', '', cleaned_output)  # Remove text after the last '}'
         cleaned_output = re.sub(r'\s+', ' ', cleaned_output).strip()  # Normalize whitespace
+        cleaned_output = cleaned_output.replace('\\"', '"')  # Fix improperly escaped quotes
         cleaned_output = re.sub(r',\s*(\}|\])', r'\1', cleaned_output)  # Remove trailing commas
 
-        # 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 root object
-            elif cleaned_output.endswith("}"):
-                cleaned_output = cleaned_output[:-1] + "]}"  # Close the array and root object
-            else:
-                cleaned_output += "]}"  # Append both array and object closures
-
-        # Step 3: Attempt to parse the cleaned output
-        try:
-            json_output = json.loads(cleaned_output)
-        except json.JSONDecodeError as e:
-            raise ValueError(f"JSON decoding error: {str(e)}")
-
-        # Step 4: Validate required structure
-        if "title" not in json_output or "sections" not in json_output:
-            raise ValueError("Missing required keys: 'title' or 'sections'.")
+        # Step 2: Attempt to parse the cleaned JSON
+        json_output = json.loads(cleaned_output)
+
+        # Step 3: Validate required keys
+        required_keys = ["title", "sections"]
+        for key in required_keys:
+            if key not in json_output:
+                raise ValueError(f"Missing key: {key}")
+
+        # Ensure "sections" is a list and has the required structure
         if not isinstance(json_output["sections"], list):
             raise ValueError("'sections' must be a list.")
         for section in json_output["sections"]:
@@ -72,8 +63,8 @@ def clean_and_format_output(output):
 
         return json_output
 
-    except ValueError as e:
-        # Return detailed error information
+    except (json.JSONDecodeError, ValueError) as e:
+        # Return error details if parsing or validation fails
         return {
             "error": "Failed to parse or validate output as JSON",
             "details": str(e),