Update src/streamlit_app.py

src/streamlit_app.py

@@ -26,51 +26,13 @@ def load_data():
 
 df_human, df_minimlist, df_cheerleader, df_poet = load_data()
 
-st.divider()
-st.subheader("🧬 TwinTrack Visualization")
-
-selected_persona = st.selectbox(
-    "Select AI Persona to Compare:",
-    ["Minimalist", "Cheerleader", "Poet"]
-)
-
-persona_map = {
-    "Minimalist": df_minimlist,  
-    "Cheerleader": df_cheerleader,
-    "Poet": df_poet
-}
-df_ai = persona_map[selected_persona]
-
-sensitivity = st.slider("Perturbation Sensitivity (Simulated)", 0.0, 1.0, 0.5)
-
-import plotly.graph_objects as go
-from plotly.subplots import make_subplots
-
-fig = make_subplots(rows=2, cols=1, shared_xaxes=True, vertical_spacing=0.1,
-                    subplot_titles=("Human: Big Flame", f"AI: {selected_persona}"))
-
-
-def create_strip(df, row_num, color_scale):
-    adjusted_values = df['tokens_est'] * (1 + sensitivity)
-    
-    return go.Heatmap(
-        z=[adjusted_values],  
-        x=df['turn'],
-        colorscale=color_scale,
-        showscale=False,      
-        hovertext=df['words'], 
-    )
-
-fig.add_trace(create_strip(df_human, 1, 'Hot'), row=1, col=1)
-
-fig.add_trace(create_strip(df_ai, 2, 'Teal'), row=2, col=1)
-
-fig.update_layout(
-    height=300,
-    margin=dict(l=20, r=20, t=40, b=20),
-    xaxis2_title="Turn Number"
-)
-fig.update_yaxes(showticklabels=False) 
-
-st.plotly_chart(fig, use_container_width=True)
+def normalize_time(df):
+    df["t_norm"] = df.index /
+(len(df) - 1)
+    return df
+
+df_human = normalize_time(df_human)
+df_minimlist = normalize_time(df_minimlist)
+df_cheerleader = normalize_time(df_cheerleader)
+df_poet = normalize_time(df_poet)