Update app.py

app.py

@@ -135,16 +135,21 @@ def main():
         # Parse the advanced analysis response
         analysis_data = parse_analysis(advanced_msg)
         
-        # Display parsed data in a collapsible section
-        with st.expander("Show Advanced Analysis"):
-            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)}")
+        if "error" in analysis_data:
+            st.error("Error in analysis: " + analysis_data["error"])
+            if "details" in analysis_data:
+                st.error("Details: " + analysis_data["details"])
+        else:
+            # Display parsed data in a collapsible section
+            with st.expander("Show Advanced Analysis"):
+                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"):