Update app.py

app.py

@@ -5,10 +5,11 @@ import requests
 import traceback
 import difflib
 
-# ---- BASIC NAME GUESS FOR FEMALE NAMES (expand as needed)
-COMMON_FEMALE_NAMES = {"alice", "mary", "lisa", "jane", "emily", "sophia", "emma", "olivia", "ava", "mia", "isabella", "charlotte", "amelia", "harper", "abigail"}
+COMMON_FEMALE_NAMES = {
+    "alice", "mary", "lisa", "jane", "emily", "sophia", "emma", "olivia",
+    "ava", "mia", "isabella", "charlotte", "amelia", "harper", "abigail"
+}
 
-# ---- SESSION STATE ----
 if "json_data" not in st.session_state:
     st.session_state.json_data = {}
 if "messages" not in st.session_state:
@@ -17,11 +18,16 @@ if "files_loaded" not in st.session_state:
     st.session_state.files_loaded = False
 if "temp_input" not in st.session_state:
     st.session_state.temp_input = ""
+if "modal_open" not in st.session_state:
+    st.session_state.modal_open = False
+if "modal_content" not in st.session_state:
+    st.session_state.modal_content = ""
+if "modal_title" not in st.session_state:
+    st.session_state.modal_title = ""
 
 st.set_page_config(page_title="Chat with Your JSONs", layout="wide")
 st.title("Chat with Your JSON Files (OpenAI function-calling, No LangChain)")
 
-# ---- UPLOAD FILES ----
 uploaded_files = st.sidebar.file_uploader(
     "Choose one or more JSON files", type="json", accept_multiple_files=True
 )
@@ -40,7 +46,6 @@ elif not uploaded_files:
     st.session_state.json_data.clear()
     st.session_state.files_loaded = False
 
-# ---- UTILS ----
 def normalize(s):
     return ' '.join(str(s).lower().replace("_", " ").replace("-", " ").replace(".", " ").split())
 
@@ -48,7 +53,6 @@ def is_fuzzy_match(a, b, threshold=0.7):
     ratio = difflib.SequenceMatcher(None, a, b).ratio()
     return ratio >= threshold or a in b or b in a
 
-# ---- SEARCH ALL KEYS FOR KEY/VALUE MATCH ----
 def search_all_jsons(key, value):
     matches = []
     value_norm = normalize(value)
@@ -71,7 +75,6 @@ def search_all_jsons(key, value):
         recursive_search(data)
     return matches
 
-# ---- FUZZY VALUE SEARCH (returns WHOLE RECORD) ----
 def fuzzy_value_search(value):
     matches = []
     value_norm = normalize(value)
@@ -93,7 +96,6 @@ def fuzzy_value_search(value):
         recursive_search(data)
     return matches
 
-# ---- LIST KEYS ----
 def list_keys(file_name):
     try:
         data = st.session_state.json_data[file_name]
@@ -106,7 +108,6 @@ def list_keys(file_name):
     except Exception as e:
         return {"error": str(e)}
 
-# ---- COUNT KEY OCCURRENCES ----
 def count_key_occurrences(file_name, key):
     try:
         data = st.session_state.json_data[file_name]
@@ -126,7 +127,6 @@ def count_key_occurrences(file_name, key):
     except Exception as e:
         return {"error": str(e)}
 
-# ---- FIND/COUNT IN ARRAYS (e.g., COMPLETED TASKS) ----
 def find_in_arrays(key, value, return_count=True):
     matches = []
     count = 0
@@ -141,7 +141,8 @@ def find_in_arrays(key, value, return_count=True):
                                 matches.append({
                                     "file": file_name,
                                     "item": item,
-                                    "array_path": key
+                                    "array_path": key,
+                                    "record": item  # show the item dict itself for popup
                                 })
                                 count += 1
                     recursive(item)
@@ -151,7 +152,6 @@ def find_in_arrays(key, value, return_count=True):
         recursive(data)
     return count if return_count else matches
 
-# ---- SUM FIELD BY NAME (e.g., TOTAL AMOUNT FOR JOHNNY) ----
 def sum_field_by_name(name, field="amount"):
     total = 0
     details = []
@@ -162,7 +162,6 @@ def sum_field_by_name(name, field="amount"):
             if isinstance(obj, dict):
                 for k, v in obj.items():
                     if isinstance(v, (str, int, float, bool)) and is_fuzzy_match(name_norm, normalize(v)):
-                        # Look for field in this or sibling dict
                         if field in obj:
                             try:
                                 amt = float(obj[field])
@@ -177,7 +176,6 @@ def sum_field_by_name(name, field="amount"):
         recursive(data)
     return {"total": total, "matches": details}
 
-# ---- COUNT FEMALE NAMES (guess from common names) ----
 def count_female_names():
     count = 0
     names = []
@@ -198,7 +196,6 @@ def count_female_names():
         recursive(data)
     return {"count": count, "names": names}
 
-# ---- FUNCTION SCHEMA for OpenAI ----
 function_schema = [
     {
         "name": "search_all_jsons",
@@ -281,7 +278,6 @@ function_schema = [
     }
 ]
 
-# ---- SYSTEM PROMPT ----
 system_message = {
     "role": "system",
     "content": (
@@ -296,7 +292,6 @@ system_message = {
     )
 }
 
-# ---- CHAT UI (with OpenAI function-calling!) ----
 st.markdown("### Ask any question about your data, just like ChatGPT.")
 
 for msg in st.session_state.messages:
@@ -307,6 +302,55 @@ for msg in st.session_state.messages:
     elif msg["role"] == "function":
         st.markdown(f"<details><summary><b>Function '{msg['name']}' output:</b></summary><pre>{msg['content']}</pre></details>", unsafe_allow_html=True)
 
+# --- JSON MODAL POPUP ---
+def show_json_links_and_modal():
+    # Find last function message
+    for msg in reversed(st.session_state.messages):
+        if msg.get("role") == "function":
+            func_name = msg.get("name")
+            try:
+                content = json.loads(msg["content"])
+            except Exception:
+                content = None
+            links_shown = False
+            if isinstance(content, list):
+                for idx, match in enumerate(content):
+                    if isinstance(match, dict) and "record" in match:
+                        if st.button(f"View JSON: {match.get('file', 'unknown')} record #{idx+1}", key=f"modal_{func_name}_{idx}"):
+                            st.session_state.modal_open = True
+                            st.session_state.modal_content = json.dumps(match["record"], indent=2)
+                            st.session_state.modal_title = f"{match.get('file', 'unknown')} record #{idx+1}"
+                        links_shown = True
+            elif isinstance(content, dict):
+                # For dicts with matches
+                if "matches" in content and isinstance(content["matches"], list):
+                    for idx, match in enumerate(content["matches"]):
+                        if isinstance(match, dict) and "record" in match:
+                            if st.button(f"View JSON: {match.get('file', 'unknown')} record #{idx+1}", key=f"modal_{func_name}_matches_{idx}"):
+                                st.session_state.modal_open = True
+                                st.session_state.modal_content = json.dumps(match["record"], indent=2)
+                                st.session_state.modal_title = f"{match.get('file', 'unknown')} record #{idx+1}"
+                            links_shown = True
+                if "names" in content and isinstance(content["names"], list):
+                    for idx, match in enumerate(content["names"]):
+                        if isinstance(match, dict) and "record" in match:
+                            if st.button(f"View JSON: {match.get('file', 'unknown')} record #{idx+1}", key=f"modal_{func_name}_names_{idx}"):
+                                st.session_state.modal_open = True
+                                st.session_state.modal_content = json.dumps(match["record"], indent=2)
+                                st.session_state.modal_title = f"{match.get('file', 'unknown')} record #{idx+1}"
+                            links_shown = True
+            if links_shown:
+                break
+
+    # Modal popup UI using st.expander as a modal hack
+    if st.session_state.modal_open:
+        with st.expander(f"JSON Record: {st.session_state.modal_title}", expanded=True):
+            st.code(st.session_state.modal_content, language="json")
+            if st.button("Close", key="close_modal"):
+                st.session_state.modal_open = False
+
+show_json_links_and_modal()
+
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 HEADERS = {
     "Authorization": f"Bearer {OPENAI_API_KEY}",
@@ -323,7 +367,7 @@ def send_message():
                 "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",
@@ -341,7 +385,6 @@ def send_message():
                 args_json = msg["function_call"]["arguments"]
                 args = json.loads(args_json)
 
-                # Call the right function
                 if func_name == "search_all_jsons":
                     result = search_all_jsons(args.get("key"), args.get("value"))
                 elif func_name == "fuzzy_value_search":
@@ -372,7 +415,6 @@ def send_message():
                     "content": json.dumps(result, indent=2),
                 })
 
-                # Let LLM observe function output and craft final answer
                 followup_messages = chat_messages + [
                     {"role": "function", "name": func_name, "content": json.dumps(result, indent=2)}
                 ]
@@ -380,7 +422,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": 1200,
@@ -391,8 +433,6 @@ def send_message():
                 final_json = final_resp.json()
                 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"]})
         st.session_state.temp_input = ""
     except Exception as e:
         st.error("Exception: " + str(e))