from flask import Flask, render_template, request, send_file, jsonify, Response, redirect, url_for
import os
import json
import logging
from utils.callbackmanager import CallbackManager
from utils.generators import (generate_pdf_from_form, generate_pdf_from_meldrx,
                              analyze_dicom_file_with_ai, analyze_hl7_file_with_ai,
                              analyze_cda_xml_file_with_ai, analyze_pdf_file_with_ai,
                              analyze_csv_file_with_ai)
from utils.oneclick import generate_discharge_paper_one_click
from utils.meldrx import MeldRxAPI

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

app = Flask(__name__)

# Set SPACE_URL from environment variable or default to the correct Space URL
SPACE_URL = os.getenv("SPACE_URL", "https://multitransformer-tonic-discharge-guard.hf.space")
REDIRECT_URI = f"{SPACE_URL}/auth/callback"
CALLBACK_MANAGER = CallbackManager(
    redirect_uri=REDIRECT_URI,
    client_secret=None,
)

# Mock display_form function for HTML rendering
def display_form(first_name, last_name, middle_initial, dob, age, sex, address, city, state, zip_code,
                 doctor_first_name, doctor_last_name, doctor_middle_initial, hospital_name, doctor_address,
                 doctor_city, doctor_state, doctor_zip, admission_date, referral_source, admission_method,
                 discharge_date, discharge_reason, date_of_death, diagnosis, procedures, medications,
                 preparer_name, preparer_job_title):
    return f"""
    <div style='color:#00FFFF; font-family: monospace;'>
    <strong>Patient Discharge Form</strong><br>
    - Name: {first_name} {middle_initial} {last_name}<br>
    - Date of Birth: {dob}, Age: {age}, Sex: {sex}<br>
    - Address: {address}, {city}, {state}, {zip_code}<br>
    - Doctor: {doctor_first_name} {doctor_middle_initial} {doctor_last_name}<br>
    - Hospital/Clinic: {hospital_name}<br>
    - Doctor Address: {doctor_address}, {doctor_city}, {doctor_state}, {doctor_zip}<br>
    - Admission Date: {admission_date}, Source: {referral_source}, Method: {admission_method}<br>
    - Discharge Date: {discharge_date}, Reason: {discharge_reason}<br>
    - Date of Death: {date_of_death}<br>
    - Diagnosis: {diagnosis}<br>
    - Procedures: {procedures}<br>
    - Medications: {medications}<br>
    - Prepared By: {preparer_name}, {preparer_job_title}
    </div>
    """

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/auth', methods=['GET', 'POST'])
def auth():
    auth_url = CALLBACK_MANAGER.get_auth_url()
    if request.method == 'POST':
        redirected_url = request.form.get('redirected_url')
        if redirected_url:
            auth_code, error = CALLBACK_MANAGER.handle_callback(redirected_url)
            result = CALLBACK_MANAGER.set_auth_code(auth_code) if auth_code else f"<span style='color:#FF4500;'>{error}</span>"
            return render_template('auth.html', auth_url=auth_url, auth_result=result)
        auth_code = request.form.get('auth_code')
        if auth_code:
            result = CALLBACK_MANAGER.set_auth_code(auth_code)
            return render_template('auth.html', auth_url=auth_url, auth_result=result)
    return render_template('auth.html', auth_url=auth_url)

@app.route('/auth/callback', methods=['GET'])
def auth_callback():
    redirected_url = request.url
    result = CALLBACK_MANAGER.handle_callback(redirected_url)
    if "Authentication successful" in result:
        return redirect(url_for('dashboard'))
    return render_template('auth.html', auth_url=CALLBACK_MANAGER.get_auth_url(), auth_result=result)

@app.route('/dashboard', methods=['GET'])
def dashboard():
    if not CALLBACK_MANAGER.access_token:
        return render_template('auth.html', auth_url=CALLBACK_MANAGER.get_auth_url(), auth_result="<span style='color:#FF8C00;'>Please authenticate first.</span>")
    
    data = CALLBACK_MANAGER.get_patient_data()
    if data.startswith('<span'):
        return render_template('dashboard.html', error=data)
    patients_data = json.loads(data)
    patients = [entry['resource'] for entry in patients_data.get('entry', []) if entry['resource'].get('resourceType') == 'Patient']
    return render_template('dashboard.html', patients=patients, authenticated=True)

@app.route('/auth/patient-data', methods=['GET'])
def patient_data():
    data = CALLBACK_MANAGER.get_patient_data()
    return jsonify(json.loads(data) if data and not data.startswith('<span') else {"error": data})

@app.route('/auth/pdf', methods=['GET'])
def generate_meldrx_pdf():
    patient_data = CALLBACK_MANAGER.get_patient_data()
    pdf_path = generate_pdf_from_meldrx(patient_data)
    return send_file(pdf_path, as_attachment=True, download_name="meldrx_patient_data.pdf")

# @app.route('/dashboard', methods=['GET'])
# def dashboard():
#     data = CALLBACK_MANAGER.get_patient_data()
#     if data.startswith('<span'):  # Indicates an error or unauthenticated state
#         return render_template('dashboard.html', error=data)
#     patients_data = json.loads(data)
#     patients = [entry['resource'] for entry in patients_data.get('entry', []) if entry['resource'].get('resourceType') == 'Patient']
#     return render_template('dashboard.html', patients=patients, authenticated=True)

@app.route('/form', methods=['GET', 'POST'])
def discharge_form():
    if request.method == 'POST':
        form_data = request.form.to_dict()
        if 'display' in request.form:
            html_form = display_form(**form_data)
            return render_template('form.html', form_output=html_form)
        elif 'generate_pdf' in request.form:
            pdf_path = generate_pdf_from_form(**form_data)
            return send_file(pdf_path, as_attachment=True, download_name="discharge_form.pdf")
    return render_template('form.html')

@app.route('/analysis', methods=['GET', 'POST'])
def file_analysis():
    if request.method == 'POST':
        file = request.files.get('file')
        file_type = request.form.get('file_type')
        if file:
            file_path = os.path.join('/tmp', file.filename)
            file.save(file_path)
            if file_type == 'dicom':
                result = analyze_dicom_file_with_ai(file_path)
            elif file_type == 'hl7':
                result = analyze_hl7_file_with_ai(file_path)
            elif file_type == 'xml' or file_type == 'ccda' or file_type == 'ccd':
                result = analyze_cda_xml_file_with_ai(file_path)
            elif file_type == 'pdf':
                result = analyze_pdf_file_with_ai(file_path)
            elif file_type == 'csv':
                result = analyze_csv_file_with_ai(file_path)
            else:
                result = "Unsupported file type"
            os.remove(file_path)
            return render_template('analysis.html', result=result, file_type=file_type)
    return render_template('analysis.html')

@app.route('/oneclick', methods=['GET', 'POST'])
def one_click():
    if request.method == 'POST':
        patient_id = request.form.get('patient_id', '')
        first_name = request.form.get('first_name', '')
        last_name = request.form.get('last_name', '')
        pdf_path, status = generate_discharge_paper_one_click(CALLBACK_MANAGER.api, patient_id, first_name, last_name)
        if pdf_path:
            return send_file(pdf_path, as_attachment=True, download_name="discharge_summary.pdf")
        return render_template('oneclick.html', status=status)
    return render_template('oneclick.html')

if __name__ == '__main__':
    port = int(os.getenv("PORT", 7860))  # Default to 5000, override with PORT env var
    app.run(debug=False, host='0.0.0.0', port=port)