Update app.py

app.py

@@ -120,11 +120,6 @@ function_schema = [
     }
 ]
 
-function_map = {
-    "search_csv": search_csv,
-    "count_unique": count_unique,
-}
-
 # --- Chat interface
 st.markdown("### Conversation")
 for i, msg in enumerate(st.session_state.messages[1:]):  # Skip system message for display
@@ -155,7 +150,7 @@ def send_message():
             "https://api.openai.com/v1/chat/completions",
             headers=HEADERS,
             json={
-                "model": "gpt-3.5-turbo-1106",
+                "model": "gpt-4.1",
                 "messages": chat_messages,
                 "functions": function_schema,
                 "function_call": "auto",
@@ -173,9 +168,11 @@ def send_message():
             func_name = msg["function_call"]["name"]
             args_json = msg["function_call"]["arguments"]
             args = json.loads(args_json)
-            # Call the correct Python function
-            if func_name in function_map:
-                function_result = function_map[func_name](**args)
+            # --- FIXED: Only pass the expected arg for each function
+            if func_name == "search_csv":
+                function_result = search_csv(args.get("query", ""))
+            elif func_name == "count_unique":
+                function_result = count_unique(args.get("column", ""))
             else:
                 function_result = {"error": f"Unknown function: {func_name}"}
             st.session_state.messages.append({
@@ -193,7 +190,7 @@ def send_message():
                 "https://api.openai.com/v1/chat/completions",
                 headers=HEADERS,
                 json={
-                    "model": "gpt-3.5-turbo-1106",
+                    "model": "gpt-4o",
                     "messages": followup_messages,
                     "temperature": 0,
                     "max_tokens": 1500,