Update app.py

app.py

@@ -33,6 +33,9 @@ static_folder = BASE_DIR / "static"
 app = Flask(__name__, static_folder=str(static_folder), static_url_path="/static")
 CORS(app)
 
+# Ensure the reports directory exists
+os.makedirs(REPORTS_ROOT, exist_ok=True)
+
 # --- LLM setup ---
 llm = ChatGroq(
     model=os.getenv("LLM_MODEL", "meta-llama/llama-4-scout-17b-16e-instruct"),
@@ -55,21 +58,22 @@ def clean_notes_with_bloatectomy(text: str, style: str = "remov") -> str:
 
 # --- Agent prompt instructions ---
 PATIENT_ASSISTANT_PROMPT = """
-You are a patient assistant helping to analyze medical records and reports. Your goal is to provide a comprehensive response based on the patient's health history and the current conversation.
+You are a patient assistant helping to analyze medical records and reports. Your primary task is to get the patient ID (PID) from the user at the start of the conversation.
+
+Once you have the PID, you will be provided with a summary of the patient's medical reports. Use this information, along with the conversation history, to provide a comprehensive response.
 
 Your tasks include:
+- **First, ask for the patient ID.** Do not proceed with any other task until you have the PID.
 - Analyzing medical records and reports to detect anomalies, redundant tests, or misleading treatments.
 - Suggesting preventive care based on the overall patient health history.
-- Optimizing healthcare costs by comparing past visits and treatments, helping patients make smarter choices.
-- Offering personalized lifestyle recommendations, such as adopting healthier food habits, daily routines, and regular health checks.
+- Optimizing healthcare costs by comparing past visits and treatments.
+- Offering personalized lifestyle recommendations.
 - Generating a natural, helpful reply to the user.
 
-You will be provided with the last user message, the conversation history, and a summary of the patient's medical reports. Use this information to give a tailored and informative response.
-
 STRICT OUTPUT FORMAT (JSON ONLY):
 Return a single JSON object with the following keys:
 - assistant_reply: string  // a natural language reply to the user (short, helpful, always present)
-- patientDetails: object  // keys may include name, problem, city, contact (update if user shared info)
+- patientDetails: object  // keys may include name, problem, pid (patient ID), city, contact (update if user shared info)
 - conversationSummary: string (optional)  // short summary of conversation + relevant patient docs
 
 Rules:
@@ -129,10 +133,30 @@ def extract_json_from_llm_response(raw_response: str) -> dict:
 def serve_frontend():
     """Serves the frontend HTML file."""
     try:
-        return app.send_static_file("frontend.html")
+        return app.send_static_file("frontend_p.html")
     except Exception:
         return "<h3>frontend2.html not found in static/ — please add your frontend2.html there.</h3>", 404
 
+@app.route("/upload_report", methods=["POST"])
+def upload_report():
+    """Handles the upload of a new PDF report for a specific patient."""
+    if 'report' not in request.files:
+        return jsonify({"error": "No file part in the request"}), 400
+    
+    file = request.files['report']
+    patient_id = request.form.get("patient_id")
+
+    if file.filename == '' or not patient_id:
+        return jsonify({"error": "No selected file or patient ID"}), 400
+
+    if file:
+        filename = secure_filename(file.filename)
+        patient_folder = REPORTS_ROOT / f"p_{patient_id}"
+        os.makedirs(patient_folder, exist_ok=True)
+        file_path = patient_folder / filename
+        file.save(file_path)
+        return jsonify({"message": f"File '{filename}' uploaded successfully for patient ID '{patient_id}'."}), 200
+
 @app.route("/chat", methods=["POST"])
 def chat():
     """Handles the chat conversation with the assistant."""
@@ -140,36 +164,9 @@ def chat():
     if not isinstance(data, dict):
         return jsonify({"error": "invalid request body"}), 400
 
-    patient_id = data.get("patient_id")
-    if not patient_id:
-        return jsonify({"error": "patient_id required"}), 400
-
     chat_history = data.get("chat_history") or []
     patient_state = data.get("patient_state") or {}
-
-    # --- Read and parse patient reports ---
-    patient_folder = REPORTS_ROOT / f"p_{patient_id}"
-    combined_text_parts = []
-    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)
+    patient_id = patient_state.get("patientDetails", {}).get("pid")
     
     # --- Prepare the state for the LLM ---
     state = patient_state.copy()
@@ -180,6 +177,49 @@ def chat():
             if msg.get("role") == "user" and msg.get("content"):
                 state["lastUserMessage"] = msg["content"]
                 break
+
+    combined_text_parts = []
+    # If a PID is not yet known, prompt the agent to ask for it.
+    if not patient_id:
+        # A simple prompt to get the agent to ask for the PID.
+        user_prompt = "Hello. I need to get the patient's ID to proceed."
+        
+        # Check if the user's last message contains a possible number for the PID
+        last_message = state.get("lastUserMessage", "")
+        # A very basic check to see if the user provided a number
+        if re.search(r'\d+', last_message):
+            inferred_pid = re.search(r'(\d+)', last_message).group(1)
+            state["patientDetails"] = {"pid": inferred_pid}
+            patient_id = inferred_pid
+            # Now that we have a PID, let the agent know to process the reports.
+            user_prompt = f"The user provided a patient ID: {inferred_pid}. Please access their reports and respond."
+        else:
+            # If no PID is found, the agent should ask for it.
+            user_prompt = "The patient has not provided a patient ID. Please ask them to provide it to proceed."
+    
+    # If a PID is known, load the patient reports.
+    if patient_id:
+        patient_folder = REPORTS_ROOT / f"p_{patient_id}"
+        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)
     
     # Update the conversation summary with the parsed documents
     base_summary = state.get("conversationSummary", "") or ""
@@ -238,5 +278,5 @@ def ping():
     return jsonify({"status": "ok"})
 
 if __name__ == "__main__":
-    port = int(os.getenv("PORT", 7860))
-    app.run(host="0.0.0.0", port=port, debug=True)
+    port = int(os.getenv("PORT", 5000))
+    app.run(host="0.0.0.0", port=port, debug=True)