import gradio as gr
import requests
import json
import os

# ==================== API Functions ====================

def analyze_provider_notes(provider_notes, api_url, progress=gr.Progress()):
    """Send provider notes to backend API for analysis"""
    if not provider_notes or not provider_notes.strip():
        return generate_empty_response()
    
    progress(0, desc="Starting analysis...")
    
    try:
        payload = {"provider_notes": provider_notes}
        
        progress(0.3, desc="Sending request to API...")
        
        response = requests.post(
            f"{api_url}/api/v1/analyze",
            json=payload,
            headers={"Content-Type": "application/json"},
            timeout=60
        )
        
        progress(0.6, desc="Processing response...")
        
        response.raise_for_status()
        result = response.json()
        
        progress(0.9, desc="Formatting results...")
        
        formatted_response = format_complete_response(result)
        
        progress(1.0, desc="Complete!")
        
        return formatted_response
        
    except requests.exceptions.Timeout:
        return "❌ **Request Timeout**\n\nThe API is taking too long to respond. Please try again."
    except requests.exceptions.HTTPError as e:
        return f"❌ **HTTP Error {e.response.status_code}**\n\n{e.response.text}"
    except requests.exceptions.RequestException as e:
        return f"❌ **Request Error**\n\n{str(e)}"
    except Exception as e:
        return f"❌ **Unexpected Error**\n\n{str(e)}"

def generate_empty_response():
    """Generate empty response message"""
    return """
### 💬 Welcome to Medical Coding (ICD-10) Assistant

Please enter provider notes in the text box below and click **Send** to analyze.

**Tips:**
- Provide detailed clinical documentation
- Include symptoms, diagnoses, and procedures
- Be specific about treatments and prescriptions

**How to use:**
1. Select an example from the sidebar, or
2. Upload documents via the Upload button
3. Type your own clinical notes below
4. Click the **Send** button
5. Review the ICD-10 and CPT codes generated
"""

def format_complete_response(result):
    """Format complete analysis response"""
    output = "---\n\n"
    
    if result.get("overall_summary"):
        output += f"### 📋 Overall Summary\n\n{result['overall_summary']}\n\n---\n\n"
    
    output += "### 🏥 ICD-10 Diagnostic Codes\n\n"
    icd_codes = result.get("icd_codes", [])
    
    if icd_codes:
        for idx, icd in enumerate(icd_codes, 1):
            output += f"**{idx}. {icd.get('code', 'N/A')}** - {icd.get('description', 'N/A')}\n\n"
            output += f"*Explanation:* {icd.get('explanation', 'N/A')}\n\n"
    else:
        output += "*No ICD-10 codes identified*\n\n"
    
    output += "---\n\n"
    
    output += "### 💼 CPT Procedure Codes\n\n"
    cpt_codes = result.get("cpt_codes", [])
    
    if cpt_codes:
        for idx, cpt in enumerate(cpt_codes, 1):
            output += f"**{idx}. {cpt.get('code', 'N/A')}** - {cpt.get('description', 'N/A')}\n\n"
            output += f"*Explanation:* {cpt.get('explanation', 'N/A')}\n\n"
    else:
        output += "*No CPT codes identified*\n\n"
    
    return output

# ==================== Example Notes ====================

EXAMPLES = {
    "Acute Bronchitis": """Patient presents with acute bronchitis. Cough for 5 days, productive with yellow sputum. Lung exam reveals diffuse wheezing. Prescribed azithromycin 500mg.""",
    
    "Type 2 Diabetes": """Patient with type 2 diabetes mellitus, uncontrolled. HbA1c 9.2%. Discussed diet and medication compliance. Adjusted insulin dosing. Referred to diabetes educator.""",
    
    "Hypertension Follow-up": """Follow-up visit for hypertension. Blood pressure 145/92. Patient reports good medication compliance. Continue current antihypertensive regimen. Return in 3 months.""",
}

def load_example(example_name):
    """Load example provider notes"""
    return EXAMPLES.get(example_name, "")

# ==================== Load External CSS ====================

def load_css():
    """Load external CSS file"""
    css_path = os.path.join(os.path.dirname(__file__), "assets", "styles.css")
    try:
        with open(css_path, "r", encoding="utf-8") as f:
            return f.read()
    except FileNotFoundError:
        print("Warning: styles.css not found")
        return ""

# ==================== Gradio Interface ====================

with gr.Blocks(css=load_css(), theme=gr.themes.Soft(), title="Medical Coding (ICD-10) Assistant") as demo:
    
    api_url_state = gr.State(value="https://Distopia22-icd-cpt-coding-api.hf.space")
    
    # Header
    gr.HTML("""
        <div class="header-container">
            <h1>🏥 Medical Coding (ICD-10) Assistant</h1>
            <p>AI-Powered Medical Coding Analysis using Groq LLaMA 3.3 70B</p>
        </div>
    """)
    
    with gr.Row(elem_classes="main-container"):
        
        # Left Sidebar
        with gr.Column(scale=1, min_width=280, elem_classes="sidebar"):
            gr.HTML('<div class="sidebar-title">📚 Example Cases</div>')
            
            example_buttons = {}
            for example_name in EXAMPLES.keys():
                btn = gr.Button(
                    f"📄 {example_name}",
                    elem_classes="example-btn"
                )
                example_buttons[example_name] = btn
            
            gr.Markdown("---")
            
            # Upload Document Section - Redirects to API Docs
            gr.HTML('<div class="sidebar-title">📤 Upload Document</div>')
            
            gr.HTML("""
                <a href="https://distopia22-icd-cpt-coding-api-backend.hf.space/docs" target="_blank" style="text-decoration: none;">
                    <button class="upload-redirect-button">
                        📁 Upload via API Docs
                    </button>
                </a>
                <div style="background: rgba(255,255,255,0.1); padding: 10px; border-radius: 8px; margin-top: 10px;">
                    <p style="color: #e0f7f4; font-size: 0.8rem; margin: 5px 0; text-align: center;">
                        📝 Only .txt format allowed
                    </p>
                    <p style="color: #ffffff; font-size: 0.75rem; margin: 5px 0; text-align: center; opacity: 0.7;">
                        Click button to upload files
                    </p>
                </div>
            """)
        
        # Right Chat Area
        with gr.Column(scale=3, elem_classes="chat-container"):
            
            output_area = gr.Markdown(
                value=generate_empty_response(),
                elem_classes="output-area",
                label=""
            )
            
            with gr.Group(elem_classes="input-container"):
                with gr.Row():
                    provider_notes_input = gr.Textbox(
                        label="",
                        placeholder="Enter clinical provider notes here...",
                        lines=3,
                        max_lines=5,
                        show_label=False,
                        elem_classes="prompt-input"
                    )
                
                with gr.Row():
                    send_btn = gr.Button("🚀 Send", elem_classes="send-button", size="sm")
                    clear_btn = gr.Button("🗑️ Clear", elem_classes="clear-button", size="sm")
    
    # Footer
    gr.HTML("""
        <div class="footer">
            <p>Powered by <strong>Groq LLaMA 3.3 70B</strong> | <strong>FastAPI</strong> | <strong>Gradio</strong></p>
            <p>© 2025 Medical Coding Assistant - Secure & HIPAA Compliant</p>
        </div>
    """)
    
    # ==================== Event Handlers ====================
    
    # Example button clicks
    for example_name, btn in example_buttons.items():
        btn.click(
            fn=lambda name=example_name: load_example(name),
            outputs=[provider_notes_input]
        )
    
    # Send button - Analyze notes
    send_btn.click(
        fn=analyze_provider_notes,
        inputs=[provider_notes_input, api_url_state],
        outputs=[output_area]
    )
    
    # Clear button
    clear_btn.click(
        fn=lambda: ("", generate_empty_response()),
        outputs=[provider_notes_input, output_area]
    )

# Launch
if __name__ == "__main__":
    demo.launch(
        server_name="0.0.0.0",
        server_port=7860,
        show_error=True
    )