Update app.py

app.py

@@ -5,6 +5,8 @@ import pandas as pd
 import plotly.express as px
 from nltk.sentiment.vader import SentimentIntensityAnalyzer
 import nltk
+import json
+
 
 # Initialize Meta AI API
 ai = MetaAI()
@@ -78,10 +80,19 @@ def advanced_analysis(query, response_message):
     return analysis_response   
 
 def parse_analysis(analysis_message):
-    # Assuming the response message contains a JSON string starting with 'Here is the JSON:'
-    json_str = analysis_message.split('Here is the JSON:\n')[1].strip()
-    analysis_data = json.loads(json_str)
-    return analysis_data     
+    try:
+        # Check if the expected delimiter 'Here is the JSON:' is present
+        if 'JSON:\n' in analysis_message:
+            json_str = analysis_message.split('JSON:\n')[1].strip()
+            analysis_data = json.loads(json_str)
+            return analysis_data
+        else:
+            # If the delimiter is not found, return a default error message or handle it
+            return {"error": "JSON delimiter not found in response"}
+    except json.JSONDecodeError as e:
+        # Handle cases where JSON decoding fails
+        return {"error": "Failed to decode JSON", "details": str(e)}
+
         
 # ---------------------------------------------------------------------------- #
 # Main Function