Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,38 +3,40 @@ import numpy as np
 import pandas as pd
 import streamlit as st
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
+st.set_page_config(layout="wide", page_title="AI Recon Visualizer")
+st.title("🧠 AI Recon Visualizer")
+st.markdown("""
+This interactive spiral chart simulates discovered targets during an automated reconnaissance workflow.
+Each point represents a discovered service or subdomain, with random weights simulating risk or exposure levels.
+""")
+
+# User-controlled parameters
+num_points = st.slider("📈 Number of discovered endpoints", 10, 10000, 800)
+num_turns = st.slider("🔄 Spiral depth (rotations)", 1, 300, 40)
+
+# Spiral coordinates
 indices = np.linspace(0, 1, num_points)
 theta = 2 * np.pi * num_turns * indices
 radius = indices
-
 x = radius * np.cos(theta)
 y = radius * np.sin(theta)
 
+# Simulated recon data
 df = pd.DataFrame({
     "x": x,
     "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
+    "target_score": indices,
+    "risk": np.random.randn(num_points),
+    "endpoint": [f"host{i}.recon.local" for i in range(num_points)]
 })
 
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# Chart definition
+chart = alt.Chart(df, height=700, width=700).mark_point(filled=True).encode(
+    x=alt.X("x", axis=None),
+    y=alt.Y("y", axis=None),
+    color=alt.Color("target_score", legend=None, scale=alt.Scale(scheme="viridis")),
+    size=alt.Size("risk", legend=None, scale=alt.Scale(range=[10, 150])),
+    tooltip=["endpoint", "risk", "target_score"]
+).interactive()
+
+st.altair_chart(chart, use_container_width=True)