Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 import os
 import json
 import requests
+import traceback
 
 # --- Page config
 st.set_page_config(page_title="JSON-Backed AI Chat Agent", layout="wide")
-
 st.title("JSON-Backed AI Chat Agent")
 
 # --- Load API key
@@ -66,7 +66,6 @@ if uploaded_files:
             "If the user does not specify a file, make your best guess based on keys/fields mentioned."
         ),
     }
-    # Reset chat if new files loaded
     st.session_state.messages = [system_message]
 else:
     st.session_state.json_data.clear()
@@ -77,10 +76,10 @@ def search_json(file_name, key, value):
     try:
         data = st.session_state.json_data[file_name]
         if isinstance(data, list):
-            results = [item for item in data if isinstance(item, dict) and item.get(key) == value]
+            results = [item for item in data if isinstance(item, dict) and str(item.get(key)) == str(value)]
             return results[:10]
         elif isinstance(data, dict):
-            if key in data and data[key] == value:
+            if key in data and str(data[key]) == str(value):
                 return [{key: value}]
             else:
                 return []
@@ -180,69 +179,81 @@ def send_message():
             chat_messages = [chat_messages[0]] + chat_messages[-9:]
         else:
             chat_messages = chat_messages.copy()
-        # OpenAI call
-        chat_resp = requests.post(
-            "https://api.openai.com/v1/chat/completions",
-            headers=HEADERS,
-            json={
-                "model": "gpt-4.1",  # Use latest available model for this purpose
-                "messages": chat_messages,
-                "functions": function_schema,
-                "function_call": "auto",
-                "temperature": 0,
-                "max_tokens": 1000,
-            },
-            timeout=60,
-        )
-        chat_resp.raise_for_status()
-        response_json = chat_resp.json()
-        msg = response_json["choices"][0]["message"]
-
-        # If OpenAI requests a function call
-        if msg.get("function_call"):
-            func_name = msg["function_call"]["name"]
-            args_json = msg["function_call"]["arguments"]
-            args = json.loads(args_json)
-            if func_name == "search_json":
-                function_result = search_json(args.get("file_name"), args.get("key"), args.get("value"))
-            elif func_name == "list_keys":
-                function_result = list_keys(args.get("file_name"))
-            elif func_name == "count_key_occurrences":
-                function_result = count_key_occurrences(args.get("file_name"), args.get("key"))
-            else:
-                function_result = {"error": f"Unknown function: {func_name}"}
-            st.session_state.messages.append({
-                "role": "function",
-                "name": func_name,
-                "content": json.dumps(function_result),
-            })
-            # Second call to OpenAI for the final answer
-            followup_messages = st.session_state.messages
-            if len(followup_messages) > 12:
-                followup_messages = [followup_messages[0]] + followup_messages[-11:]
-            else:
-                followup_messages = followup_messages.copy()
-            final_resp = requests.post(
+        try:
+            # OpenAI call
+            chat_resp = requests.post(
                 "https://api.openai.com/v1/chat/completions",
                 headers=HEADERS,
                 json={
                     "model": "gpt-4.1",
-                    "messages": followup_messages,
+                    "messages": chat_messages,
+                    "functions": function_schema,
+                    "function_call": "auto",
                     "temperature": 0,
-                    "max_tokens": 1500,
+                    "max_tokens": 1000,
                 },
                 timeout=60,
             )
-            final_resp.raise_for_status()
-            answer = final_resp.json()["choices"][0]["message"]["content"]
-            st.session_state.messages.append({"role": "assistant", "content": answer})
-        else:
-            st.session_state.messages.append({"role": "assistant", "content": msg["content"]})
+            chat_resp.raise_for_status()
+            response_json = chat_resp.json()
+            st.write("RAW LLM RESPONSE:", response_json)  # DEBUG
 
-        st.session_state.temp_input = ""
+            msg = response_json["choices"][0]["message"]
+
+            # If OpenAI requests a function call
+            if msg.get("function_call"):
+                func_name = msg["function_call"]["name"]
+                args_json = msg["function_call"]["arguments"]
+                args = json.loads(args_json)
+                st.write("Function Call:", func_name, "Args:", args)  # DEBUG
+
+                if func_name == "search_json":
+                    result = search_json(
+                        args.get("file_name"), args.get("key"), args.get("value")
+                    )
+                elif func_name == "list_keys":
+                    result = list_keys(args.get("file_name"))
+                elif func_name == "count_key_occurrences":
+                    result = count_key_occurrences(args.get("file_name"), args.get("key"))
+                else:
+                    result = {"error": f"Unknown function: {func_name}"}
+
+                st.session_state.messages.append({
+                    "role": "function",
+                    "name": func_name,
+                    "content": json.dumps(result),
+                })
+                # Second call to OpenAI for the final answer
+                followup_messages = st.session_state.messages
+                if len(followup_messages) > 12:
+                    followup_messages = [followup_messages[0]] + followup_messages[-11:]
+                else:
+                    followup_messages = followup_messages.copy()
+                final_resp = requests.post(
+                    "https://api.openai.com/v1/chat/completions",
+                    headers=HEADERS,
+                    json={
+                        "model": "gpt-4.1",
+                        "messages": followup_messages,
+                        "temperature": 0,
+                        "max_tokens": 1500,
+                    },
+                    timeout=60,
+                )
+                final_resp.raise_for_status()
+                final_json = final_resp.json()
+                st.write("RAW FINAL RESPONSE:", final_json)  # DEBUG
+                answer = final_json["choices"][0]["message"]["content"]
+                st.session_state.messages.append({"role": "assistant", "content": answer})
+            else:
+                st.session_state.messages.append({"role": "assistant", "content": msg["content"]})
+        except Exception as e:
+            st.error("Exception: " + str(e))
+            st.code(traceback.format_exc())
+        finally:
+            st.session_state.temp_input = ""
 
 if st.session_state.json_data:
     st.text_input("Your message:", key="temp_input", on_change=send_message)
 else:
     st.info("Please upload at least one JSON file to start chatting.")
-