Update app.py

app.py

@@ -228,6 +228,7 @@ def chat():
 
     wants_records = needs_pid_for_query(state.get("lastUserMessage", "") or "")
 
+    # If user wants records but no PID yet, ask for PID (same behavior as before)
     if wants_records and (not patient_id or patient_id_str.strip() == ""):
         if not state.get("asked_for_pid", False):
             assistant_reply = "Please provide the patient ID (PID) to retrieve previous records."
@@ -248,12 +249,10 @@ def chat():
             }
             return jsonify(response_payload)
 
-    # If we have a patient_id, check for files — if none, return a clear assistant reply.
+    # If we have a PID, check whether any allowed files exist for that PID.
+    has_allowed_files = False
     if patient_id and str(patient_id).strip() != "":
         patient_folder = REPORTS_ROOT / f"{patient_id}"
-
-        # If folder does not exist or contains no allowed files, short-circuit with a friendly LLM-like reply.
-        has_allowed_files = False
         if patient_folder.exists() and patient_folder.is_dir():
             for f in patient_folder.iterdir():
                 if f.is_file():
@@ -262,44 +261,37 @@ def chat():
                         has_allowed_files = True
                         break
 
-        if not has_allowed_files:
-            # update state so caller knows we checked and found nothing
-            state.setdefault("patientDetails", {})["pid"] = str(patient_id)
-            state["no_records_found"] = True
-            # Optionally append to conversationSummary to keep a trace
-            base_summary = state.get("conversationSummary", "") or ""
-            note = f"No medical records found for patient ID {patient_id}."
-            state["conversationSummary"] = (base_summary + "\n\n" + note).strip()
-
-            assistant_reply = f"No medical records are available for patient ID {patient_id}."
-            response_payload = {
-                "assistant_reply": assistant_reply,
-                "updated_state": state,
-            }
-            return jsonify(response_payload)
+        # IMPORTANT: do NOT short-circuit here.
+        # If the user explicitly asked for previous records (wants_records == True)
+        # and we have no files, we will tell the LLM that there are no uploaded records
+        # via the SYSTEM_HINT (so LLM can respond appropriately). We DO NOT return early,
+        # and we DO NOT add any extra JSON fields to the response.
+        if has_allowed_files:
+            # read files and build combined_text_parts (existing behavior)
+            for fname in sorted(os.listdir(patient_folder)):
+                file_path = patient_folder / fname
+                page_text = ""
+                if partition_pdf is not None and str(file_path).lower().endswith('.pdf'):
+                    try:
+                        elements = partition_pdf(filename=str(file_path))
+                        page_text = "\n".join([el.text for el in elements if hasattr(el, 'text') and el.text])
+                    except Exception:
+                        logger.exception("Failed to parse PDF %s", file_path)
+                else:
+                    try:
+                        page_text = file_path.read_text(encoding='utf-8', errors='ignore')
+                    except Exception:
+                        page_text = ""
+
+                if page_text:
+                    cleaned = clean_notes_with_bloatectomy(page_text, style="remov")
+                    if cleaned:
+                        combined_text_parts.append(cleaned)
+        else:
+            # no files: do not modify state or return. We'll include a hint for the LLM below
+            logger.info("No uploaded files found for PID %s. Will inform LLM only if user asked for records.", patient_id)
 
-        # If we do have files, proceed to read them (existing behavior)
-        for fname in sorted(os.listdir(patient_folder)):
-            file_path = patient_folder / fname
-            page_text = ""
-            if partition_pdf is not None and str(file_path).lower().endswith('.pdf'):
-                try:
-                    elements = partition_pdf(filename=str(file_path))
-                    page_text = "\n".join([el.text for el in elements if hasattr(el, 'text') and el.text])
-                except Exception:
-                    logger.exception("Failed to parse PDF %s", file_path)
-            else:
-                try:
-                    page_text = file_path.read_text(encoding='utf-8', errors='ignore')
-                except Exception:
-                    page_text = ""
-
-            if page_text:
-                cleaned = clean_notes_with_bloatectomy(page_text, style="remov")
-                if cleaned:
-                    combined_text_parts.append(cleaned)
-
-    # (rest of the original function continues unchanged)
+    # Build conversationSummary from any docs we read (unchanged)
     base_summary = state.get("conversationSummary", "") or ""
     docs_summary = "\n\n".join(combined_text_parts)
     if docs_summary:
@@ -307,8 +299,15 @@ def chat():
     else:
         state["conversationSummary"] = base_summary
 
+    # Prepare the action hint. If user asked for records but there are no uploaded files,
+    # explicitly tell the LLM so it can respond like "No records available for PID X".
     if patient_id and str(patient_id).strip() != "":
-        action_hint = f"Use the patient ID {patient_id} to retrieve and summarize any relevant reports."
+        if wants_records and not has_allowed_files:
+            action_hint = (
+                f"User asked about prior records. NOTE: there are NO uploaded medical records for patient ID {patient_id}."
+            )
+        else:
+            action_hint = f"Use the patient ID {patient_id} to retrieve and summarize any relevant reports."
     else:
         action_hint = "No PID provided and the user's request does not need prior records. Provide helpful, general medical guidance and offer to retrieve records if the user later supplies a PID."