Update app.py

app.py

@@ -104,19 +104,18 @@ def fetch_advanced_analysis(query, msg):
 
 def parse_analysis(analysis_message):
     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()
+        # Find the starting point of the JSON string
+        start = analysis_message.find('{')
+        if start != -1:
+            # Extract the JSON string
+            json_str = analysis_message[start:]
+            # Convert string to a Python dictionary
             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"}
-            st.write(json_str)
+            return {"error": "JSON data not found in the response"}
     except json.JSONDecodeError as e:
-        # Handle cases where JSON decoding fails
         return {"error": "Failed to decode JSON", "details": str(e)}
-
         
 # ---------------------------------------------------------------------------- #
 # Main Function
@@ -156,10 +155,6 @@ def main():
         advanced_response = fetch_advanced_analysis(user_query, msg)
         advanced_msg = advanced_response.get('message', 'No advanced analysis available.')
 
-        # Display the advanced analysis results in a collapsible section
-        with st.expander("Show Advanced Analysis"):
-            st.write(advanced_msg)
-        
         # Parse the advanced analysis response
         analysis_data = parse_analysis(advanced_msg)
         
@@ -170,7 +165,15 @@ def main():
         else:
             # Display parsed data in a collapsible section
             with st.expander("Show Advanced Analysis"):
-                st.write(advanced_msg)
+                st.write("### User Intent")
+                st.write(analysis_data['user_intent'])
+                st.write("### Follow-up Questions")
+                for question in analysis_data['follow_up_questions']:
+                    st.write("- " + question)
+                st.write("### Identified Entities")
+                for entity_type, entities in analysis_data['entities'].items():
+                    st.write(f"**{entity_type.capitalize()}**: {', '.join(entities)}")
+
 
         # Display the sentiment in a collapsible section
         with st.expander("Show Sentiment"):