Update app.py

app.py

@@ -1,112 +1,804 @@
 import gradio as gr
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.enums import TA_CENTER
 import datetime
 
-def decision_logic(bladder, saddle, deficit_prog, myelopathy,
-                   cancer, fever, immuno,
-                   trauma,
-                   radiculopathy,
-                   duration,
-                   no_improvement,
-                   post_op,
-                   impact):
 
-    if bladder or saddle or deficit_prog or myelopathy:
-        return "🔴 MR PILNE", "Objawy alarmowe neurologiczne (CES/mielopatia)"
+AUTHOR_NAME = "prof. dr hab. n. med. Rafał Obuchowicz"
+AUTHOR_AFFILIATION = "Luxmed Diagnostyka"
 
-    if cancer or fever or immuno or post_op:
-        return "🟠 MR PRZYSPIESZONE", "Podejrzenie nowotworu / infekcji / stan pooperacyjny"
 
-    if radiculopathy and duration == ">6 tygodni" and no_improvement and impact:
-        return "🟡 MR PLANOWE", "Radikulopatia >6 tygodni bez poprawy"
+def yes_no(value: bool) -> str:
+    return "Tak" if value else "Nie"
 
-    if trauma:
-        return "🔵 RTG/TK NAJPIERW", "Podejrzenie urazu"
 
-    return "🟢 BRAK WSKAZAŃ", "Leczenie zachowawcze"
+def validate_form(
+    first_episode, recurrent_episode,
+    no_deficit, root_deficit, multilevel_bilateral, cord_signs,
+    conservative_none, conservative_lt6, conservative_ge6,
+    improved, no_improvement, worsening
+):
+    warnings = []
 
+    if first_episode and recurrent_episode:
+        warnings.append("Zaznaczono jednocześnie: 'Pierwszy epizod' i 'Nawrót / zaostrzenie przewlekłe'.")
 
-def generate_pdf(rec, just):
-    filename = "raport.pdf"
-    doc = SimpleDocTemplate(filename)
+    neuro_flags = [no_deficit, root_deficit, multilevel_bilateral, cord_signs]
+    if no_deficit and any([root_deficit, multilevel_bilateral, cord_signs]):
+        warnings.append("Zaznaczono jednocześnie 'Brak deficytu' i obecność deficytu neurologicznego.")
+
+    treatment_flags = [conservative_none, conservative_lt6, conservative_ge6]
+    if sum(treatment_flags) > 1:
+        warnings.append("W sekcji leczenia zachowawczego wybrano więcej niż jedną sprzeczną opcję czasową.")
+
+    response_flags = [improved, no_improvement, worsening]
+    if sum(response_flags) > 1:
+        warnings.append("W sekcji odpowiedzi na leczenie zaznaczono więcej niż jedną sprzeczną opcję.")
+
+    return warnings
+
+
+def build_recommendation(
+    spine_section,
+    duration,
+    first_episode,
+    recurrent_episode,
+    post_op,
+    axial_pain,
+    radicular_pain,
+    paresthesia,
+    weakness,
+    gait_balance,
+    sphincter,
+    saddle,
+    progressive_deficit,
+    myelopathy,
+    cancer_history,
+    weight_loss,
+    night_pain,
+    fever,
+    immunosuppression,
+    infection_suspicion,
+    trauma_high,
+    trauma_low_risk,
+    osteoporosis_steroids,
+    no_deficit,
+    root_deficit,
+    multilevel_bilateral,
+    cord_signs,
+    conservative_none,
+    conservative_lt6,
+    conservative_ge6,
+    improved,
+    no_improvement,
+    worsening,
+    suspect_disc,
+    suspect_cord,
+    suspect_tumor,
+    suspect_infection,
+    qualify_surgery,
+    qualify_injection,
+    treatment_impact,
+):
+    reasons = []
+    alternatives = []
+    urgency = "routine"
+    contrast = "nie"
+    imaging = "Brak wskazań do MR na tym etapie"
+    category = "🟢 BRAK WSKAZAŃ DO MR"
+    clinical_path = "Leczenie zachowawcze / obserwacja"
+
+    if sphincter or saddle or progressive_deficit or myelopathy or cord_signs:
+        category = "🔴 MR PILNE"
+        imaging = f"Pilne MR odcinka {spine_section}"
+        clinical_path = "Pilna konsultacja specjalistyczna / tryb pilny"
+        urgency = "pilne"
+        reasons.append("Stwierdzono objawy alarmowe neurologiczne")
+        if sphincter:
+            reasons.append("Zaburzenia zwieraczy")
+        if saddle:
+            reasons.append("Znieczulenie okolicy krocza")
+        if progressive_deficit:
+            reasons.append("Postępujący deficyt neurologiczny")
+        if myelopathy or cord_signs:
+            reasons.append("Objawy mielopatii / uszkodzenia rdzenia")
+
+        return {
+            "category": category,
+            "imaging": imaging,
+            "clinical_path": clinical_path,
+            "urgency": urgency,
+            "contrast": contrast,
+            "reasons": reasons,
+            "alternatives": alternatives,
+            "color": "#b91c1c",
+        }
+
+    if (
+        cancer_history
+        or weight_loss
+        or night_pain
+        or fever
+        or immunosuppression
+        or infection_suspicion
+        or suspect_tumor
+        or suspect_infection
+        or (post_op and (worsening or weakness or paresthesia or radicular_pain))
+    ):
+        category = "🟠 MR PRZYSPIESZONE"
+        imaging = f"MR odcinka {spine_section}"
+        clinical_path = "Przyspieszona diagnostyka specjalistyczna"
+        urgency = "przyspieszone"
+        contrast = "rozważyć"
+        reasons.append("Występują przesłanki nowotworowe, infekcyjne lub pooperacyjne")
+        if cancer_history:
+            reasons.append("Wywiad nowotworowy")
+        if weight_loss:
+            reasons.append("Niezamierzona utrata masy ciała")
+        if night_pain:
+            reasons.append("Ból nocny / niemechaniczny")
+        if fever:
+            reasons.append("Gorączka")
+        if immunosuppression:
+            reasons.append("Immunosupresja")
+        if infection_suspicion:
+            reasons.append("Podejrzenie infekcji")
+        if post_op:
+            reasons.append("Stan po operacji kręgosłupa")
+
+        alternatives.append("Rozważyć badania laboratoryjne: CRP, OB, morfologia")
+        if trauma_high or trauma_low_risk:
+            alternatives.append("Jeśli współistnieje uraz: rozważyć RTG/TK struktur kostnych")
+
+        return {
+            "category": category,
+            "imaging": imaging,
+            "clinical_path": clinical_path,
+            "urgency": urgency,
+            "contrast": contrast,
+            "reasons": reasons,
+            "alternatives": alternatives,
+            "color": "#ea580c",
+        }
+
+    if trauma_high or trauma_low_risk or osteoporosis_steroids:
+        category = "🔵 RTG / TK NAJPIERW"
+        imaging = f"RTG lub TK odcinka {spine_section}"
+        clinical_path = "Najpierw diagnostyka kostna; MR warunkowo"
+        urgency = "pilne" if trauma_high else "routine"
+        reasons.append("Dominują wskazania urazowe / kostne")
+        if trauma_high:
+            reasons.append("Uraz wysokiej energii")
+        if trauma_low_risk:
+            reasons.append("Uraz niskoenergetyczny z ryzykiem")
+        if osteoporosis_steroids:
+            reasons.append("Osteoporoza / przewlekła steroidoterapia")
+
+        alternatives.append("MR rozważyć przy deficycie neurologicznym lub podejrzeniu uszkodzeń tkanek miękkich")
+
+        return {
+            "category": category,
+            "imaging": imaging,
+            "clinical_path": clinical_path,
+            "urgency": urgency,
+            "contrast": contrast,
+            "reasons": reasons,
+            "alternatives": alternatives,
+            "color": "#2563eb",
+        }
+
+    radicular_pattern = radicular_pain or paresthesia or weakness or root_deficit or multilevel_bilateral
+
+    if (
+        radicular_pattern
+        and conservative_ge6
+        and (no_improvement or worsening)
+        and treatment_impact == "TAK"
+    ):
+        category = "🟡 MR PLANOWE"
+        imaging = f"MR odcinka {spine_section} bez kontrastu"
+        clinical_path = "Planowa diagnostyka obrazowa"
+        urgency = "planowe"
+        reasons.append("Objawy korzeniowe / neurologiczne utrzymują się mimo leczenia")
+        reasons.append("Leczenie zachowawcze ≥ 6 tygodni")
+        if no_improvement:
+            reasons.append("Brak poprawy")
+        if worsening:
+            reasons.append("Pogorszenie")
+        reasons.append("Wynik badania może wpłynąć na leczenie")
+
+        if qualify_surgery:
+            reasons.append("Pacjent potencjalnie kwalifikowany do zabiegu")
+        if qualify_injection:
+            reasons.append("Pacjent potencjalnie kwalifikowany do blokady / iniekcji")
+
+        return {
+            "category": category,
+            "imaging": imaging,
+            "clinical_path": clinical_path,
+            "urgency": urgency,
+            "contrast": contrast,
+            "reasons": reasons,
+            "alternatives": alternatives,
+            "color": "#ca8a04",
+        }
+
+    if (
+        axial_pain
+        and not radicular_pattern
+        and not weakness
+        and not gait_balance
+        and not conservative_ge6
+        and treatment_impact != "TAK"
+    ):
+        category = "🟢 BRAK WSKAZAŃ DO MR"
+        imaging = "Brak wskazań do MR na tym etapie"
+        clinical_path = "Leczenie zachowawcze / obserwacja / kontrola"
+        urgency = "routine"
+        reasons.append("Dominują objawy bólu osiowego bez czerwonych flag")
+        reasons.append("Brak przesłanek do pilnej diagnostyki obrazowej")
+        alternatives.append("Kontynuacja leczenia zachowawczego")
+        alternatives.append("Rozważyć fizjoterapię / farmakoterapię / kontrolę kliniczną")
+
+        return {
+            "category": category,
+            "imaging": imaging,
+            "clinical_path": clinical_path,
+            "urgency": urgency,
+            "contrast": contrast,
+            "reasons": reasons,
+            "alternatives": alternatives,
+            "color": "#15803d",
+        }
+
+    category = "🟡 WYMAGA DOPRECYZOWANIA"
+    imaging = "Rozważyć MR po uzupełnieniu oceny klinicznej"
+    clinical_path = "Uzupełnić wywiad / badanie / odpowiedź na leczenie"
+    urgency = "routine"
+    reasons.append("Dane nie są jednoznaczne dla automatycznej kwalifikacji")
+    if treatment_impact == "NIEPEWNE":
+        reasons.append("Niepewny wpływ wyniku MR na decyzję terapeutyczną")
+    alternatives.append("Zweryfikować czas leczenia zachowawczego")
+    alternatives.append("Uściślić deficyty neurologiczne i pytanie kliniczne")
+
+    return {
+        "category": category,
+        "imaging": imaging,
+        "clinical_path": clinical_path,
+        "urgency": urgency,
+        "contrast": contrast,
+        "reasons": reasons,
+        "alternatives": alternatives,
+        "color": "#a16207",
+    }
+
+
+def generate_pdf(data, result, warnings):
+    filename = "raport_mr_kregoslup_pro_plus.pdf"
+    doc = SimpleDocTemplate(filename, pagesize=A4, rightMargin=36, leftMargin=36, topMargin=36, bottomMargin=36)
     styles = getSampleStyleSheet()
-    content = []
 
-    content.append(Paragraph("Raport kwalifikacji do MR kręgosłupa", styles["Title"]))
-    content.append(Spacer(1, 12))
-    content.append(Paragraph(f"Data: {datetime.datetime.now()}", styles["Normal"]))
-    content.append(Spacer(1, 12))
-    content.append(Paragraph(f"Rekomendacja: {rec}", styles["Normal"]))
-    content.append(Spacer(1, 12))
-    content.append(Paragraph(f"Uzasadnienie: {just}", styles["Normal"]))
+    title_style = ParagraphStyle(
+        "CustomTitle",
+        parent=styles["Title"],
+        alignment=TA_CENTER,
+        fontSize=18,
+        spaceAfter=10
+    )
+
+    subtitle_style = ParagraphStyle(
+        "CustomSubtitle",
+        parent=styles["Normal"],
+        alignment=TA_CENTER,
+        fontSize=10,
+        textColor=colors.grey
+    )
+
+    story = []
+    story.append(Paragraph("Raport kwalifikacji do badania MR kręgosłupa", title_style))
+    story.append(Paragraph("Clinical Decision Support – PRO+", subtitle_style))
+    story.append(Spacer(1, 8))
+    story.append(Paragraph(f"Autor narzędzia: <b>{AUTHOR_NAME}</b>", styles["Normal"]))
+    story.append(Paragraph(f"Afiliacja: <b>{AUTHOR_AFFILIATION}</b>", styles["Normal"]))
+    story.append(Paragraph(f"Data wygenerowania: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}", styles["Normal"]))
+    story.append(Spacer(1, 14))
+
+    summary_table = Table(
+        [
+            ["Rekomendacja", result["category"]],
+            ["Sugerowane badanie", result["imaging"]],
+            ["Tryb", result["urgency"]],
+            ["Kontrast", result["contrast"]],
+            ["Ścieżka kliniczna", result["clinical_path"]],
+        ],
+        colWidths=[160, 350],
+    )
+    summary_table.setStyle(
+        TableStyle(
+            [
+                ("BACKGROUND", (0, 0), (0, -1), colors.HexColor("#EAEAEA")),
+                ("GRID", (0, 0), (-1, -1), 0.5, colors.grey),
+                ("VALIGN", (0, 0), (-1, -1), "TOP"),
+                ("FONTSIZE", (0, 0), (-1, -1), 10),
+                ("PADDING", (0, 0), (-1, -1), 6),
+            ]
+        )
+    )
+    story.append(summary_table)
+    story.append(Spacer(1, 14))
+
+    if warnings:
+        story.append(Paragraph("<b>Ostrzeżenia walidacyjne:</b>", styles["Heading3"]))
+        for w in warnings:
+            story.append(Paragraph(f"• {w}", styles["Normal"]))
+        story.append(Spacer(1, 10))
+
+    story.append(Paragraph("<b>Uzasadnienie:</b>", styles["Heading3"]))
+    for r in result["reasons"]:
+        story.append(Paragraph(f"• {r}", styles["Normal"]))
+    story.append(Spacer(1, 10))
+
+    if result["alternatives"]:
+        story.append(Paragraph("<b>Alternatywy / dalsze kroki:</b>", styles["Heading3"]))
+        for a in result["alternatives"]:
+            story.append(Paragraph(f"• {a}", styles["Normal"]))
+        story.append(Spacer(1, 10))
+
+    details = [
+        ["Odcinek", data["spine_section"]],
+        ["Czas objawów", data["duration"]],
+        ["Pierwszy epizod", data["first_episode"]],
+        ["Nawrót", data["recurrent_episode"]],
+        ["Stan po operacji", data["post_op"]],
+        ["Ból osiowy", data["axial_pain"]],
+        ["Ból promieniujący", data["radicular_pain"]],
+        ["Parestezje", data["paresthesia"]],
+        ["Osłabienie siły", data["weakness"]],
+        ["Zaburzenia chodu / równowagi", data["gait_balance"]],
+        ["Zaburzenia zwieraczy", data["sphincter"]],
+        ["Znieczulenie krocza", data["saddle"]],
+        ["Postępujący deficyt", data["progressive_deficit"]],
+        ["Objawy mielopatii", data["myelopathy"]],
+        ["Wywiad nowotworowy", data["cancer_history"]],
+        ["Utrata masy ciała", data["weight_loss"]],
+        ["Ból nocny", data["night_pain"]],
+        ["Gorączka", data["fever"]],
+        ["Immunosupresja", data["immunosuppression"]],
+        ["Podejrzenie infekcji", data["infection_suspicion"]],
+        ["Uraz wysokiej energii", data["trauma_high"]],
+        ["Uraz niskoenergetyczny + ryzyko", data["trauma_low_risk"]],
+        ["Osteoporoza / steroidy", data["osteoporosis_steroids"]],
+        ["Brak deficytu", data["no_deficit"]],
+        ["Deficyt korzeniowy", data["root_deficit"]],
+        ["Deficyt wielopoziomowy / obustronny", data["multilevel_bilateral"]],
+        ["Objawy rdzeniowe", data["cord_signs"]],
+        ["Brak leczenia zachowawczego", data["conservative_none"]],
+        ["Leczenie < 6 tygodni", data["conservative_lt6"]],
+        ["Leczenie ≥ 6 tygodni", data["conservative_ge6"]],
+        ["Poprawa", data["improved"]],
+        ["Brak poprawy", data["no_improvement"]],
+        ["Pogorszenie", data["worsening"]],
+        ["Podejrzenie dyskopatii", data["suspect_disc"]],
+        ["Podejrzenie ucisku rdzenia", data["suspect_cord"]],
+        ["Podejrzenie nowotworu", data["suspect_tumor"]],
+        ["Podejrzenie infekcji", data["suspect_infection"]],
+        ["Kwalifikacja do zabiegu", data["qualify_surgery"]],
+        ["Kwalifikacja do blokady", data["qualify_injection"]],
+        ["Czy wynik MR zmieni leczenie?", data["treatment_impact"]],
+    ]
+
+    details_table = Table(details, colWidths=[220, 290])
+    details_table.setStyle(
+        TableStyle(
+            [
+                ("BACKGROUND", (0, 0), (0, -1), colors.HexColor("#F3F4F6")),
+                ("GRID", (0, 0), (-1, -1), 0.4, colors.lightgrey),
+                ("VALIGN", (0, 0), (-1, -1), "TOP"),
+                ("FONTSIZE", (0, 0), (-1, -1), 9),
+                ("PADDING", (0, 0), (-1, -1), 5),
+            ]
+        )
+    )
+    story.append(Paragraph("<b>Dane formularza:</b>", styles["Heading3"]))
+    story.append(details_table)
+    story.append(Spacer(1, 14))
+    story.append(Paragraph("Uwaga: narzędzie ma charakter wspomagający i nie zastępuje oceny lekarza.", styles["Italic"]))
 
-    doc.build(content)
+    doc.build(story)
     return filename
 
 
-def run(bladder, saddle, deficit_prog, myelopathy,
-        cancer, fever, immuno,
-        trauma,
-        radiculopathy,
-        duration,
-        no_improvement,
-        post_op,
-        impact):
+def render_html_banner(category, imaging, color):
+    return f"""
+    <div style="padding:16px;border-radius:14px;background:{color};color:white;
+                font-size:18px;font-weight:700;text-align:center;margin:8px 0;">
+        {category}<br>
+        <span style="font-size:14px;font-weight:500;">{imaging}</span>
+    </div>
+    """
+
+
+def run_app(
+    spine_section,
+    duration,
+    first_episode,
+    recurrent_episode,
+    post_op,
+    axial_pain,
+    radicular_pain,
+    paresthesia,
+    weakness,
+    gait_balance,
+    sphincter,
+    saddle,
+    progressive_deficit,
+    myelopathy,
+    cancer_history,
+    weight_loss,
+    night_pain,
+    fever,
+    immunosuppression,
+    infection_suspicion,
+    trauma_high,
+    trauma_low_risk,
+    osteoporosis_steroids,
+    no_deficit,
+    root_deficit,
+    multilevel_bilateral,
+    cord_signs,
+    conservative_none,
+    conservative_lt6,
+    conservative_ge6,
+    improved,
+    no_improvement,
+    worsening,
+    suspect_disc,
+    suspect_cord,
+    suspect_tumor,
+    suspect_infection,
+    qualify_surgery,
+    qualify_injection,
+    treatment_impact,
+):
+    warnings = validate_form(
+        first_episode, recurrent_episode,
+        no_deficit, root_deficit, multilevel_bilateral, cord_signs,
+        conservative_none, conservative_lt6, conservative_ge6,
+        improved, no_improvement, worsening
+    )
 
-    rec, just = decision_logic(
-        bladder, saddle, deficit_prog, myelopathy,
-        cancer, fever, immuno,
-        trauma,
-        radiculopathy,
+    result = build_recommendation(
+        spine_section,
         duration,
-        no_improvement,
+        first_episode,
+        recurrent_episode,
         post_op,
-        impact
+        axial_pain,
+        radicular_pain,
+        paresthesia,
+        weakness,
+        gait_balance,
+        sphincter,
+        saddle,
+        progressive_deficit,
+        myelopathy,
+        cancer_history,
+        weight_loss,
+        night_pain,
+        fever,
+        immunosuppression,
+        infection_suspicion,
+        trauma_high,
+        trauma_low_risk,
+        osteoporosis_steroids,
+        no_deficit,
+        root_deficit,
+        multilevel_bilateral,
+        cord_signs,
+        conservative_none,
+        conservative_lt6,
+        conservative_ge6,
+        improved,
+        no_improvement,
+        worsening,
+        suspect_disc,
+        suspect_cord,
+        suspect_tumor,
+        suspect_infection,
+        qualify_surgery,
+        qualify_injection,
+        treatment_impact,
+    )
+
+    reasons_text = "\n".join([f"- {x}" for x in result["reasons"]]) if result["reasons"] else "-"
+    alt_text = "\n".join([f"- {x}" for x in result["alternatives"]]) if result["alternatives"] else "-"
+    warnings_text = "\n".join([f"- {x}" for x in warnings]) if warnings else "Brak ostrzeżeń walidacyjnych."
+
+    form_data = {
+        "spine_section": spine_section,
+        "duration": duration,
+        "first_episode": yes_no(first_episode),
+        "recurrent_episode": yes_no(recurrent_episode),
+        "post_op": yes_no(post_op),
+        "axial_pain": yes_no(axial_pain),
+        "radicular_pain": yes_no(radicular_pain),
+        "paresthesia": yes_no(paresthesia),
+        "weakness": yes_no(weakness),
+        "gait_balance": yes_no(gait_balance),
+        "sphincter": yes_no(sphincter),
+        "saddle": yes_no(saddle),
+        "progressive_deficit": yes_no(progressive_deficit),
+        "myelopathy": yes_no(myelopathy),
+        "cancer_history": yes_no(cancer_history),
+        "weight_loss": yes_no(weight_loss),
+        "night_pain": yes_no(night_pain),
+        "fever": yes_no(fever),
+        "immunosuppression": yes_no(immunosuppression),
+        "infection_suspicion": yes_no(infection_suspicion),
+        "trauma_high": yes_no(trauma_high),
+        "trauma_low_risk": yes_no(trauma_low_risk),
+        "osteoporosis_steroids": yes_no(osteoporosis_steroids),
+        "no_deficit": yes_no(no_deficit),
+        "root_deficit": yes_no(root_deficit),
+        "multilevel_bilateral": yes_no(multilevel_bilateral),
+        "cord_signs": yes_no(cord_signs),
+        "conservative_none": yes_no(conservative_none),
+        "conservative_lt6": yes_no(conservative_lt6),
+        "conservative_ge6": yes_no(conservative_ge6),
+        "improved": yes_no(improved),
+        "no_improvement": yes_no(no_improvement),
+        "worsening": yes_no(worsening),
+        "suspect_disc": yes_no(suspect_disc),
+        "suspect_cord": yes_no(suspect_cord),
+        "suspect_tumor": yes_no(suspect_tumor),
+        "suspect_infection": yes_no(suspect_infection),
+        "qualify_surgery": yes_no(qualify_surgery),
+        "qualify_injection": yes_no(qualify_injection),
+        "treatment_impact": treatment_impact,
+    }
+
+    pdf_file = generate_pdf(form_data, result, warnings)
+    banner_html = render_html_banner(result["category"], result["imaging"], result["color"])
+
+    return (
+        banner_html,
+        result["category"],
+        result["imaging"],
+        result["clinical_path"],
+        result["urgency"],
+        result["contrast"],
+        reasons_text,
+        alt_text,
+        warnings_text,
+        pdf_file,
     )
 
-    pdf = generate_pdf(rec, just)
-    return rec, just, pdf
 
+with gr.Blocks(theme=gr.themes.Soft(), title="MR Spine CDS PRO+") as app:
+    gr.Markdown(
+        f"""
+        # MR Spine Clinical Decision Support – PRO+
+        **Autor:** {AUTHOR_NAME}  
+        **Afiliacja:** {AUTHOR_AFFILIATION}
+        
+        Ustrukturyzowany formularz kwalifikacji do badania MR kręgosłupa.  
+        **Uwaga:** narzędzie wspomagające decyzję kliniczną, nie zastępuje oceny lekarza.
+        """
+    )
 
-with gr.Blocks() as app:
+    with gr.Group():
+        gr.Markdown("## A. Identyfikacja problemu")
+        with gr.Row():
+            spine_section = gr.Radio(
+                ["szyjnego", "piersiowego", "lędźwiowego"],
+                label="Odcinek kręgosłupa",
+                value="lędźwiowego",
+            )
+            duration = gr.Radio(
+                ["< 2 tygodni", "2–6 tygodni", "6–12 tygodni", "> 12 tygodni"],
+                label="Czas trwania objawów",
+                value="2–6 tygodni",
+            )
+        with gr.Row():
+            first_episode = gr.Checkbox(label="Pierwszy epizod")
+            recurrent_episode = gr.Checkbox(label="Nawrót / zaostrzenie przewlekłe")
+            post_op = gr.Checkbox(label="Pacjent po operacji kręgosłupa")
 
-    gr.Markdown("## 🧠 MR Spine Decision Support Tool")
-    gr.Markdown("⚠️ Narzędzie wspomagające decyzję kliniczną — nie zastępuje lekarza")
+    with gr.Group():
+        gr.Markdown("## B. Charakter objawów")
+        with gr.Row():
+            axial_pain = gr.Checkbox(label="Ból miejscowy (osiowy)")
+            radicular_pain = gr.Checkbox(label="Ból promieniujący (radikulopatia)")
+            paresthesia = gr.Checkbox(label="Parestezje / drętwienie")
+            weakness = gr.Checkbox(label="Osłabienie siły mięśniowej")
+            gait_balance = gr.Checkbox(label="Zaburzenia chodu / równowagi")
 
-    bladder = gr.Checkbox(label="Zaburzenia zwieraczy")
-    saddle = gr.Checkbox(label="Znieczulenie krocza")
-    deficit_prog = gr.Checkbox(label="Postępujący niedowład")
-    myelopathy = gr.Checkbox(label="Mielopatia")
+    with gr.Group():
+        gr.Markdown("## C. Red flags")
+        with gr.Row():
+            sphincter = gr.Checkbox(label="Zaburzenia zwieraczy")
+            saddle = gr.Checkbox(label="Znieczulenie krocza")
+            progressive_deficit = gr.Checkbox(label="Postępujący niedowład")
+            myelopathy = gr.Checkbox(label="Objawy mielopatii")
+        with gr.Row():
+            cancer_history = gr.Checkbox(label="Wywiad nowotworowy")
+            weight_loss = gr.Checkbox(label="Utrata masy ciała")
+            night_pain = gr.Checkbox(label="Ból nocny / niemechaniczny")
+        with gr.Row():
+            fever = gr.Checkbox(label="Gorączka")
+            immunosuppression = gr.Checkbox(label="Immunosupresja")
+            infection_suspicion = gr.Checkbox(label="Podejrzenie infekcji")
+        with gr.Row():
+            trauma_high = gr.Checkbox(label="Uraz wysokiej energii")
+            trauma_low_risk = gr.Checkbox(label="Uraz niskoenergetyczny + ryzyko")
+            osteoporosis_steroids = gr.Checkbox(label="Osteoporoza / przewlekłe steroidy")
 
-    cancer = gr.Checkbox(label="Nowotwór")
-    fever = gr.Checkbox(label="Gorączka")
-    immuno = gr.Checkbox(label="Immunosupresja")
+    with gr.Group():
+        gr.Markdown("## D. Badanie neurologiczne")
+        with gr.Row():
+            no_deficit = gr.Checkbox(label="Brak deficytu")
+            root_deficit = gr.Checkbox(label="Deficyt korzeniowy")
+            multilevel_bilateral = gr.Checkbox(label="Deficyt wielopoziomowy / obustronny")
+            cord_signs = gr.Checkbox(label="Objawy rdzeniowe")
 
-    trauma = gr.Checkbox(label="Uraz")
+    with gr.Group():
+        gr.Markdown("## E. Dotychczasowe leczenie")
+        with gr.Row():
+            conservative_none = gr.Checkbox(label="Brak leczenia zachowawczego")
+            conservative_lt6 = gr.Checkbox(label="Leczenie < 6 tygodni")
+            conservative_ge6 = gr.Checkbox(label="Leczenie ≥ 6 tygodni")
+        with gr.Row():
+            improved = gr.Checkbox(label="Poprawa")
+            no_improvement = gr.Checkbox(label="Brak poprawy")
+            worsening = gr.Checkbox(label="Pogorszenie")
 
-    radiculopathy = gr.Checkbox(label="Radikulopatia")
-    duration = gr.Radio(["<6 tygodni", ">6 tygodni"], label="Czas objawów")
+    with gr.Group():
+        gr.Markdown("## F. Pytanie kliniczne")
+        with gr.Row():
+            suspect_disc = gr.Checkbox(label="Podejrzenie dyskopatii")
+            suspect_cord = gr.Checkbox(label="Podejrzenie ucisku rdzenia")
+            suspect_tumor = gr.Checkbox(label="Podejrzenie nowotworu")
+            suspect_infection = gr.Checkbox(label="Podejrzenie infekcji")
+        with gr.Row():
+            qualify_surgery = gr.Checkbox(label="Kwalifikacja do zabiegu")
+            qualify_injection = gr.Checkbox(label="Kwalifikacja do blokady / iniekcji")
+            treatment_impact = gr.Radio(
+                ["TAK", "NIE", "NIEPEWNE"],
+                label="Czy wynik MR zmieni leczenie?",
+                value="NIEPEWNE",
+            )
 
-    no_improvement = gr.Checkbox(label="Brak poprawy")
-    post_op = gr.Checkbox(label="Po operacji")
-    impact = gr.Checkbox(label="Wpływ na leczenie")
+    with gr.Row():
+        run_button = gr.Button("Generuj rekomendację", variant="primary")
+        clear_button = gr.Button("Wyczyść formularz")
 
-    btn = gr.Button("🔍 Oceń")
+    gr.Markdown("## Wynik")
+    out_banner = gr.HTML()
+    out_category = gr.Textbox(label="Kategoria rekomendacji")
+    out_imaging = gr.Textbox(label="Sugerowane badanie")
+    out_path = gr.Textbox(label="Ścieżka kliniczna")
+    out_urgency = gr.Textbox(label="Tryb")
+    out_contrast = gr.Textbox(label="Kontrast")
+    out_reasons = gr.Textbox(label="Uzasadnienie", lines=8)
+    out_alternatives = gr.Textbox(label="Alternatywy / dalsze kroki", lines=5)
+    out_warnings = gr.Textbox(label="Ostrzeżenia walidacyjne", lines=4)
+    out_pdf = gr.File(label="Pobierz raport PDF")
 
-    out1 = gr.Textbox(label="Rekomendacja")
-    out2 = gr.Textbox(label="Uzasadnienie")
-    out3 = gr.File(label="Pobierz PDF")
+    inputs = [
+        spine_section,
+        duration,
+        first_episode,
+        recurrent_episode,
+        post_op,
+        axial_pain,
+        radicular_pain,
+        paresthesia,
+        weakness,
+        gait_balance,
+        sphincter,
+        saddle,
+        progressive_deficit,
+        myelopathy,
+        cancer_history,
+        weight_loss,
+        night_pain,
+        fever,
+        immunosuppression,
+        infection_suspicion,
+        trauma_high,
+        trauma_low_risk,
+        osteoporosis_steroids,
+        no_deficit,
+        root_deficit,
+        multilevel_bilateral,
+        cord_signs,
+        conservative_none,
+        conservative_lt6,
+        conservative_ge6,
+        improved,
+        no_improvement,
+        worsening,
+        suspect_disc,
+        suspect_cord,
+        suspect_tumor,
+        suspect_infection,
+        qualify_surgery,
+        qualify_injection,
+        treatment_impact,
+    ]
 
-    btn.click(run,
-              inputs=[bladder, saddle, deficit_prog, myelopathy,
-                      cancer, fever, immuno,
-                      trauma,
-                      radiculopathy,
-                      duration,
-                      no_improvement,
-                      post_op,
-                      impact],
-              outputs=[out1, out2, out3])
+    outputs = [
+        out_banner,
+        out_category,
+        out_imaging,
+        out_path,
+        out_urgency,
+        out_contrast,
+        out_reasons,
+        out_alternatives,
+        out_warnings,
+        out_pdf,
+    ]
+
+    run_button.click(fn=run_app, inputs=inputs, outputs=outputs)
+
+    clear_button.click(
+        fn=lambda: (
+            "lędźwiowego",
+            "2–6 tygodni",
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            False,
+            "NIEPEWNE",
+            "",
+            "",
+            "",
+            "",
+            "",
+            "",
+            "",
+            "",
+            None,
+        ),
+        inputs=[],
+        outputs=inputs + outputs,
+    )
 
 app.launch()