Update app.py

app.py

@@ -42,8 +42,14 @@ def command_handler(user_input):
     return None
 
 # Function to get the response from OpenAI with professionalism and energy
-def get_groq_response(message, user_language):
+def get_groq_response(message, user_language, custom_data=None):
     try:
+        # If custom data is available, include it in the AI prompt
+        if custom_data:
+            prompt = f"Use the following information for analysis: {custom_data}. Then answer the user's query: {message}"
+        else:
+            prompt = message
+
         response = openai.ChatCompletion.create(
             model="llama-3.1-70b-versatile",
             messages=[
@@ -61,7 +67,7 @@ def get_groq_response(message, user_language):
                         f"Your primary goal is to assist users by providing accurate, personalized, and actionable insights related to private markets. Always maintain professionalism and conciseness. Ensure that responses are easy to interpret, and avoid jargon whenever possible.\n\n"
                     )
                 },
-                {"role": "user", "content": message}
+                {"role": "user", "content": prompt}
             ]
         )
         return response.choices[0].message["content"]
@@ -73,14 +79,29 @@ def format_response(data):
     if not data:
         return "No data available for this query."
 
+    formatted_response = ""
+
     if isinstance(data, dict):
-        formatted_response = "### Insights:\n\n"
+        formatted_response += "### Key Insights:\n\n"
         for key, value in data.items():
-            formatted_response += f"**{key.capitalize()}**: {value}\n\n"
+            if isinstance(value, dict):
+                # For nested dictionaries, add a heading for the sub-keys
+                formatted_response += f"**{key.capitalize()}**:\n"
+                for sub_key, sub_value in value.items():
+                    formatted_response += f"  - **{sub_key.capitalize()}**: {sub_value}\n"
+            elif isinstance(value, list):
+                # If the value is a list, show it as a bullet-point list
+                formatted_response += f"**{key.capitalize()}**:\n"
+                for idx, item in enumerate(value):
+                    formatted_response += f"  - {item}\n"
+            else:
+                # For other types of data, simply display key-value pairs
+                formatted_response += f"**{key.capitalize()}**: {value}\n"
+
     elif isinstance(data, list):
-        formatted_response = "### Insights:\n\n" + "\n".join(f"{idx+1}. {item}" for idx, item in enumerate(data))
+        formatted_response += "### Insights:\n\n" + "\n".join(f"{idx+1}. {item}" for idx, item in enumerate(data))
     else:
-        formatted_response = f"### Key Insights:\n\n{data}"
+        formatted_response = f"### Key Insight:\n\n{data}"
 
     return formatted_response.strip()
 
@@ -116,7 +137,7 @@ def market_analysis_agent(user_input, history=[]):
             response = workforce
         else:
             # Get dynamic AI response if query doesn't match predefined terms
-            response = get_groq_response(user_input, user_language)
+            response = get_groq_response(user_input, user_language, custom_data=json.dumps(company_profile))
 
         # Format the response for easy readability and highlighting
         formatted_response = format_response(response)