Update logic/nlp_report.py

logic/nlp_report.py

@@ -15,7 +15,7 @@ model_path = hf_hub_download(
 
 llm = Llama(
     model_path=model_path,
-    n_ctx=32048,
+    n_ctx=2048,
     chat_format="chatml",
     verbose=False
 )
@@ -72,3 +72,23 @@ Write a concise and insightful clinical summary, including key gaps and social c
     except Exception as e:
         logging.error(f"LLM error: {e}")
         return "[Error generating summary]"
+
+# Additional: Quality validation helper for UI feedback
+
+def summarize_data_quality(df):
+    issues = []
+    required_cols = ['patient_id', 'age', 'gender', 'hcc_codes']
+    for col in required_cols:
+        if col not in df.columns:
+            issues.append(f"Missing column: {col}")
+        elif df[col].isnull().any():
+            issues.append(f"Null values in column: {col}")
+
+    percent_missing = df.isnull().mean().round(2) * 100
+    high_missing = percent_missing[percent_missing > 30].to_dict()
+    if high_missing:
+        for col, pct in high_missing.items():
+            issues.append(f"Over 30% missing in: {col} ({pct:.0f}%)")
+
+    summary_df = pd.DataFrame({"Issues": issues})
+    return summary_df