Update app.py

app.py

@@ -3,11 +3,12 @@
 import os
 import tempfile
 import hashlib
+import json
 import gradio as gr
 
 from embedding_manager import EmbeddingManager
 from summarizer import PatientChartSummarizer
-from hedis_engine import HedisComplianceEngine
+from interactive_hedis_engine import InteractiveHedisComplianceEngine
 
 APP_TITLE = "ChartWise AI"
 DEFAULT_MEASURE_YEAR = 2024
@@ -17,6 +18,11 @@ DEFAULT_MEASURE_YEAR = 2024
 _VDB_CACHE = {}
 _LAST_HASH = None  # optional: for quick reuse/debug
 
+# Global state for interactive HEDIS analysis
+_CURRENT_ENGINE = None
+_CURRENT_CRITERIA = None
+_ANALYSIS_STATE = "idle"  # idle, criteria_review, analysis_complete
+
 def _pdf_hash(pdf_bytes: bytes) -> str:
     return hashlib.sha256(pdf_bytes).hexdigest()
 
@@ -52,6 +58,72 @@ def _get_vectordb_from_bytes(pdf_bytes: bytes):
         except Exception:
             pass
 
+def format_criteria_for_display(criteria: dict) -> str:
+    """Format criteria dictionary into markdown for display."""
+    if "raw_output" in criteria:
+        return f"```\n{criteria['raw_output']}\n```"
+    
+    markdown = f"# HEDIS Measure Criteria\n\n"
+    
+    if "measure" in criteria:
+        markdown += f"**📋 Measure:** {criteria['measure']}\n\n"
+    
+    if "required_tests" in criteria:
+        markdown += f"## 🔬 Required Tests/Procedures\n"
+        if isinstance(criteria['required_tests'], list):
+            for test in criteria['required_tests']:
+                markdown += f"• {test}\n"
+        else:
+            markdown += f"{criteria['required_tests']}\n"
+        markdown += "\n"
+    
+    if "required_medications" in criteria:
+        markdown += f"## 💊 Required Medications\n"
+        if isinstance(criteria['required_medications'], list):
+            for med in criteria['required_medications']:
+                markdown += f"• {med}\n"
+        else:
+            markdown += f"{criteria['required_medications']}\n"
+        markdown += "\n"
+    
+    if "codes" in criteria:
+        markdown += f"## 🏷️ Relevant Codes\n"
+        if isinstance(criteria['codes'], dict):
+            for code_type, codes in criteria['codes'].items():
+                markdown += f"**{code_type}:** {codes}\n"
+        else:
+            markdown += f"{criteria['codes']}\n"
+        markdown += "\n"
+    
+    if "timeframes" in criteria:
+        markdown += f"## ⏰ Timeframes\n"
+        if isinstance(criteria['timeframes'], dict):
+            for requirement, timeframe in criteria['timeframes'].items():
+                markdown += f"**{requirement}:** {timeframe}\n"
+        else:
+            markdown += f"{criteria['timeframes']}\n"
+        markdown += "\n"
+    
+    if "inclusions" in criteria:
+        markdown += f"## ✅ Inclusion Criteria\n"
+        if isinstance(criteria['inclusions'], list):
+            for inclusion in criteria['inclusions']:
+                markdown += f"• {inclusion}\n"
+        else:
+            markdown += f"{criteria['inclusions']}\n"
+        markdown += "\n"
+    
+    if "exclusions" in criteria:
+        markdown += f"## ❌ Exclusion Criteria\n"
+        if isinstance(criteria['exclusions'], list):
+            for exclusion in criteria['exclusions']:
+                markdown += f"• {exclusion}\n"
+        else:
+            markdown += f"{criteria['exclusions']}\n"
+        markdown += "\n"
+    
+    return markdown
+
 # --- Handlers ---
 def generate_patient_summary(pdf_file):
     try:
@@ -67,23 +139,158 @@ def generate_patient_summary(pdf_file):
     except Exception as e:
         return f"❌ Error processing chart: {e}"
 
-def generate_hedis_analysis(pdf_file, measure_name, measurement_year):
+def start_hedis_analysis(pdf_file, measure_name, measurement_year):
+    """Start interactive HEDIS analysis by getting initial criteria."""
+    global _CURRENT_ENGINE, _CURRENT_CRITERIA, _ANALYSIS_STATE
+    
     try:
         if pdf_file is None:
-            return "⚠️ Please upload a PDF file first."
+            return (
+                "⚠️ Please upload a PDF file first.",
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False)
+            )
         if not measure_name:
-            return "⚠️ Please enter a HEDIS measure code (e.g., COL, BCS, CCS)."
+            return (
+                "⚠️ Please enter a HEDIS measure code (e.g., COL, BCS, CCS).",
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False)
+            )
         if not measurement_year:
-            return "⚠️ Please enter a measurement year."
+            return (
+                "⚠️ Please enter a measurement year.",
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False)
+            )
 
         # Reuse embeddings if already built for this file
         vectordb = _get_vectordb_from_bytes(pdf_file)
 
-        hedis = HedisComplianceEngine(vectordb, measure_name, int(measurement_year))
-        result = hedis.run()
-        return result
+        # Create engine and get initial criteria
+        _CURRENT_ENGINE = InteractiveHedisComplianceEngine(vectordb, measure_name, int(measurement_year))
+        _CURRENT_CRITERIA = _CURRENT_ENGINE.get_hedis_criteria_with_feedback()
+        _ANALYSIS_STATE = "criteria_review"
+        
+        criteria_display = format_criteria_for_display(_CURRENT_CRITERIA)
+        
+        return (
+            criteria_display,
+            gr.update(visible=True),  # suggestions textbox
+            gr.update(visible=True),  # approve button
+            gr.update(visible=True),  # modify button
+            gr.update(visible=True)   # cancel button
+        )
+    except Exception as e:
+        _ANALYSIS_STATE = "idle"
+        return (
+            f"❌ Error getting HEDIS criteria: {e}",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False)
+        )
+
+def approve_criteria():
+    """Approve current criteria and proceed with full analysis."""
+    global _CURRENT_ENGINE, _CURRENT_CRITERIA, _ANALYSIS_STATE
+    
+    try:
+        if _ANALYSIS_STATE != "criteria_review" or _CURRENT_ENGINE is None:
+            return (
+                "❌ No criteria to approve. Please start a new analysis.",
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False)
+            )
+        
+        # Set approved criteria and run full analysis
+        _CURRENT_ENGINE.hedis_criteria = _CURRENT_CRITERIA
+        result = _CURRENT_ENGINE.run_full_analysis()
+        _ANALYSIS_STATE = "analysis_complete"
+        
+        return (
+            f"## ✅ Analysis Complete\n\n{result}",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False)
+        )
     except Exception as e:
-        return f"❌ Error processing HEDIS analysis: {e}"
+        _ANALYSIS_STATE = "idle"
+        return (
+            f"❌ Error running analysis: {e}",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False)
+        )
+
+def modify_criteria(suggestions):
+    """Modify criteria based on suggestions."""
+    global _CURRENT_ENGINE, _CURRENT_CRITERIA, _ANALYSIS_STATE
+    
+    try:
+        if _ANALYSIS_STATE != "criteria_review" or _CURRENT_ENGINE is None:
+            return (
+                "❌ No criteria to modify. Please start a new analysis.",
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False),
+                gr.update(visible=False)
+            )
+        
+        if not suggestions.strip():
+            return (
+                "⚠️ Please provide suggestions for modification.",
+                gr.update(visible=True),
+                gr.update(visible=True),
+                gr.update(visible=True),
+                gr.update(visible=True)
+            )
+        
+        # Get modified criteria
+        _CURRENT_CRITERIA = _CURRENT_ENGINE.get_hedis_criteria_with_feedback(suggestions.strip())
+        criteria_display = format_criteria_for_display(_CURRENT_CRITERIA)
+        
+        return (
+            f"## 🔄 Modified Criteria\n\n{criteria_display}",
+            gr.update(visible=True, value=""),  # clear suggestions
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(visible=True)
+        )
+    except Exception as e:
+        _ANALYSIS_STATE = "idle"
+        return (
+            f"❌ Error modifying criteria: {e}",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False)
+        )
+
+def cancel_analysis():
+    """Cancel the current analysis."""
+    global _CURRENT_ENGINE, _CURRENT_CRITERIA, _ANALYSIS_STATE
+    
+    _CURRENT_ENGINE = None
+    _CURRENT_CRITERIA = None
+    _ANALYSIS_STATE = "idle"
+    
+    return (
+        "❌ Analysis cancelled. You can start a new analysis.",
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=False)
+    )
 
 # --- Gradio Theme ---
 simple_theme = gr.themes.Soft(
@@ -101,7 +308,7 @@ simple_theme = gr.themes.Soft(
 # --- Interface ---
 with gr.Blocks(theme=simple_theme, title="ChartWise AI") as interface:
     gr.HTML("<h1 style='text-align:center;color:#1e40af;'>🏥 ChartWise AI</h1>")
-    gr.HTML("<p style='text-align:center;color:#64748b;'>Patient Chart Analysis &amp; HEDIS Compliance</p>")
+    gr.HTML("<p style='text-align:center;color:#64748b;'>Patient Chart Analysis &amp; Interactive HEDIS Compliance</p>")
 
     with gr.Row(equal_height=True):
         # Left: Patient Summary
@@ -115,9 +322,9 @@ with gr.Blocks(theme=simple_theme, title="ChartWise AI") as interface:
                 height=400
             )
 
-        # Right: HEDIS
+        # Right: Interactive HEDIS
         with gr.Column():
-            gr.HTML("<h3 style='color:#1e40af;'>🎯 HEDIS Measure Analysis</h3>")
+            gr.HTML("<h3 style='color:#1e40af;'>🎯 Interactive HEDIS Analysis</h3>")
             hedis_measure = gr.Textbox(
                 label="HEDIS Measure Code",
                 placeholder="e.g., COL, BCS, CCS, AAB",
@@ -129,16 +336,58 @@ with gr.Blocks(theme=simple_theme, title="ChartWise AI") as interface:
                 precision=0,
                 info="e.g., 2024"
             )
-            hedis_btn = gr.Button("🎯 Run HEDIS Analysis", variant="secondary")
+            start_hedis_btn = gr.Button("🎯 Start HEDIS Analysis", variant="secondary")
+            
             hedis_output = gr.Markdown(
                 label="HEDIS Analysis Results",
-                value="Enter measure and year, then click 'Run HEDIS Analysis'.",
-                height=400
+                value="Enter measure and year, then click 'Start HEDIS Analysis' to begin interactive review.",
+                height=300
             )
+            
+            # Interactive feedback components (initially hidden)
+            with gr.Group():
+                suggestions_input = gr.Textbox(
+                    label="Suggestions for Criteria Modification",
+                    placeholder="Enter your suggestions to modify the criteria...",
+                    visible=False,
+                    lines=3
+                )
+                
+                with gr.Row():
+                    approve_btn = gr.Button("✅ Approve & Continue", variant="primary", visible=False)
+                    modify_btn = gr.Button("🔄 Modify Criteria", variant="secondary", visible=False)
+                    cancel_btn = gr.Button("❌ Cancel Analysis", visible=False)
 
     # Wire events
-    summary_btn.click(fn=generate_patient_summary, inputs=[pdf_upload], outputs=[summary_output])
-    hedis_btn.click(fn=generate_hedis_analysis, inputs=[pdf_upload, hedis_measure, measurement_year], outputs=[hedis_output])
+    summary_btn.click(
+        fn=generate_patient_summary,
+        inputs=[pdf_upload],
+        outputs=[summary_output]
+    )
+    
+    start_hedis_btn.click(
+        fn=start_hedis_analysis,
+        inputs=[pdf_upload, hedis_measure, measurement_year],
+        outputs=[hedis_output, suggestions_input, approve_btn, modify_btn, cancel_btn]
+    )
+    
+    approve_btn.click(
+        fn=approve_criteria,
+        inputs=[],
+        outputs=[hedis_output, suggestions_input, approve_btn, modify_btn, cancel_btn]
+    )
+    
+    modify_btn.click(
+        fn=modify_criteria,
+        inputs=[suggestions_input],
+        outputs=[hedis_output, suggestions_input, approve_btn, modify_btn, cancel_btn]
+    )
+    
+    cancel_btn.click(
+        fn=cancel_analysis,
+        inputs=[],
+        outputs=[hedis_output, suggestions_input, approve_btn, modify_btn, cancel_btn]
+    )
 
 # Spaces auto-runs the script; these hints are fine, too:
 if __name__ == "__main__":