Update app.py

app.py

@@ -46,7 +46,10 @@ 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: Handle common issues systematically
+        # Step 2: Fix common structural issues
+        # Remove extra closing brackets in arrays
+        cleaned_output = re.sub(r'\]\]', ']', cleaned_output)
+
         # Fix missing commas between objects in arrays
         cleaned_output = re.sub(r'(\})(\s*{)', r'\1,\2', cleaned_output)
 
@@ -60,23 +63,10 @@ def clean_and_format_learning_content(output):
         if open_brackets > close_brackets:
             cleaned_output += ']' * (open_brackets - close_brackets)
 
-        # Ensure "sections" array is closed properly
-        if 'sections": [' in cleaned_output and not cleaned_output.strip().endswith(']}'):
-            cleaned_output = cleaned_output.rstrip('}') + ']}'
-
-        # Handle misplaced or extra braces/brackets
-        cleaned_output = re.sub(r'}\s*}\s*]', '}]', cleaned_output)  # Extra closing brace in arrays
-
-        # Step 3: Attempt to parse and repair malformed JSON
-        try:
-            json_output = json.loads(cleaned_output)  # Attempt to parse as-is
-        except json.JSONDecodeError as e:
-            # Debugging logs for detailed errors (can be commented out in production)
-            print(f"JSON Decode Error at char {e.pos}: {e.msg}")
-            print("Cleaned Output:", cleaned_output)
-            raise e  # Re-raise the exception after logging
+        # Step 3: Attempt to parse the cleaned output as JSON
+        json_output = json.loads(cleaned_output)  # Parse JSON
 
-        # Step 4: Validate the required structure
+        # Step 4: Validate 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'.")
@@ -96,6 +86,14 @@ def clean_and_format_learning_content(output):
             "output": cleaned_output
         }
 
+    except (json.JSONDecodeError, ValueError) as e:
+        # Provide detailed error information for debugging
+        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):