Update app.py

app.py

@@ -33,7 +33,8 @@ def format_prompt(topic, description, difficulty):
 # Function to clean and format the AI output
 def clean_and_format_learning_content(output):
     """
-    Cleans, validates, and parses JSON output for learning content.
+    Cleans, validates, and attempts to auto-repair JSON output for learning content.
+    Handles truncated and malformed JSON gracefully.
     """
     try:
         # Step 1: Clean the raw output
@@ -45,16 +46,23 @@ 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: Attempt to parse the cleaned JSON
+        # Step 2: Detect truncation and auto-repair if necessary
+        if not cleaned_output.endswith("]}"):
+            # Check for open array or object and close them
+            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
+
+        # Step 3: Attempt to parse the cleaned JSON
         json_output = json.loads(cleaned_output)
 
-        # Step 3: Validate required keys
+        # Step 4: Validate the structure
         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 "title" not in json_output or "sections" not in json_output:
+            raise ValueError("Missing required keys: 'title' or 'sections'.")
         if not isinstance(json_output["sections"], list):
             raise ValueError("'sections' must be a list.")
         for section in json_output["sections"]:
@@ -64,14 +72,13 @@ def clean_and_format_learning_content(output):
         return json_output
 
     except (json.JSONDecodeError, ValueError) as e:
-        # Return error details if parsing or validation fails
+        # Return error details and cleaned output 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):
     """