Update app.py

app.py

@@ -1,76 +1,181 @@
 import streamlit as st
-from graphviz import Digraph
-import re
+import pandas as pd
+import plotly.graph_objects as go
+import base64
+from io import StringIO
 
-def create_org_chart(data, direction='TB'):
-    """Create organization chart using Graphviz"""
-    dot = Digraph(comment='Organization Chart', format='png')
-    dot.attr(rankdir=direction)
-    
-    # Track all added nodes to prevent duplicates
-    added_nodes = set()
-    
-    # Add relationships
-    for line in data:
-        if ':' in line:
-            parent, children = line.split(':', 1)
-            parent = parent.strip()
-            children = [child.strip() for child in children.split(',') if child.strip()]
-            
-            # Add parent node if not exists
-            if parent and parent not in added_nodes:
-                dot.node(parent)
-                added_nodes.add(parent)
-            
-            # Add children and relationships
-            for child in children:
-                if child and child not in added_nodes:
-                    dot.node(child)
-                    added_nodes.add(child)
-                if parent and child:
-                    dot.edge(parent, child)
-    
-    return dot
+# Custom CSS for modern UI
+st.markdown("""
+<style>
+    .stTextInput>div>div>input, .stTextArea>div>div>textarea {
+        background-color: #f0f2f6;
+        border-radius: 8px;
+        padding: 12px;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        border-radius: 8px;
+        padding: 10px 24px;
+        border: none;
+        font-size: 16px;
+    }
+    .stDownloadButton>button {
+        background-color: #008CBA;
+        color: white;
+        border-radius: 8px;
+        padding: 10px 24px;
+        border: none;
+        font-size: 16px;
+    }
+    .stSelectbox>div>div>select {
+        background-color: #f0f2f6;
+        border-radius: 8px;
+        padding: 8px;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+def create_org_chart(df, title, direction='TB'):
+    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)
+    
+    # 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')
+    
+    node_trace = go.Scatter(
+        x=node_x, y=node_y,
+        mode='markers+text',
+        text=node_text,
+        textposition="top center",
+        marker=dict(
+            symbol='square',
+            size=[100]*len(node_x),
+            color='#4CAF50',
+            line=dict(width=2, color='DarkSlateGrey')
+        ),
+        hoverinfo='text',
+        textfont=dict(size=14, color='white')
+    )
+    
+    fig = go.Figure(data=[edge_trace, node_trace],
+                 layout=go.Layout(
+                    title=f'<b>{title}</b><br>',
+                    titlefont_size=24,
+                    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))
+                    )
+    
+    return fig
 
 def main():
-    st.title("Organization Chart Generator")
-    st.markdown("### Enter your organizational hierarchy")
+    st.title("Professional Organization Chart Generator")
+    
+    # Title input
+    chart_title = st.text_input("Enter Chart Title", "Organization Chart")
+    
+    # File upload
+    uploaded_file = st.file_uploader("Upload CSV File", type=["csv"])
     
-    # Input format explanation
-    with st.expander("How to format your input"):
+    # Input format
+    with st.expander("CSV Format Instructions"):
         st.markdown("""
-        **Format:**  
-        `Parent Position: Child1, Child2, ...`  
-        
-        **Example:**  
-        ```
-        CEO: CTO, CFO, CMO
-        CTO: Engineering Manager, QA Manager
-        CFO: Accounting Manager, Tax Specialist
-        Engineering Manager: Senior Developer, Junior Developer
-        ```
+        **CSV Format:**  
+        Parent,Child  
+        CEO,CTO  
+        CEO,CFO  
+        CTO,Engineering Manager  
+        CFO,Accounting Manager  
         """)
     
-    # User input
-    direction = st.selectbox('Chart Direction', 
-                           ('Top-Bottom (TB)', 'Left-Right (LR)'), 
-                           index=0).split(' ')[0]
+    # Input data
+    if uploaded_file:
+        df = pd.read_csv(uploaded_file)
+    else:
+        default_data = """Parent,Child
+CEO,CTO
+CEO,CFO
+CTO,Engineering Manager
+CFO,Accounting Manager"""
+        df = pd.read_csv(StringIO(default_data))
     
-    input_data = st.text_area("Enter your organizational structure", 
-                            height=300,
-                            value="CEO: CTO, CFO\nCTO: Engineering Manager\nCFO: Accounting Manager")
+    # Show editable dataframe
+    edited_df = st.experimental_data_editor(df, num_rows="dynamic")
     
-    # Process input
+    # Generate chart
     if st.button('Generate Organization Chart'):
-        if input_data:
-            # Split and clean input
-            lines = [line.strip() for line in input_data.split('\n') if line.strip()]
+        if not edited_df.empty:
+            fig = create_org_chart(edited_df, chart_title)
+            st.plotly_chart(fig, use_container_width=True)
             
-            # Create chart
-            chart = create_org_chart(lines, direction)
+            # PDF download options
+            st.subheader("Download Options")
+            col1, col2 = st.columns(2)
+            with col1:
+                orientation = st.selectbox("PDF Orientation", ["Portrait", "Landscape"])
+            with col2:
+                paper_size = st.selectbox("Paper Size", ["A4", "Letter", "Legal"])
             
-            # Display chart
-            st.graphviz_chart(chart)
+            # 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)
+                st.download_button(
+                    label="Download PDF",
+                    data=img_bytes,
+                    file_name="org_chart.pdf",
+                    mime="application/pdf"
+                )
         else:
             st.warning("Please enter some organizational data first")