nrstitcher-preprocessing/refactor.py at master · allison714/nrstitcher-preprocessing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import sys

with open('app.py', 'r', encoding='utf-8') as f:
    lines = f.readlines()

out_lines = []
in_gen_bundle = False
in_analytics = False
in_drift = False

for i, line in enumerate(lines):
    # 1. Expanders top block
    if line.startswith('# Display Verification'):
        out_lines.append('# --- Top Level UI Expanders ---\n')
        out_lines.append('exp_generate = st.expander("🚀 Generate Run Bundle", expanded=True)\n\n')
        out_lines.append('with exp_generate:\n')
        out_lines.append('    ' + line)
        in_gen_bundle = True
        continue

    # Stop indenting right before the post-generation tab definition
    if in_gen_bundle and line.startswith('# --- Tabs for Post-Generation / Utilities ---'):
        in_gen_bundle = False
        out_lines.append(line)
        continue

    # Post gen tabs section replacer
    if line.startswith('tab_verify, tab_analytics = st.tabs'):
        out_lines.append('exp_verify = st.expander("✅ Verification & Metadata", expanded=True)\n')
        out_lines.append('exp_warping = st.expander("🗺️ Warping Diagnostics", expanded=False)\n')
        out_lines.append('exp_drift = st.expander("📉 Intensity Drift Analysis", expanded=False)\n')
        continue

    if line.startswith('with tab_verify:'):
        out_lines.append('with exp_verify:\n')
        continue

    if line.startswith('with tab_analytics:'):
        out_lines.append('with exp_warping:\n')
        in_analytics = True
        continue

    # Split drift from analytics
    if in_analytics and line.strip() == 'st.write("### Intensity Drift Analysis")':
        out_lines.append('with exp_drift:\n')
        out_lines.append('    ' + line.lstrip())
        in_analytics = False
        in_drift = True
        continue

    if in_analytics and line.strip() == 'st.markdown("---")' and lines[i+1].strip() == 'st.write("### Intensity Drift Analysis")':
        # Skip the markdown line before drift analysis
        continue

    # Indent core bundle logic
    if in_gen_bundle:
        if line.strip() == '':
            out_lines.append(line)
        else:
            out_lines.append('    ' + line)
    else:
        out_lines.append(line)

with open('app.py', 'w', encoding='utf-8') as f:
    f.writelines(out_lines)
print("SUCCESS")