Update app.py

app.py

@@ -7,24 +7,27 @@ def create_amygdala_hijacking_graph():
     g = Digraph('Amygdala_Hijacking', format='png')
 
     g.attr(rankdir='LR')
-
-    g.node('1', 'Sensory Input', shape='oval')
-    g.node('2', 'Thalamus', shape='oval')
-    g.node('3', 'Amygdala', shape='oval', color='red', style='filled')
-    g.node('4', 'Hippocampus', shape='oval')
-    g.node('5', 'Prefrontal Cortex', shape='oval')
-    g.node('6', 'Response', shape='oval')
-
-    g.edge('1', '2', label='Receives signals')
-    g.edge('2', '3', label='Quick, emotional response')
-    g.edge('2', '4', label='Sends signals to')
-    g.edge('4', '5', label='Relays information')
-    g.edge('5', '3', label='Rational control (if not hijacked)')
-    g.edge('3', '6', label='Generates response')
+    g.attr('node', shape='oval', fontname='Arial', fontsize='16', fontcolor='black')
+    g.attr('edge', fontname='Arial', fontsize='12', fontcolor='blue')
+
+    g.node('1', 'Sensory Input', shape='rect', style='filled', fillcolor='lightblue')
+    g.node('2', 'Thalamus', shape='ellipse', style='filled', fillcolor='lightgreen')
+    g.node('3', 'Amygdala', shape='ellipse', color='red', style='filled', fillcolor='red', fontcolor='white')
+    g.node('4', 'Hippocampus', shape='ellipse', style='filled', fillcolor='lightyellow')
+    g.node('5', 'Prefrontal Cortex', shape='ellipse', style='filled', fillcolor='lightpink')
+    g.node('6', 'Response', shape='rect', style='filled', fillcolor='lightgray')
+
+    g.edge('1', '2', label='Receives Signals')
+    g.edge('2', '3', label='Quick, Emotional Response')
+    g.edge('2', '4', label='Sends Signals To')
+    g.edge('4', '5', label='Relays Information')
+    g.edge('5', '3', label='Rational Control (If Not Hijacked)')
+    g.edge('3', '6', label='Generates Response')
 
     return g
 
 
+
 def main():
     st.title("Amygdala Hijacking Visualization")
     st.text("A simple graph model to represent amygdala hijacking in the brain.")