Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import pandas as pd
-from graphviz import Digraph
-import base64
+import plotly.graph_objects as go
 from io import StringIO
+import base64
 
 # Custom CSS for professional UI
 st.markdown("""
@@ -10,12 +10,14 @@ st.markdown("""
     /* Main container */
     .stApp {
         background-color: #f5f5f5;
+        font-family: 'Arial', sans-serif;
     }
     
     /* Sidebar */
     .stSidebar {
         background-color: #2c3e50;
         color: white;
+        padding: 20px;
     }
     
     /* Buttons */
@@ -52,28 +54,104 @@ st.markdown("""
         background-color: #4CAF50;
         border-radius: 8px;
     }
+    
+    /* Organization chart container */
+    .org-chart-container {
+        background-color: white;
+        border-radius: 12px;
+        padding: 20px;
+        box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+    }
 </style>
 """, unsafe_allow_html=True)
 
 # Function to create organization chart
 def create_org_chart(df, title):
-    dot = Digraph(comment='Organization Chart')
-    dot.attr(rankdir='TB', labelloc='t', label=title, fontsize='20', fontname='Arial')
+    positions = {}
+    edge_x = []
+    edge_y = []
+    node_x = []
+    node_y = []
+    node_text = []
     
-    # Add nodes and edges
+    # Create hierarchical positions
+    levels = {}
     for _, row in df.iterrows():
         parent = row['Parent']
         child = row['Child']
-        
-        # 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)
+        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)
+    
+    # Create 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'
+    )
     
-    return dot
+    node_trace = go.Scatter(
+        x=node_x, y=node_y,
+        mode='markers+text',
+        text=node_text,
+        textposition="middle center",
+        marker=dict(
+            symbol='square',
+            size=[100] * len(node_x),  # Scalable size
+            color='#4CAF50',
+            line=dict(width=2, color='DarkSlateGrey'),
+            opacity=0.8
+        ),
+        hoverinfo='text',
+        textfont=dict(size=14, color='white'),
+        texttemplate='%{text}<br>',  # Wrap text
+    )
+    
+    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),
+                    plot_bgcolor='white',
+                    paper_bgcolor='white'
+                    )
+                    )
+    
+    return fig
 
 # Function to download CSV template
 def download_csv_template():
@@ -131,10 +209,8 @@ CFO,Accounting Manager"""
         # Generate chart
         if st.button("Generate Organization Chart"):
             if not edited_df.empty:
-                dot = create_org_chart(edited_df, chart_title)
-                
-                # Display the chart
-                st.graphviz_chart(dot)
+                fig = create_org_chart(edited_df, chart_title)
+                st.plotly_chart(fig, use_container_width=True)
                 
                 # PDF download options
                 st.markdown("### Download Options")
@@ -146,10 +222,12 @@ CFO,Accounting Manager"""
                 
                 # Convert to PDF
                 if st.button("Download as PDF"):
-                    pdf_bytes = dot.pipe(format='pdf')
+                    img_bytes = fig.to_image(format="pdf", 
+                                           width=800 if orientation == "Portrait" else 1200,
+                                           height=1200 if orientation == "Portrait" else 800)
                     st.download_button(
                         label="Download PDF",
-                        data=pdf_bytes,
+                        data=img_bytes,
                         file_name="org_chart.pdf",
                         mime="application/pdf"
                     )