Update app.py

app.py

@@ -4,11 +4,9 @@ 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
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 if not OPENAI_API_KEY:
     st.error("❌ OPENAI_API_KEY not set in Settings → Secrets.")
@@ -19,13 +17,11 @@ HEADERS = {
     "Content-Type": "application/json",
 }
 
-# --- Sidebar: Multiple JSON upload & preview
 st.sidebar.header("Upload Multiple JSON Files")
 uploaded_files = st.sidebar.file_uploader(
     "Choose one or more JSON files", type="json", accept_multiple_files=True
 )
 
-# --- Session State for data and chat
 if "json_data" not in st.session_state:
     st.session_state.json_data = {}
 if "messages" not in st.session_state:
@@ -33,7 +29,6 @@ if "messages" not in st.session_state:
 if "temp_input" not in st.session_state:
     st.session_state.temp_input = ""
 
-# --- Load all JSON files
 if uploaded_files:
     st.session_state.json_data.clear()
     file_summaries = []
@@ -53,7 +48,6 @@ if uploaded_files:
         except Exception as e:
             st.sidebar.error(f"Error reading {f.name}: {e}")
 
-    # Compose system prompt for the LLM
     system_message = {
         "role": "system",
         "content": (
@@ -69,7 +63,6 @@ if uploaded_files:
 else:
     st.session_state.json_data.clear()
 
-# --- Functions for querying JSON files
 def search_json(file_name, key, value):
     try:
         data = st.session_state.json_data[file_name]
@@ -110,7 +103,6 @@ def count_key_occurrences(file_name, key):
     except Exception as e:
         return {"error": str(e)}
 
-# --- Function schemas for OpenAI
 function_schema = [
     {
         "name": "search_json",
@@ -150,13 +142,16 @@ function_schema = [
     },
 ]
 
-# --- Chat interface
 st.markdown("### Conversation")
-for i, msg in enumerate(st.session_state.messages[1:]):  # Skip system message for display
+for i, msg in enumerate(st.session_state.messages[1:]):
     if msg["role"] == "user":
         st.markdown(f"<div style='color: #4F8BF9;'><b>User:</b> {msg['content']}</div>", unsafe_allow_html=True)
     elif msg["role"] == "assistant":
-        st.markdown(f"<div style='color: #1C6E4C;'><b>Agent:</b> {msg['content']}</div>", unsafe_allow_html=True)
+        content = msg.get("content", "")
+        if content.strip():
+            st.markdown(f"<div style='color: #1C6E4C;'><b>Agent:</b> {content}</div>", unsafe_allow_html=True)
+        else:
+            st.markdown(f"<div style='color: #DC143C;'><b>Agent:</b> [No response generated]</div>", unsafe_allow_html=True)
     elif msg["role"] == "function":
         try:
             result = json.loads(msg["content"])
@@ -164,7 +159,6 @@ for i, msg in enumerate(st.session_state.messages[1:]):  # Skip system message f
         except Exception:
             st.markdown(f"<b>Function '{msg['name']}' output:</b> {msg['content']}", unsafe_allow_html=True)
 
-# --- Sending user input and OpenAI call logic using a callback
 def send_message():
     user_input = st.session_state.temp_input
     if user_input and user_input.strip():
@@ -175,12 +169,11 @@ def send_message():
         else:
             chat_messages = chat_messages.copy()
         try:
-            # OpenAI call
             chat_resp = requests.post(
                 "https://api.openai.com/v1/chat/completions",
                 headers=HEADERS,
                 json={
-                    "model": "gpt-4.1",
+                    "model": "gpt-4o",
                     "messages": chat_messages,
                     "functions": function_schema,
                     "function_call": "auto",
@@ -193,7 +186,6 @@ def send_message():
             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"]
@@ -215,7 +207,6 @@ def send_message():
                     "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:]
@@ -225,7 +216,7 @@ def send_message():
                     "https://api.openai.com/v1/chat/completions",
                     headers=HEADERS,
                     json={
-                        "model": "gpt-4.1",
+                        "model": "gpt-4o",
                         "messages": followup_messages,
                         "temperature": 0,
                         "max_tokens": 1500,