Update app.py

app.py

@@ -3,7 +3,6 @@ import json
 import difflib
 import re
 
-# --- Fuzzy search utilities ---
 def normalize(s):
     return ' '.join(str(s).lower().replace("_", " ").replace("-", " ").replace(".", " ").split())
 
@@ -11,7 +10,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
 
-# --- Fuzzy, global string value search (all keys/fields) ---
 def recursive_fuzzy_value_search(target_value):
     matches = []
     norm_target = normalize(target_value)
@@ -19,8 +17,8 @@ def recursive_fuzzy_value_search(target_value):
         def _search(obj, path):
             if isinstance(obj, dict):
                 for k, v in obj.items():
-                    # Match ANY string value (not just specific keys)
-                    if isinstance(v, str) and is_fuzzy_match(norm_target, normalize(v)):
+                    # Match ANY primitive value (str, int, float, bool)
+                    if isinstance(v, (str, int, float, bool)) and is_fuzzy_match(norm_target, normalize(v)):
                         matches.append({
                             "match_path": path + [k],
                             "matched_value": v,
@@ -31,7 +29,7 @@ def recursive_fuzzy_value_search(target_value):
                     # Check inside nested dicts
                     if isinstance(v, dict):
                         for nk, nv in v.items():
-                            if isinstance(nv, str) and is_fuzzy_match(norm_target, normalize(nv)):
+                            if isinstance(nv, (str, int, float, bool)) and is_fuzzy_match(norm_target, normalize(nv)):
                                 matches.append({
                                     "match_path": path + [k, nk],
                                     "matched_value": nv,
@@ -46,18 +44,17 @@ def recursive_fuzzy_value_search(target_value):
         _search(data, [])
     return matches
 
-# --- Show all string values (for debug) ---
 def show_all_strings():
     found = []
     for file_name, data in st.session_state.json_data.items():
         def recursive(obj, path):
             if isinstance(obj, dict):
                 for k, v in obj.items():
-                    if isinstance(v, str):
+                    if isinstance(v, (str, int, float, bool)):
                         found.append(f"{file_name} | {'.'.join(path + [k])} = {v}")
                     elif isinstance(v, dict):
                         for nk, nv in v.items():
-                            if isinstance(nv, str):
+                            if isinstance(nv, (str, int, float, bool)):
                                 found.append(f"{file_name} | {'.'.join(path + [k, nk])} = {nv}")
                     recursive(v, path + [k])
             elif isinstance(obj, list):
@@ -66,9 +63,7 @@ def show_all_strings():
         recursive(data, [])
     return found
 
-# --- Query handler (searches for any string value) ---
 def handle_user_query(query):
-    # Extract likely search string (more flexible, still supports old user-centric queries)
     patterns = [
         r"(?:last\s*login.*?for|when\s+did)\s+([a-zA-Z0-9 _\-\.@]+)",
         r"when\s+was\s+([a-zA-Z0-9 _\-\.@]+)\s+last\s+(?:login|logged\s*in)",
@@ -82,8 +77,8 @@ def handle_user_query(query):
             found_value = m.group(1).strip()
             break
     if not found_value:
-        # Fallback: any word/phrase of 3+ chars (letters, digits, spaces, dashes, underscores, dots)
-        m = re.search(r"([A-Za-z0-9][A-Za-z0-9 _\-\.@]{2,})", query)
+        # Fallback: any word/phrase of 1+ char (letters, digits, spaces, dashes, underscores, dots)
+        m = re.search(r"([A-Za-z0-9][A-Za-z0-9 _\-\.@]*)", query)
         if m:
             found_value = m.group(1).strip()
     if found_value:
@@ -97,7 +92,7 @@ def handle_user_query(query):
             )
         return "\n\n".join(answers)
     else:
-        return "No valid search value detected. Try a person's name, product, device, etc."
+        return "No valid search value detected. Try a person's name, number, product, device, etc."
 
 # --- Streamlit UI setup ---
 if "json_data" not in st.session_state:
@@ -110,7 +105,7 @@ if "files_loaded" not in st.session_state:
     st.session_state.files_loaded = False
 
 st.set_page_config(page_title="Flexible JSON Fuzzy Search", layout="wide")
-st.title("Instant JSON-Backed Q&A (Flexible Fuzzy Search — All Keys!)")
+st.title("Instant JSON-Backed Q&A (Flexible Fuzzy Search — All Keys & Types!)")
 
 uploaded_files = st.sidebar.file_uploader(
     "Choose one or more JSON files", type="json", accept_multiple_files=True
@@ -130,7 +125,7 @@ elif not uploaded_files:
     st.session_state.json_data.clear()
     st.session_state.files_loaded = False
 
-st.markdown("### Ask about ANY value (name, product, device, etc) — partials/typos/substring OK!")
+st.markdown("### Ask about ANY value (name, product, number, device, etc) — partials/typos/substring OK!")
 for msg in st.session_state.messages:
     if msg["role"] == "user":
         st.markdown(f"<div style='color: #4F8BF9;'><b>User:</b> {msg['content']}</div>", unsafe_allow_html=True)
@@ -147,7 +142,7 @@ def send_message():
 
 if st.session_state.json_data:
     st.text_input("Your message:", key="temp_input", on_change=send_message)
-    if st.button("Show all strings in uploaded JSONs"):
+    if st.button("Show all values in uploaded JSONs"):
         st.write(show_all_strings())
 else:
     st.info("Please upload at least one JSON file to start chatting.")