Update app.py

app.py

@@ -5,9 +5,8 @@ import random
 import time
 from scipy.stats import entropy as scipy_entropy
 
-# --- НАСТРОЙКИ ---
+# Константы
 seqlen = 60
-steps = 10000
 min_run, max_run = 1, 2
 ANGLE_MAP = {'A': 60.0, 'C': 180.0, 'G': -60.0, 'T': -180.0, 'N': 0.0}
 bases = ['A', 'C', 'G', 'T']
@@ -16,7 +15,7 @@ lags_shown = 6
 st.set_page_config(layout="wide")
 st.title("🌌 Визуализация торсионных биомашин")
 
-# --- ИНТЕРФЕЙС ---
+# UI
 col1, col2, col3 = st.columns([1,1,2])
 with col1:
     if 'running' not in st.session_state:
@@ -36,7 +35,7 @@ with col2:
 with col3:
     speed = st.slider("⏱ Скорость обновления (мс)", 10, 1000, 200, step=10)
 
-# --- ИНИЦИАЛИЗАЦИЯ ---
+# Init
 if 'seq' not in st.session_state:
     st.session_state.seq = ''.join(random.choices(bases, k=seqlen))
 if 'step' not in st.session_state:
@@ -46,7 +45,7 @@ if 'stat_bist_counts' not in st.session_state:
     st.session_state.stat_entropy = []
     st.session_state.stat_autocorr = []
 
-# --- ФУНКЦИИ ---
+# Функции
 def find_local_min_runs(profile, min_run=1, max_run=2):
     result = []
     N = len(profile)
@@ -101,35 +100,31 @@ def bio_mutate(seq):
         if len(seq) > 10:
             start = random.randint(0, len(seq)-6)
             end = start + random.randint(3,6)
-            subseq = seq[start:end]
-            subseq = subseq[::-1]
+            subseq = seq[start:end][::-1]
             seq = seq[:start] + subseq + seq[end:]
     while len(seq) < seqlen:
         seq += random.choice(bases)
-    if len(seq) > seqlen:
-        seq = seq[:seqlen]
-    return seq
+    return seq[:seqlen]
 
-# --- ВИЗУАЛИЗАЦИЯ ---
+# Визуализация
 plot_area = st.empty()
-fig, axs = plt.subplots(3, 1, figsize=(10, 8))
-plt.subplots_adjust(hspace=0.5)
 
-# --- ЦИКЛ ВИЗУАЛИЗАЦИИ ---
 while st.session_state.running:
+    # Мутация
     st.session_state.seq = bio_mutate(st.session_state.seq)
     torsion_profile = np.array([ANGLE_MAP.get(nt, 0.0) for nt in st.session_state.seq])
     runs = find_local_min_runs(torsion_profile, min_run, max_run)
     ent = compute_entropy(torsion_profile)
     acorr = compute_autocorr(torsion_profile)
 
+    # Сохраняем последние 50
     st.session_state.stat_bist_counts = st.session_state.stat_bist_counts[-50:] + [len(runs)]
     st.session_state.stat_entropy = st.session_state.stat_entropy[-50:] + [ent]
     st.session_state.stat_autocorr = st.session_state.stat_autocorr[-50:] + [acorr]
 
-    axs[0].cla()
-    axs[1].cla()
-    axs[2].cla()
+    # График
+    fig, axs = plt.subplots(3, 1, figsize=(10, 8))
+    plt.subplots_adjust(hspace=0.5)
 
     axs[0].plot(torsion_profile, color='royalblue')
     for start, end, val in runs:
@@ -142,8 +137,8 @@ while st.session_state.running:
     axs[1].set_title("Число 'биомашин'")
 
     axs[2].bar(np.arange(lags_shown), acorr[:lags_shown], color='teal')
-    axs[2].set_title("Автокорреляция углового профиля")
+    axs[2].set_title("Автокорреляция")
 
-    plot_area.pyplot(fig, clear_figure=True)
+    plot_area.pyplot(fig)
     st.session_state.step += 1
     time.sleep(speed / 1000.0)