app.py

import streamlit as st
import pandas as pd
import numpy as np

# Seed for reproducibility
np.random.seed(42)

# Function to generate synthetic BreastCancer data
def generate_breast_cancer_data(num_patients):
    primary_keys = [f"PPK_{i+1:05d}" for i in range(num_patients)]
    
    ages = []
    menopausal_status = []
    tumor_sizes = []
    lymph_nodes = []
    grades = []
    stages = []
    er_status = []
    pr_status = []
    her2_status = []
    ki67_level = []
    tnbc_status = []
    brca_mutation = []
    overall_health = []
    genomic_score = []
    treatment = []

    for i in range(num_patients):
        age = int(np.random.normal(60, 10))
        age = max(30, min(age, 80))
        ages.append(age)

        menopausal = "Post-menopausal" if age >= 50 else "Pre-menopausal"
        menopausal_status.append(menopausal)

        tumor_size = round(np.random.lognormal(mean=0.7, sigma=0.5), 2)
        tumor_sizes.append(tumor_size)

        lymph_node = (
            "Positive"
            if (tumor_size > 2.0 and np.random.rand() < 0.6)
            or (tumor_size <= 2.0 and np.random.rand() < 0.3)
            else "Negative"
        )
        lymph_nodes.append(lymph_node)

        grade = np.random.choice([1, 2, 3], p=[0.1, 0.4, 0.5] if tumor_size > 2.0 else [0.3, 0.5, 0.2])
        grades.append(grade)

        if tumor_size <= 2.0 and lymph_node == "Negative":
            stage = "I"
        elif (tumor_size > 2.0 and tumor_size <= 5.0) and lymph_node == "Negative":
            stage = "II"
        elif lymph_node == "Positive" or tumor_size > 5.0:
            stage = "III"
        else:
            stage = "II"
        if np.random.rand() < 0.05:
            stage = "IV"
        stages.append(stage)

        er = np.random.choice(["Positive", "Negative"], p=[0.75, 0.25])
        pr = "Positive" if er == "Positive" and np.random.rand() > 0.1 else "Negative"
        er_status.append(er)
        pr_status.append(pr)

        her2 = np.random.choice(["Positive", "Negative"], p=[0.3, 0.7] if grade == 3 else [0.15, 0.85])
        her2_status.append(her2)

        ki67 = "High" if grade == 3 and np.random.rand() < 0.8 else "Low"
        ki67_level.append(ki67)

        tnbc = "Positive" if er == "Negative" and pr == "Negative" and her2 == "Negative" else "Negative"
        tnbc_status.append(tnbc)

        brca = "Positive" if (tnbc == "Positive" or age < 40) and np.random.rand() < 0.2 else "Negative"
        brca_mutation.append(brca)

        health = "Good" if age < 65 and np.random.rand() < 0.9 else "Poor"
        overall_health.append(health)

        recurrence_score = (
            np.random.choice(["Low", "Intermediate", "High"], p=[0.6, 0.3, 0.1])
            if er == "Positive" and her2 == "Negative"
            else "N/A"
        )
        genomic_score.append(recurrence_score)

        if stage in ["I", "II"]:
            if tnbc == "Positive":
                treat = "Surgery, Chemotherapy, and Radiation Therapy"
            elif er == "Positive" and recurrence_score != "N/A":
                if recurrence_score == "High":
                    treat = "Surgery, Chemotherapy, Hormone Therapy, and Radiation Therapy"
                elif recurrence_score == "Intermediate":
                    treat = "Surgery, Consider Chemotherapy, Hormone Therapy, and Radiation Therapy"
                else:
                    treat = "Surgery, Hormone Therapy, and Radiation Therapy"
            elif her2 == "Positive":
                treat = "Surgery, HER2-Targeted Therapy, Chemotherapy, and Radiation Therapy"
            else:
                treat = "Surgery, Chemotherapy, and Radiation Therapy"
        elif stage == "III":
            treat = (
                "Neoadjuvant Chemotherapy, Surgery, Radiation Therapy"
                + (", HER2-Targeted Therapy" if her2 == "Positive" else "")
                + (", Hormone Therapy" if er == "Positive" else "")
            )
        else:
            treat = "Systemic Therapy (Palliative Care)"
        treatment.append(treat)

    breast_cancer_data = {
        "PRIMARY_PERSON_KEY": primary_keys,
        "Age": ages,
        "Menopausal Status": menopausal_status,
        "Tumor Size (cm)": tumor_sizes,
        "Lymph Node Involvement": lymph_nodes,
        "Tumor Grade": grades,
        "Tumor Stage": stages,
        "ER Status": er_status,
        "PR Status": pr_status,
        "HER2 Status": her2_status,
        "Ki-67 Level": ki67_level,
        "TNBC Status": tnbc_status,
        "BRCA Mutation": brca_mutation,
        "Overall Health": overall_health,
        "Genomic Recurrence Score": genomic_score,
        "Treatment": treatment,
    }

    return pd.DataFrame(breast_cancer_data)

# Function to generate Members from BreastCancer
def generate_members_from_breast_cancer(breast_cancer_df):
    return pd.DataFrame({
        "MEMBER_ID": breast_cancer_df["PRIMARY_PERSON_KEY"],
        "PRIMARY_PERSON_KEY": breast_cancer_df["PRIMARY_PERSON_KEY"],
        "MEM_GENDER": ["F"] * len(breast_cancer_df),
        "MEM_ETHNICITY": np.random.choice(["Hispanic", "Non-Hispanic", None], len(breast_cancer_df)),
        "MEM_RACE": np.random.choice(["White", "Black", "Asian", None], len(breast_cancer_df)),
        "MEM_STATE": np.random.choice(["MI", "HI", "CA"], len(breast_cancer_df)),
        "MEM_ZIP3": np.random.randint(100, 999, len(breast_cancer_df)),
    })

# Function to generate Enrollments from BreastCancer
def generate_enrollments_from_breast_cancer(breast_cancer_df):
    return pd.DataFrame({
        "PRIMARY_PERSON_KEY": breast_cancer_df["PRIMARY_PERSON_KEY"],
        "MEM_STAT": np.random.choice(["ACTIVE", "INACTIVE"], len(breast_cancer_df)),
        "PAYER_LOB": np.random.choice(["MEDICAID", "COMMERCIAL", "MEDICARE"], len(breast_cancer_df)),
        "PAYER_TYPE": np.random.choice(["PPO", "HMO"], len(breast_cancer_df)),
        "RELATION": np.random.choice(["SUBSCRIBER", "DEPENDENT"], len(breast_cancer_df)),
    })

# Function to generate Services from BreastCancer
def generate_services(num_services, primary_keys):
    return pd.DataFrame({
        "PRIMARY_PERSON_KEY": np.random.choice(primary_keys, num_services),
        "SERVICE_SETTING": np.random.choice(["OUTPATIENT", "INPATIENT"], num_services),
        "PROC_CODE": np.random.randint(1000, 9999, num_services),
        "SERVICE_DATE": pd.date_range(start="2023-01-01", periods=num_services).to_numpy(),
        "AMOUNT_BILLED": np.random.uniform(500, 15000, num_services),
        "AMOUNT_PAID": np.random.uniform(500, 15000, num_services),
        "CLAIM_STATUS": np.random.choice(["PAID", "DENIED", "PENDING"], num_services),
        "RELATION": np.random.choice(["SUBSCRIBER", "DEPENDENT"], num_services),
    })

# Main Streamlit App
st.title("Synthetic Medical Data Generator")

# Sliders
num_patients = st.slider("Number of Breast Cancer Patients to Generate", 10, 1000, 100)
num_services = st.slider("Number of Services to Generate", 10, 2000, 500)

if st.button("Generate Data"):
    breast_cancer_df = generate_breast_cancer_data(num_patients)
    members_df = generate_members_from_breast_cancer(breast_cancer_df)
    enrollments_df = generate_enrollments_from_breast_cancer(breast_cancer_df)
    services_df = generate_services(num_services, breast_cancer_df["PRIMARY_PERSON_KEY"].tolist())

    # Display and download data
    st.subheader("Breast Cancer Data")
    st.dataframe(breast_cancer_df.head())
    st.download_button("Download Breast Cancer Data", breast_cancer_df.to_csv(index=False), "breast_cancer.csv")

    st.subheader("Members Data")
    st.dataframe(members_df.head())
    st.download_button("Download Members Data", members_df.to_csv(index=False), "members.csv")

    st.subheader("Enrollments Data")
    st.dataframe(enrollments_df.head())
    st.download_button("Download Enrollments Data", enrollments_df.to_csv(index=False), "enrollments.csv")

    st.subheader("Services Data")
    st.dataframe(services_df.head())
    st.download_button("Download Services Data", services_df.to_csv(index=False), "services.csv")