Update app.py

app.py

@@ -248,37 +248,66 @@ def chat():
             }
             return jsonify(response_payload)
 
-    if patient_id and patient_id_str.strip() != "":
+    # If we have a patient_id, check for files — if none, return a clear assistant reply.
+    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 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)
-
-        base_summary = state.get("conversationSummary", "") or ""
-        docs_summary = "\n\n".join(combined_text_parts)
-        if docs_summary:
-            state["conversationSummary"] = (base_summary + "\n\n" + docs_summary).strip()
-        else:
-            state["conversationSummary"] = base_summary
+            for f in patient_folder.iterdir():
+                if f.is_file():
+                    ext = f.suffix.lower().lstrip(".")
+                    if ext in ALLOWED_EXTENSIONS:
+                        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)
+
+        # 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)
+    base_summary = state.get("conversationSummary", "") or ""
+    docs_summary = "\n\n".join(combined_text_parts)
+    if docs_summary:
+        state["conversationSummary"] = (base_summary + "\n\n" + docs_summary).strip()
+    else:
+        state["conversationSummary"] = base_summary
 
-    if patient_id and patient_id_str.strip() != "":
+    if patient_id and str(patient_id).strip() != "":
         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."