Update app.py

app.py

@@ -46,17 +46,19 @@ 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: Fix Markdown-style links (if unintended)
-        cleaned_output = re.sub(r'\[([^\]]+)\]\([^\)]+\)', r'\1', cleaned_output)
-
-        # Step 3: Fix extra closing braces or brackets
-        cleaned_output = re.sub(r'\]\]', ']', cleaned_output)  # Remove extra closing brackets
-        cleaned_output = re.sub(r'\}\}', '}', cleaned_output)  # Remove extra closing braces
-
-        # Step 4: Fix missing commas between JSON objects
+        # Step 2: Fix the 'sections' structure
+        # Ensure 'sections' is an array, not a single object
+        sections_pattern = r'"sections":\s*{'
+        if re.search(sections_pattern, cleaned_output):
+            # Replace the invalid opening brace with an array opening
+            cleaned_output = re.sub(sections_pattern, '"sections": [ {', cleaned_output)
+            # Add a closing array bracket at the end of 'sections'
+            cleaned_output = re.sub(r'(}\s*,?\s*)]}', r'} ] }', cleaned_output)
+
+        # Step 3: Fix missing commas between JSON objects
         cleaned_output = re.sub(r'(\})(\s*{)', r'\1,\2', cleaned_output)
 
-        # Step 5: Balance brackets and braces
+        # Step 4: Fix unbalanced brackets and braces
         open_braces = cleaned_output.count('{')
         close_braces = cleaned_output.count('}')
         open_brackets = cleaned_output.count('[')
@@ -66,10 +68,10 @@ def clean_and_format_learning_content(output):
         if open_brackets > close_brackets:
             cleaned_output += ']' * (open_brackets - close_brackets)
 
-        # Step 6: Attempt to parse JSON
+        # Step 5: Attempt to parse JSON
         json_output = json.loads(cleaned_output)
 
-        # Step 7: Validate JSON structure
+        # Step 6: Validate JSON 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'.")
@@ -89,6 +91,7 @@ def clean_and_format_learning_content(output):
             "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):
     """