Update app.py

app.py

@@ -6,6 +6,7 @@ import json
 import os  # For API key usage
 from pathlib import Path
 import time
+import re
 import plotly.express as px
 import pandas as pd
 
@@ -275,54 +276,33 @@ def call_gemini_api(prompt):
             api_status.success(f"✅ Response received from AI ({model.model_name}) in {end_time - start_time:.2f}s.")
             time.sleep(1)
             api_status.empty()
+            
+            # Improved JSON extraction using regex
+            json_response_text = response.text.strip()
+            json_response_text = json_response_text.replace("```json", "").replace("```", "")
+            match = re.search(r'({.*})', json_response_text, re.DOTALL)
+            if match:
+                final_json_text = match.group(1)
+                insights = json.loads(final_json_text)
+                return insights, None
+            else:
+                st.warning("⚠️ Could not find valid JSON object.")
+                return {"raw_response": response.text}, "AI response did not contain clear JSON object."
+        except json.JSONDecodeError as json_err:
+            st.error(f"🚨 Error parsing JSON: {json_err}")
+            st.code(response.text, language='text')
+            return None, f"AI response not valid JSON: {json_err}"
+        except AttributeError:
+            st.error("🚨 Unexpected API response structure (AttributeError).")
+            st.code(f"Response object: {response}", language='text')
+            return None, "Unexpected response structure (AttributeError)."
+        except Exception as e:
+            st.error(f"🚨 Unexpected issue processing response: {e}")
             try:
-                json_response_text = response.text.strip()
-                if json_response_text.startswith("```json"):
-                    json_response_text = json_response_text[7:]
-                if json_response_text.startswith("```"):
-                    json_response_text = json_response_text[3:]
-                if json_response_text.endswith("```"):
-                    json_response_text = json_response_text[:-3]
-                json_start = json_response_text.find('{')
-                json_end = json_response_text.rfind('}') + 1
-                if json_start != -1 and json_end != -1 and json_end > json_start:
-                    final_json_text = json_response_text[json_start:json_end]
-                    insights = json.loads(final_json_text)
-                    return insights, None
-                else:
-                    st.warning("⚠️ Could not find valid JSON object.")
-                    return {"raw_response": response.text}, "AI response did not contain clear JSON object."
-            except json.JSONDecodeError as json_err:
-                st.error(f"🚨 Error parsing JSON: {json_err}")
-                st.code(response.text, language='text')
-                return None, f"AI response not valid JSON: {json_err}"
-            except AttributeError:
-                st.error("🚨 Unexpected API response structure (AttributeError).")
                 st.code(f"Response object: {response}", language='text')
-                return None, "Unexpected response structure (AttributeError)."
-            except Exception as e:
-                st.error(f"🚨 Unexpected issue processing response: {e}")
-                try:
-                    st.code(f"Response object: {response}", language='text')
-                except Exception:
-                    pass
-                return None, f"Unexpected response structure: {e}"
-        except Exception as e:
-            api_status.empty()
-            st.error(f"🚨 API call error: {e}")
-            error_msg = f"API call failed: {e}"
-            if hasattr(e, 'message'):
-                if "429" in e.message:
-                    error_msg = "API Quota Exceeded or Rate Limit hit."
-                elif "API key not valid" in e.message:
-                    error_msg = "Invalid Gemini API Key."
-                elif "permission denied" in e.message.lower():
-                    error_msg = f"Permission Denied for model '{st.session_state.selected_model_name}'. Check API key access."
-                elif "blocked" in e.message.lower():
-                    error_msg = "Content blocked due to safety settings."
-            elif "block_reason: SAFETY" in str(e):
-                error_msg = "Content blocked due to safety settings."
-            return None, error_msg
+            except Exception:
+                pass
+            return None, f"Unexpected response structure: {e}"
 
 def display_results(results_json, requested_analyses):
     """