Update app.py

app.py

@@ -41,7 +41,6 @@ if uploaded_files:
         try:
             content = json.load(f)
             st.session_state.json_data[f.name] = content
-            # For summary in system prompt
             if isinstance(content, dict):
                 keys = list(content.keys())
             elif isinstance(content, list) and content and isinstance(content[0], dict):
@@ -72,7 +71,6 @@ else:
 
 # --- Functions for querying JSON files
 def search_json(file_name, key, value):
-    """Return all records in the given JSON file (list of dicts) where key == value."""
     try:
         data = st.session_state.json_data[file_name]
         if isinstance(data, list):
@@ -89,7 +87,6 @@ def search_json(file_name, key, value):
         return {"error": str(e)}
 
 def list_keys(file_name):
-    """Return all top-level keys of the JSON file."""
     try:
         data = st.session_state.json_data[file_name]
         if isinstance(data, dict):
@@ -102,7 +99,6 @@ def list_keys(file_name):
         return {"error": str(e)}
 
 def count_key_occurrences(file_name, key):
-    """Count number of occurrences of a given key in the JSON file."""
     try:
         data = st.session_state.json_data[file_name]
         if isinstance(data, dict):
@@ -173,7 +169,6 @@ def send_message():
     user_input = st.session_state.temp_input
     if user_input and user_input.strip():
         st.session_state.messages.append({"role": "user", "content": user_input})
-        # Limit history for context size
         chat_messages = st.session_state.messages
         if len(chat_messages) > 10:
             chat_messages = [chat_messages[0]] + chat_messages[-9:]
@@ -196,8 +191,6 @@ def send_message():
             )
             chat_resp.raise_for_status()
             response_json = chat_resp.json()
-            st.write("RAW LLM RESPONSE:", response_json)  # DEBUG
-
             msg = response_json["choices"][0]["message"]
 
             # If OpenAI requests a function call
@@ -205,7 +198,6 @@ def send_message():
                 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(
@@ -242,7 +234,6 @@ def send_message():
                 )
                 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: