Update app.py

app.py

@@ -52,19 +52,27 @@ if uploaded_files:
 
     # --- System prompt with explicit example ---
     system_message = {
-        "role": "system",
-        "content": (
-            "You are an AI data analyst for uploaded JSON files. "
-            "Each file may have different structures and keys, including lists and nested dictionaries. "
-            "You have access to a function 'search_all_jsons' that finds all records in all JSON files where a key matches a value, recursively. "
-            "If a user asks about groups of people or wants to know counts such as 'How many females are there?', "
-            "interpret this as 'search for all records where gender equals female'. "
-            "Always use the 'search_all_jsons' function with key='gender' and value='female' for such queries, unless another key/value is clear from context. "
-            "Example:\n"
-            "User: How many females are there?\n"
-            "Assistant: (Call search_all_jsons with key='gender', value='female')"
-        )
-    }
+    "role": "system",
+    "content": (
+        "You are an AI data analyst for uploaded JSON files. "
+        "Each file may have different structures and keys, including lists and nested dictionaries. "
+        "You have access to a function 'search_all_jsons' that finds all records in all JSON files where a key matches a value, recursively. "
+        "If a user asks about groups of people or wants to know counts such as 'How many females are there?', "
+        "interpret this as 'search for all records where gender equals female'. "
+        "Always use the 'search_all_jsons' function with key='gender' and value='female' for such queries, unless another key/value is clear from context. "
+        "If someone asks 'How many males?', search for gender equals male, etc. "
+        "EXAMPLES:\n"
+        "User: How many females are there?\n"
+        "Assistant: (Call search_all_jsons with key='gender', value='female')\n"
+        "User: How many males are there?\n"
+        "Assistant: (Call search_all_jsons with key='gender', value='male')\n"
+        "User: Show all females\n"
+        "Assistant: (Call search_all_jsons with key='gender', value='female')\n"
+        "User: How many people are named Emily?\n"
+        "Assistant: (Call search_all_jsons with key='firstName', value='Emily')"
+    )
+}
+
     st.session_state.messages = [system_message]
 else:
     st.session_state.json_data.clear()