Update app.py

app.py

@@ -67,6 +67,21 @@ def sentiment_analysis(text):
     df['confidence'] = df[['neg', 'neu', 'pos']].max(axis=1)
 
     return df
+
+# ---------------------------------------------------------------------------- #
+# Advanced Analysis
+# ---------------------------------------------------------------------------- #
+
+ def advanced_analysis(query, response_message):
+    analysis_prompt = f"Analyze the question of the user '{query}', analyze the response '{response_message}' and provide a JSON that includes: the user's intent, up to four follow-up questions, the entities in the response."
+    analysis_response = ai.prompt(message=analysis_prompt)
+    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     
         
 # ---------------------------------------------------------------------------- #
 # Main Function
@@ -102,6 +117,24 @@ def main():
         # Run sentiment analysis
         df_sentiment = sentiment_analysis(msg)
 
+        # Advanced analysis with second AI call
+        advanced_response = advanced_analysis(user_query, msg)
+        advanced_msg = advanced_response.get('message', 'No advanced analysis available.')
+        
+        # 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)}")
+        
         # Display the sentiment in a collapsible section
         with st.expander("Show Sentiment"):
             # Display negative sentence locations