Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import pandas as pd
-import plotly.graph_objects as go
-from io import StringIO
+from graphviz import Digraph
 import base64
+from io import StringIO
 
 # Custom CSS for professional UI
 st.markdown("""
@@ -57,86 +57,23 @@ st.markdown("""
 
 # Function to create organization chart
 def create_org_chart(df, title):
-    positions = {}
-    edge_x = []
-    edge_y = []
-    node_x = []
-    node_y = []
-    node_text = []
-    
-    # Create hierarchical positions
-    levels = {}
-    for _, row in df.iterrows():
-        parent = row['Parent']
-        child = row['Child']
-        if parent not in levels:
-            levels[parent] = 0
-        if child not in levels:
-            levels[child] = levels[parent] + 1
-    
-    # Assign coordinates
-    max_level = max(levels.values())
-    level_positions = {level: [] for level in range(max_level + 1)}
-    
-    for node, level in levels.items():
-        level_positions[level].append(node)
-    
-    for level, nodes in level_positions.items():
-        y_pos = max_level - level
-        x_positions = [i - (len(nodes)-1)/2 for i in range(len(nodes))]
-        for node, x_pos in zip(nodes, x_positions):
-            positions[node] = (x_pos, y_pos)
+    dot = Digraph(comment='Organization Chart')
+    dot.attr(rankdir='TB', labelloc='t', label=title, fontsize='20', fontname='Arial')
     
-    # Create edges
+    # Add nodes and edges
     for _, row in df.iterrows():
         parent = row['Parent']
         child = row['Child']
-        x0, y0 = positions[parent]
-        x1, y1 = positions[child]
-        edge_x.extend([x0, x1, None])
-        edge_y.extend([y0, y1, None])
-    
-    # Create nodes
-    for node, (x, y) in positions.items():
-        node_x.append(x)
-        node_y.append(y)
-        node_text.append(node)
-    
-    # Create 3D effect
-    edge_trace = go.Scatter(
-        x=edge_x, y=edge_y,
-        line=dict(width=2, color='#888'),
-        hoverinfo='none',
-        mode='lines'
-    )
-    
-    node_trace = go.Scatter(
-        x=node_x, y=node_y,
-        mode='markers+text',
-        text=node_text,
-        textposition="middle center",
-        marker=dict(
-            symbol='square',
-            size=[50] * len(node_x),  # Reduced size for better fit
-            color='#4CAF50',
-            line=dict(width=2, color='DarkSlateGrey')
-        ),
-        hoverinfo='text',
-        textfont=dict(size=12, color='white')
-    )
-    
-    fig = go.Figure(data=[edge_trace, node_trace],
-                 layout=go.Layout(
-                    title=f'<b>{title}</b><br>',
-                    titlefont_size=20,
-                    showlegend=False,
-                    hovermode='closest',
-                    margin=dict(b=20, l=5, r=5, t=40),
-                    xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-                    yaxis=dict(showgrid=False, zeroline=False, showticklabels=False))
-                    )
+        
+        # Add parent node if not already added
+        if parent not in dot.body:
+            dot.node(parent, shape='box', style='filled', fillcolor='#4CAF50', fontname='Arial', fontcolor='white')
+        
+        # Add child node and edge
+        dot.node(child, shape='box', style='filled', fillcolor='#4CAF50', fontname='Arial', fontcolor='white')
+        dot.edge(parent, child)
     
-    return fig
+    return dot
 
 # Function to download CSV template
 def download_csv_template():
@@ -194,8 +131,10 @@ CFO,Accounting Manager"""
         # Generate chart
         if st.button("Generate Organization Chart"):
             if not edited_df.empty:
-                fig = create_org_chart(edited_df, chart_title)
-                st.plotly_chart(fig, use_container_width=True)
+                dot = create_org_chart(edited_df, chart_title)
+                
+                # Display the chart
+                st.graphviz_chart(dot)
                 
                 # PDF download options
                 st.markdown("### Download Options")
@@ -207,12 +146,10 @@ CFO,Accounting Manager"""
                 
                 # Convert to PDF
                 if st.button("Download as PDF"):
-                    img_bytes = fig.to_image(format="pdf", 
-                                           width=800 if orientation == "Portrait" else 1200,
-                                           height=1200 if orientation == "Portrait" else 800)
+                    pdf_bytes = dot.pipe(format='pdf')
                     st.download_button(
                         label="Download PDF",
-                        data=img_bytes,
+                        data=pdf_bytes,
                         file_name="org_chart.pdf",
                         mime="application/pdf"
                     )