Update app.py

app.py

@@ -147,7 +147,6 @@ def search_text_list(text_list: Optional[List[str]], search_terms: List[str]) ->
                 break
     return found_snippets
 
-
 # --- Other Helper Functions (parse_bp, check_red_flags, format_patient_data_for_prompt) ---
 def parse_bp(bp_string: str) -> Optional[tuple[int, int]]:
     if not isinstance(bp_string, str):
@@ -235,7 +234,6 @@ def format_patient_data_for_prompt(data: dict) -> str:
             prompt_str += f"**{section_title}:** {value}\n"
     return prompt_str.strip()
 
-
 # --- Tool Definitions ---
 class LabOrderInput(BaseModel):
     test_name: str = Field(...)
@@ -261,6 +259,9 @@ class FlagRiskInput(BaseModel):
 
 @tool("order_lab_test", args_schema=LabOrderInput)
 def order_lab_test(test_name: str, reason: str, priority: str = "Routine") -> str:
+    """
+    Orders a lab test with the specified test name, reason, and priority.
+    """
     print(f"Executing order_lab_test: {test_name}, Reason: {reason}, Priority: {priority}")
     return json.dumps({
         "status": "success",
@@ -270,6 +271,9 @@ def order_lab_test(test_name: str, reason: str, priority: str = "Routine") -> st
 
 @tool("prescribe_medication", args_schema=PrescriptionInput)
 def prescribe_medication(medication_name: str, dosage: str, route: str, frequency: str, duration: str, reason: str) -> str:
+    """
+    Prepares a prescription for the specified medication including dosage, route, frequency, and duration.
+    """
     print(f"Executing prescribe_medication: {medication_name} {dosage}...")
     return json.dumps({
         "status": "success",
@@ -279,6 +283,9 @@ def prescribe_medication(medication_name: str, dosage: str, route: str, frequenc
 
 @tool("check_drug_interactions", args_schema=InteractionCheckInput)
 def check_drug_interactions(potential_prescription: str, current_medications: Optional[List[str]] = None, allergies: Optional[List[str]] = None) -> str:
+    """
+    Checks for potential drug interactions and allergy risks for the given prescription.
+    """
     print(f"\n--- Executing REAL check_drug_interactions ---")
     print(f"Checking potential prescription: '{potential_prescription}'")
     warnings = []
@@ -359,6 +366,9 @@ def check_drug_interactions(potential_prescription: str, current_medications: Op
 
 @tool("flag_risk", args_schema=FlagRiskInput)
 def flag_risk(risk_description: str, urgency: str) -> str:
+    """
+    Flags a clinical risk with the provided description and urgency.
+    """
     print(f"Executing flag_risk: {risk_description}, Urgency: {urgency}")
     st.error(f"🚨 **{urgency.upper()} RISK FLAGGED by AI:** {risk_description}", icon="🚨")
     return json.dumps({