Update app.py

app.py

@@ -252,6 +252,17 @@ def extract_json_from_text(text):
                 return text[start:i+1]
     return None
 
+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)
+    return text
+
 def call_gemini_api(prompt):
     """
     Calls the Gemini API using the provided prompt.
@@ -302,14 +313,24 @@ def call_gemini_api(prompt):
             if match:
                 final_json_text = match.group(1)
             else:
-                # Fallback: extract using our balanced-braces function
+                # Fallback: extract using balanced braces method
                 final_json_text = extract_json_from_text(json_response_text)
-            if final_json_text:
-                insights = json.loads(final_json_text)
-                return insights, None
-            else:
+            if not final_json_text:
                 st.warning("⚠️ Could not extract a valid JSON object.")
                 return {"raw_response": response.text}, "AI response did not contain clear JSON object."
+            try:
+                insights = json.loads(final_json_text)
+                return insights, None
+            except json.JSONDecodeError as json_err:
+                # Attempt to fix the JSON string and re-parse
+                fixed_text = fix_json(final_json_text)
+                try:
+                    insights = json.loads(fixed_text)
+                    return insights, None
+                except json.JSONDecodeError as json_err2:
+                    st.error(f"🚨 Error parsing JSON after fix attempt: {json_err2}")
+                    st.code(response.text, language='text')
+                    return None, f"AI response not valid JSON: {json_err2}"
         except json.JSONDecodeError as json_err:
             st.error(f"🚨 Error parsing JSON: {json_err}")
             st.code(response.text, language='text')