Update app.py

app.py

@@ -257,10 +257,10 @@ def fix_json(text):
     Attempt to fix common JSON formatting issues.
     This function removes trailing commas and inserts commas between adjacent objects.
     """
-    # Remove trailing commas before closing braces
-    text = re.sub(r',\s*}', "}", text)
-    # Insert missing commas between adjacent objects in an array if needed
-    text = re.sub(r'}\s*{', "},{", text)
+    # Remove trailing commas before a closing brace or bracket.
+    text = re.sub(r',\s*([}\]])', r'\1', text)
+    # Insert commas between adjacent objects if missing.
+    text = re.sub(r'}\s*{', '},{', text)
     return text
 
 def call_gemini_api(prompt):
@@ -306,14 +306,14 @@ def call_gemini_api(prompt):
             time.sleep(1)
             api_status.empty()
             
-            # Remove markdown formatting if present
+            # Remove markdown formatting if present.
             json_response_text = response.text.strip().replace("```json", "").replace("```", "")
-            # First try regex extraction
+            # First, attempt regex extraction.
             match = re.search(r'({.*})', json_response_text, re.DOTALL)
             if match:
                 final_json_text = match.group(1)
             else:
-                # Fallback: extract using balanced braces method
+                # Fallback: extract using balanced braces.
                 final_json_text = extract_json_from_text(json_response_text)
             if not final_json_text:
                 st.warning("⚠️ Could not extract a valid JSON object.")
@@ -322,7 +322,7 @@ def call_gemini_api(prompt):
                 insights = json.loads(final_json_text)
                 return insights, None
             except json.JSONDecodeError as json_err:
-                # Attempt to fix the JSON string and re-parse
+                # Attempt to fix the JSON string.
                 fixed_text = fix_json(final_json_text)
                 try:
                     insights = json.loads(fixed_text)