Update app.py

app.py

@@ -1,9 +1,6 @@
-# app.py - Interactive Version with HEDIS Feedback
-# app.py - Interactive Version with HEDIS Feedback
 import os
 import tempfile
 import hashlib
-import json
 import gradio as gr
 
 from embedding_manager import EmbeddingManager
@@ -18,11 +15,6 @@ 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()
 
@@ -58,112 +50,6 @@ def _get_vectordb_from_bytes(pdf_bytes: bytes):
         except Exception:
             pass
 
-def clean_json_response(response_text: str) -> dict:
-    """Clean and parse JSON response from agents."""
-    if not response_text:
-        return {"raw_output": "Empty response", "parse_error": "No content received"}
-        
-    try:
-        # Clean the response string
-        cleaned = str(response_text).strip()
-        
-        # Remove markdown code blocks if present
-        if cleaned.startswith("```json"):
-            cleaned = cleaned[7:]
-        elif cleaned.startswith("```"):
-            cleaned = cleaned[3:]
-        
-        if cleaned.endswith("```"):
-            cleaned = cleaned[:-3]
-        
-        # Remove any leading/trailing whitespace and newlines
-        cleaned = cleaned.strip()
-        
-        # Handle quoted JSON strings
-        if len(cleaned) >= 2 and cleaned.startswith('"') and cleaned.endswith('"'):
-            try:
-                # Try to unquote and parse
-                unquoted = cleaned[1:-1]
-                # Handle common escape sequences
-                unquoted = unquoted.replace('\\"', '"').replace('\\n', '\n').replace('\\t', '\t')
-                test_parse = json.loads(unquoted)
-                cleaned = unquoted
-            except:
-                # If unquoting fails, keep original
-                pass
-        
-        # Try to parse as JSON
-        return json.loads(cleaned)
-        
-    except json.JSONDecodeError as e:
-        print(f"JSON parsing error: {e}")
-        print(f"Original response: {repr(response_text)}")
-        return {"raw_output": str(response_text), "parse_error": f"JSON Parse Error: {str(e)}"}
-    """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:
@@ -179,219 +65,23 @@ def generate_patient_summary(pdf_file):
     except Exception as e:
         return f"❌ Error processing chart: {e}"
 
-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
-    
+def generate_hedis_analysis(pdf_file, measure_name, measurement_year):
     try:
         if pdf_file is None:
-            return (
-                "⚠️ Please upload a PDF file first.",
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False)
-            )
+            return "⚠️ Please upload a PDF file first."
         if not measure_name:
-            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)
-            )
+            return "⚠️ Please enter a HEDIS measure code (e.g., COL, BCS, CCS)."
         if not measurement_year:
-            return (
-                "⚠️ Please enter a measurement year.",
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False)
-            )
+            return "⚠️ Please enter a measurement year."
 
         # Reuse embeddings if already built for this file
         vectordb = _get_vectordb_from_bytes(pdf_file)
 
-        # Create engine and get initial criteria
-        _CURRENT_ENGINE = HedisComplianceEngine(vectordb, measure_name, int(measurement_year))
-        
-        try:
-            # Get initial criteria with better error handling
-            raw_criteria = _CURRENT_ENGINE.get_hedis_criteria_with_feedback()
-            
-            # Handle different response types
-            if isinstance(raw_criteria, dict):
-                if "raw_output" in raw_criteria:
-                    # Try to clean and re-parse if it's raw output
-                    _CURRENT_CRITERIA = clean_json_response(raw_criteria["raw_output"])
-                else:
-                    _CURRENT_CRITERIA = raw_criteria
-            else:
-                # If it's not a dict, treat as raw output
-                _CURRENT_CRITERIA = clean_json_response(str(raw_criteria))
-            
-            _ANALYSIS_STATE = "criteria_review"
-            
-        except Exception as e:
-            print(f"Error getting criteria: {e}")
-            return (
-                f"❌ Error getting HEDIS criteria: {str(e)}\n\nPlease try again or check the measure code.",
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False),
-                gr.update(visible=False)
-            )
-        
-        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:
-        _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)
-            )
-        
-        try:
-            # Get modified criteria with better error handling
-            raw_criteria = _CURRENT_ENGINE.get_hedis_criteria_with_feedback(suggestions.strip())
-            
-            # Handle different response types
-            if isinstance(raw_criteria, dict):
-                if "raw_output" in raw_criteria:
-                    # Try to clean and re-parse if it's raw output
-                    _CURRENT_CRITERIA = clean_json_response(raw_criteria["raw_output"])
-                else:
-                    _CURRENT_CRITERIA = raw_criteria
-            else:
-                # If it's not a dict, treat as raw output
-                _CURRENT_CRITERIA = clean_json_response(str(raw_criteria))
-            
-            criteria_display = format_criteria_for_display(_CURRENT_CRITERIA)
-            
-        except Exception as e:
-            print(f"Error modifying criteria: {e}")
-            return (
-                f"❌ Error modifying criteria: {str(e)}\n\nPlease try again with different suggestions.",
-                gr.update(visible=True, value=""),
-                gr.update(visible=True),
-                gr.update(visible=True),
-                gr.update(visible=True)
-            )
-        
-        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)
-        )
+        hedis = HedisComplianceEngine(vectordb, measure_name, int(measurement_year))
+        result = hedis.run()
+        return result
     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 toggle_analysis_mode(mode):
-    """Toggle between interactive and standard analysis modes."""
-    if mode == "Interactive (with criteria review)":
-        return (
-            gr.update(visible=True),   # start_hedis_btn
-            gr.update(visible=False),  # run_standard_btn
-            "Select analysis mode and enter measure details, then click 'Start Interactive Analysis' to begin criteria review."
-        )
-    else:
-        return (
-            gr.update(visible=False),  # start_hedis_btn
-            gr.update(visible=True),   # run_standard_btn
-            "Select analysis mode and enter measure details, then click 'Run Standard Analysis' for direct 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)
-    )
+        return f"❌ Error processing HEDIS analysis: {e}"
 
 # --- Gradio Theme ---
 simple_theme = gr.themes.Soft(
@@ -409,7 +99,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; Interactive HEDIS Compliance</p>")
+    gr.HTML("<p style='text-align:center;color:#64748b;'>Patient Chart Analysis &amp; HEDIS Compliance</p>")
 
     with gr.Row(equal_height=True):
         # Left: Patient Summary
@@ -423,26 +113,9 @@ with gr.Blocks(theme=simple_theme, title="ChartWise AI") as interface:
                 height=400
             )
 
-def generate_hedis_analysis_original(pdf_file, measure_name, measurement_year):
-    """Original HEDIS analysis without interactive feedback."""
-    try:
-        if pdf_file is None:
-            return "⚠️ Please upload a PDF file first."
-        if not measure_name:
-            return "⚠️ Please enter a HEDIS measure code (e.g., COL, BCS, CCS)."
-        if not measurement_year:
-            return "⚠️ Please enter a measurement year."
-
-        # 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()  # Use original non-interactive run method
-        return result
-    except Exception as e:
-        return f"❌ Error processing HEDIS analysis: {e}"
+        # Right: HEDIS
         with gr.Column():
-            gr.HTML("<h3 style='color:#1e40af;'>🎯 Interactive HEDIS Analysis</h3>")
+            gr.HTML("<h3 style='color:#1e40af;'>🎯 HEDIS Measure Analysis</h3>")
             hedis_measure = gr.Textbox(
                 label="HEDIS Measure Code",
                 placeholder="e.g., COL, BCS, CCS, AAB",
@@ -454,74 +127,18 @@ def generate_hedis_analysis_original(pdf_file, measure_name, measurement_year):
                 precision=0,
                 info="e.g., 2024"
             )
-            start_hedis_btn = gr.Button("🎯 Start HEDIS Analysis", variant="secondary")
-            
+            hedis_btn = gr.Button("🎯 Run HEDIS Analysis", variant="secondary")
             hedis_output = gr.Markdown(
                 label="HEDIS Analysis Results",
-                value="Enter measure and year, then click 'Start HEDIS Analysis' to begin interactive review.",
-                height=300
+                value="Enter measure and year, then click 'Run HEDIS Analysis'.",
+                height=400
             )
-            
-            # 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]
-    )
-    
-    # Mode toggle
-    analysis_mode.change(
-        fn=toggle_analysis_mode,
-        inputs=[analysis_mode],
-        outputs=[start_hedis_btn, run_standard_btn, hedis_output]
-    )
-    
-    # Interactive analysis
-    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]
-    )
-    
-    # Standard analysis
-    run_standard_btn.click(
-        fn=generate_hedis_analysis_original,
-        inputs=[pdf_upload, hedis_measure, measurement_year],
-        outputs=[hedis_output]
-    )
-    
-    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]
-    )
+    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])
 
 # Spaces auto-runs the script; these hints are fine, too:
 if __name__ == "__main__":
-    interface.queue().launch(server_name="0.0.0.0", server_port=int(os.environ.get("GRADIO_SERVER_PORT", "7860")))
+    interface.queue().launch(server_name="0.0.0.0", server_port=int(os.environ.get("GRADIO_SERVER_PORT", "7860")))
+