Update app.py

app.py

@@ -74,10 +74,33 @@ def sentiment_analysis(text):
 # 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 fetch_advanced_analysis(query, msg):
+    analysis_prompt = f"""
+    Analyze the user's request: '{query}', and the response: '{msg}'. 
+    Based on this analysis, generate a detailed JSON response including:
+    1. The user's intent,
+    2. Up to four follow-up questions,
+    3. The main entities mentioned in the response.
+
+    Example of expected JSON format:
+    {{
+        "user_intent": "Identify the effects of climate change on polar bears",
+        "follow_up_questions": [
+            "What are the primary threats to polar bears today?",
+            "How does the melting ice affect their habitat?",
+            "What conservation efforts are in place for polar bears?",
+            "How can individuals contribute to these efforts?"
+        ],
+        "entities": {{
+            "animal": ["polar bears"],
+            "issue": ["climate change"],
+            "actions": ["conservation efforts"]
+        }}
+    }}
+    """
+    # Assume ai is an initialized MetaAI instance that can send prompts to the AI service
+    advanced_response = ai.prompt(message=analysis_prompt)
+    return advanced_response
 
 def parse_analysis(analysis_message):
     try:
@@ -129,9 +152,13 @@ def main():
         # Run sentiment analysis
         df_sentiment = sentiment_analysis(msg)
 
-        # Advanced analysis with second AI call
-        advanced_response = advanced_analysis(user_query, msg)
+        # Fetch advanced analysis
+        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)
@@ -143,15 +170,8 @@ def main():
         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)}")
-        
+                st.write(advanced_msg)
+
         # Display the sentiment in a collapsible section
         with st.expander("Show Sentiment"):
             # Display negative sentence locations