codebase-brain 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

brain_parser/bug_detector.py

@@ -9,32 +9,33 @@ def detect_circular_dependencies(brain, temp_path=None):
     G = build_graph(brain)
     bugs = []
 
-    def clean(path):
+    def rel(path):
         p = path.replace('\\', '/')
         idx = p.find('/temp/')
         if idx != -1:
             after_temp = p[idx + 6:]
             parts = after_temp.split('/', 1)
-            if len(parts) > 1:
-                return parts[1]
-            return parts[0]
-        # fallback — just filename
-        return p.split('/')[-1]
+            p = parts[1] if len(parts) > 1 else parts[0]
+        if '/src/' in p:
+            p = p[p.find('/src/') + 1:]
+        elif p.startswith('./'):
+            p = p[2:]
+        return p
 
     try:
-        cycles = list(nx.simple_cycles(G))
-        for cycle in cycles:
-            clean_cycle = [clean(f) for f in cycle]
-            # format as numbered list not one long line
-            chain = '\n  '.join([f"{i+1}. {f}" for i, f in enumerate(clean_cycle)])
-            chain += f"\n  → back to {clean_cycle[0]}"
+        sccs = [scc for scc in nx.strongly_connected_components(G) if len(scc) > 1]
+
+        for scc in sccs:
+            clean_files = sorted(list(set([rel(f).split('/')[-1] for f in scc])))
+            chain = '\n  '.join([f"{i+1}. {f}" for i, f in enumerate(clean_files)])
             bugs.append({
                 'type': 'circular_dependency',
                 'severity': 'HIGH',
-                'files': clean_cycle,
-                'message': f"Circular dependency ({len(cycle)} files):\n  {chain}",
+                'files': clean_files,
+                'message': f"Circular dependency group ({len(clean_files)} files):\n  {chain}",
                 'fix': 'Extract shared logic into a separate file that both can import from.'
             })
+
     except Exception as e:
         print(f"Cycle detection error: {e}")
 

brain_parser/codebase_walker.py

@@ -53,18 +53,23 @@ def walk_codebase(root_path):
 
     return brain
 
-BRAIN_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'brain.json')
+def get_brain_path():
+    """Always returns brain.json in the current working directory."""
+    return os.path.join(os.getcwd(), 'brain.json')
 
-def save_brain(brain, output_path=BRAIN_PATH):
+def save_brain(brain, output_path=None):
+    if output_path is None:
+        output_path = get_brain_path()
     with open(output_path, "w") as f:
         json.dump(brain, f, indent=4)
     print(f"Brain saved to {output_path}")
 
-def load_brain(input_path=BRAIN_PATH):
-    if not os.path.exists(input_path):
-        print("No brain found. Run walker first.")
+def load_brain(output_path=None):
+    if output_path is None:
+        output_path = get_brain_path()
+    if not os.path.exists(output_path):
         return None
-    with open(input_path, "r") as f:
+    with open(output_path, 'r') as f:
         return json.load(f)
 
 

brain_parser/graph_builder.py

@@ -82,71 +82,185 @@ def calculate_risk(G):
     return risk
 
 def visualize_interactive(G):
-    risk = calculate_risk(G)\
+    from pyvis.network import Network
+    import os
 
-    color_map = {
-        "HIGH": "#ff4444",
-        "MEDIUM": "#ffa500",
-        "LOW": "#44cc44"
-    }
+    risk = calculate_risk(G)
 
     net = Network(
-        height="750px",
+        height="100vh",
         width="100%",
-        bgcolor="#1a1a2e",
-        font_color="white",
-        directed=True
+        bgcolor="#0d0d0d",
+        font_color="#ffffff",
+        directed=True,
+        cdn_resources='in_line'
     )
 
+    color_map = {
+        "HIGH": "#ff3860",
+        "MEDIUM": "#ffdd57",
+        "LOW": "#23d160"
+    }
+
+    # build node map first — ensures consistent keys for edges
+    node_map = {}
     for node in G.nodes():
-        risk_level = risk[node]
+        node_map[node] = str(node)
+
+    for node, node_id in node_map.items():
+        risk_level = risk.get(node, 'LOW')
         color = color_map[risk_level]
-        label = node.split("\\")[-1].split("/")[-1]
+        label = node.replace('\\', '/').split('/')[-1]
+        size = 18 + (G.in_degree(node) * 5)
 
         net.add_node(
-            node,
+            node_id,
             label=label,
-            color=color,
-            size=20 + (G.in_degree(node) * 3),
-            title=f"File: {label}\nRisk: {risk_level}\nConnections: {G.in_degree(node)}"
+            color={
+                'background': color,
+                'border': color,
+                'highlight': {'background': '#ffffff', 'border': color}
+            },
+            size=size,
+            font={'size': 14, 'color': '#ffffff', 'face': 'monospace'},
+            borderWidth=2,
+            shadow=True,
+             title=f"File: {label}\nRisk: {risk_level}\nConnections: {G.in_degree(node)}"
         )
 
     for edge in G.edges():
-        net.add_edge(edge[0], edge[1])
+        src = node_map.get(edge[0], str(edge[0]))
+        dst = node_map.get(edge[1], str(edge[1]))
+        if src in net.get_nodes() and dst in net.get_nodes():
+            net.add_edge(
+                src, dst,
+                color={'color': '#ffffff33'},
+                width=1.5,
+                arrows={'to': {'enabled': True, 'scaleFactor': 0.6}}
+            )
 
     net.set_options("""
     {
       "nodes": {
         "borderWidth": 2,
-        "shadow": true
+        "shadow": {
+          "enabled": true,
+          "color": "rgba(0,0,0,0.8)",
+          "size": 15,
+          "x": 0,
+          "y": 0
+        }
       },
       "edges": {
         "smooth": {
-          "type": "dynamic"
+          "type": "curvedCW",
+          "roundness": 0.2
         },
-        "shadow": true
+        "shadow": false
       },
       "interaction": {
         "hover": true,
-        "navigationButtons": true,
-        "hideEdgesOnDrag": true
+        "navigationButtons": false,
+        "hideEdgesOnDrag": true,
+        "tooltipDelay": 100
       },
       "physics": {
-        "stabilization": true,
+        "stabilization": {
+          "enabled": true,
+          "iterations": 200
+        },
         "barnesHut": {
-          "gravitationalConstant": -8000,
-          "springLength": 200,
-          "springConstant": 0.02
+          "gravitationalConstant": -12000,
+          "centralGravity": 0.1,
+          "springLength": 250,
+          "springConstant": 0.02,
+          "damping": 0.09
         }
       }
     }
     """)
-    net.save_graph("brain_map.html")
+
+    output_path = os.path.join(os.getcwd(), 'brain_map.html')
+
+    html_content = net.generate_html()
+    with open(output_path, 'w', encoding='utf-8') as f:
+        f.write(html_content)
+
+    with open(output_path, 'r', encoding='utf-8') as f:
+        html = f.read()
+
+    custom_style = """
+<style>
+* { margin: 0; padding: 0; box-sizing: border-box; }
+body { background: #0d0d0d; font-family: 'Courier New', monospace; }
+#mynetwork {
+    background: radial-gradient(ellipse at center, #1a1a2e 0%, #0d0d0d 100%);
+    border: none !important;
+}
+.legend {
+    position: fixed;
+    top: 20px;
+    left: 20px;
+    background: rgba(13,13,13,0.9);
+    border: 1px solid #333;
+    border-radius: 12px;
+    padding: 16px 20px;
+    color: #fff;
+    font-family: monospace;
+    font-size: 13px;
+    z-index: 1000;
+    backdrop-filter: blur(10px);
+}
+.legend-title {
+    color: #888;
+    font-size: 11px;
+    letter-spacing: 2px;
+    text-transform: uppercase;
+    margin-bottom: 12px;
+}
+.legend-item { display: flex; align-items: center; gap: 10px; margin: 8px 0; }
+.dot { width: 12px; height: 12px; border-radius: 50%; box-shadow: 0 0 8px currentColor; }
+.high { background: #ff3860; color: #ff3860; }
+.medium { background: #ffdd57; color: #ffdd57; }
+.low { background: #23d160; color: #23d160; }
+.brain-title {
+    position: fixed;
+    top: 20px;
+    right: 20px;
+    color: #444;
+    font-family: monospace;
+    font-size: 12px;
+    letter-spacing: 3px;
+    text-transform: uppercase;
+    z-index: 1000;
+}
+div.vis-tooltip {
+    background: #1a1a2e !important;
+    border: 1px solid #ff3860 !important;
+    border-radius: 8px !important;
+    color: #fff !important;
+    font-family: monospace !important;
+    font-size: 12px !important;
+    padding: 10px 14px !important;
+}
+</style>
+<div class="legend">
+    <div class="legend-title">Risk Level</div>
+    <div class="legend-item"><div class="dot high"></div><span>High Risk</span></div>
+    <div class="legend-item"><div class="dot medium"></div><span>Medium Risk</span></div>
+    <div class="legend-item"><div class="dot low"></div><span>Low Risk</span></div>
+</div>
+<div class="brain-title">Codebase Brain</div>
+"""
+
+    html = html.replace('<body>', '<body>' + custom_style)
+
+    with open(output_path, 'w', encoding='utf-8') as f:
+        f.write(html)
+
     print(f"Nodes in graph: {len(G.nodes())}")
     print(f"Edges in graph: {len(G.edges())}")
-    import webbrowser
-    webbrowser.open("brain_map.html")
-    print("Interactive graph saved to brain_map.html")
+    print(f"Graph saved to {output_path}")
 
 
 def get_impact(G, filepath):

{codebase_brain-0.1.0.dist-info → codebase_brain-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebase-brain
-Version: 0.1.0
+Version: 0.2.0
 Summary: An AI brain that lives inside your codebase
 Requires-Python: >=3.8
 Requires-Dist: watchdog

{codebase_brain-0.1.0.dist-info → codebase_brain-0.2.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 brain_cli.py,sha256=sdJ5r60B4ELMruWR8B9azejpnizFkhdDDv0IrPhMLvE,11580
 brain_parser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 brain_parser/ast_parser.py,sha256=3Vy1XXyInjgA2xn6IBYv4K1juEdtFwAo58upqukCHXI,1799
-brain_parser/bug_detector.py,sha256=gvXBUFFk3RYoNQC-mToo80ytR7VZr_t1TfFEiSStmvo,7652
-brain_parser/codebase_walker.py,sha256=6uU1yJ6leqmHcdlSm-xcq1DOG1mdv4F0-JupSMZrSwE,3582
+brain_parser/bug_detector.py,sha256=IbEpyXfatL0r1cjaIZ7LXzYPg7z0molyRjg2ULu4828,7653
+brain_parser/codebase_walker.py,sha256=rWAHtHWurpZ7HhHAXLSuySMn07TIfg20LrRJ8wyCxKQ,3714
 brain_parser/file_watcher.py,sha256=cuok95kDJy6uKfsz_HByKWLBvRIPUAz7djAaXQ-2T3Q,2565
-brain_parser/graph_builder.py,sha256=fpQqY46oq4uqTzysPyhbUAaAwwJGV_yH5kQBW4xtuSs,5218
+brain_parser/graph_builder.py,sha256=gDeh4SCnpbjGGi0n1cX5kaj8uQ_-a6maKOMXK7Kn8z0,8730
 brain_parser/llm_router.py,sha256=f86O4tUw3MV1s2dq6axEzwJHEQfyboaqPr_QDR8CtK0,2447
 brain_parser/query_engine.py,sha256=Lunb_-9z1Lk2VYOE7f_-jF8X2AQtby1tPFXgFUjqnpA,3671
 brain_parser/root_cause.py,sha256=jvRB9OzFREU0oguGhfoU8wV2mCJ7A85y2BohybzIQ-0,5326
 brain_parser/universal_parser.py,sha256=47StXj-GPGJb4g0WzM5VYgM3tIc1tOzwEa8mcL_q9AY,6694
-codebase_brain-0.1.0.dist-info/METADATA,sha256=Eep4Qd7cPOarafSDhm6aysPe6lbhMk1fcPWJGqd3UVg,327
-codebase_brain-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-codebase_brain-0.1.0.dist-info/entry_points.txt,sha256=O9yHP4zy_uoVPxmewGbdNjNksmxRYUv34MQJDRXXodc,41
-codebase_brain-0.1.0.dist-info/top_level.txt,sha256=RmSvznU63-vmm5zLSKAlYYsFzQ8X3RyPrB-m0q0uQfI,23
-codebase_brain-0.1.0.dist-info/RECORD,,
+codebase_brain-0.2.0.dist-info/METADATA,sha256=fxLp9D79h6tMcXPXsS5T7kD4zJ93t6RgEmWaBRL-MQg,327
+codebase_brain-0.2.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+codebase_brain-0.2.0.dist-info/entry_points.txt,sha256=O9yHP4zy_uoVPxmewGbdNjNksmxRYUv34MQJDRXXodc,41
+codebase_brain-0.2.0.dist-info/top_level.txt,sha256=RmSvznU63-vmm5zLSKAlYYsFzQ8X3RyPrB-m0q0uQfI,23
+codebase_brain-0.2.0.dist-info/RECORD,,