Update app.py

app.py

@@ -1,74 +1,26 @@
 import streamlit as st
 import pandas as pd
 from graphviz import Digraph
+import os
 import base64
 from io import StringIO
 
+# Add Graphviz to system PATH (critical for Hugging Face Spaces)
+os.environ["PATH"] += os.pathsep + '/usr/bin/graphviz'
+
 # Custom CSS for professional UI
 st.markdown("""
 <style>
-    /* Main container */
-    .stApp {
-        background-color: #f9f9f9;
-        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-    }
-    
-    /* Titles */
-    h1 {
-        color: #2c3e50;
-        border-bottom: 3px solid #4CAF50;
-        padding-bottom: 10px;
-    }
-    
-    /* Sidebar */
-    .stSidebar {
-        background-color: #2c3e50;
+    .stDownloadButton button {
+        background: linear-gradient(45deg, #4CAF50, #45a049) !important;
         color: white !important;
+        border-radius: 25px !important;
+        padding: 12px 28px !important;
+        transition: all 0.3s !important;
     }
-    
-    /* Buttons */
-    .stButton>button {
-        background-color: #4CAF50;
-        color: white;
-        border-radius: 25px;
-        padding: 12px 28px;
-        border: none;
-        font-size: 16px;
-        transition: all 0.3s;
-    }
-    .stButton>button:hover {
-        background-color: #45a049;
+    .stDownloadButton button:hover {
         transform: scale(1.05);
-    }
-    
-    /* File uploader */
-    .stFileUploader>div>div>div>div {
-        background-color: #ffffff;
-        border-radius: 10px;
-        padding: 15px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    
-    /* Data editor */
-    .stDataEditor {
-        background-color: white;
-        border-radius: 10px;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    
-    /* Tabs */
-    .stTabs [data-baseweb="tab-list"] {
-        gap: 10px;
-    }
-    .stTabs [data-baseweb="tab"] {
-        background-color: #e8f5e9;
-        border-radius: 8px;
-        padding: 12px 24px;
-        transition: all 0.3s;
-    }
-    .stTabs [aria-selected="true"] {
-        background-color: #4CAF50;
-        color: white !important;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.2);
     }
 </style>
 """, unsafe_allow_html=True)
@@ -76,97 +28,74 @@ st.markdown("""
 def create_org_chart(df, title):
     """Create professional organization chart with Graphviz"""
     dot = Digraph(comment=title)
-    dot.attr(rankdir='TB', labelloc='t', label=title, fontsize='24', fontname='Arial')
+    dot.attr(rankdir='TB', labelloc='t', label=title, 
+            fontsize='24', fontname='Arial', margin='0.5')
     dot.attr('node', shape='box', style='filled', 
-            fillcolor='linear-gradient(#4CAF50, #45a049)', fontname='Arial',
-            fontcolor='white', margin='0.3', width='2', height='0.8',
-            fixedsize='false', color='#2c3e50', fontsize='14')
+           fillcolor='linear-gradient(#4CAF50, #45a049)', fontname='Arial',
+           fontcolor='white', margin='0.3', width='2', height='0.8',
+           fixedsize='false', color='#2c3e50', fontsize='14')
     dot.attr('edge', color='#666666', arrowsize='0.8')
 
-    # Track all nodes and relationships
-    nodes = set()
-    edges = set()
-
+    # Process relationships
+    relationships = set()
     for _, row in df.iterrows():
         parent = row['Parent'].strip()
         child = row['Child'].strip()
-        
         if parent and child:
-            nodes.add(parent)
-            nodes.add(child)
-            edges.add((parent, child))
-
-    # Add nodes first to maintain order
+            relationships.add((parent, child))
+    
+    # Add nodes and edges
+    nodes = set()
+    for parent, child in relationships:
+        nodes.add(parent)
+        nodes.add(child)
+        dot.edge(parent, child)
+    
     for node in nodes:
         dot.node(node, node)
-        
-    # Add edges
-    for edge in edges:
-        dot.edge(edge[0], edge[1])
-
+    
     return dot
 
-def download_csv_template():
-    """Generate CSV template download link"""
-    csv_template = "Parent,Child\nCEO,CTO\nCEO,CFO\nCTO,Engineering Manager\nCFO,Accounting Manager"
-    b64 = base64.b64encode(csv_template.encode()).decode()
-    href = f'<a href="data:file/csv;base64,{b64}" download="org_chart_template.csv" style="\
-        background-color: #4CAF50; color: white; padding: 12px 24px; \
-        border-radius: 25px; text-decoration: none;">Download CSV Template</a>'
-    st.markdown(href, unsafe_allow_html=True)
-
 def main():
     st.title("🏢 Professional Organization Chart Generator")
     
     with st.sidebar:
         st.header("Configuration")
-        chart_title = st.text_input("Chart Title", "Company Organization Chart")
-        orientation = st.selectbox("PDF Orientation", ["Portrait", "Landscape"])
-        paper_size = st.select_slider("Paper Size", options=["A3", "A4", "Letter", "Legal"])
-
-    tab1, tab2 = st.tabs(["📥 Data Input", "📊 Organization Chart"])
-
-    with tab1:
-        st.subheader("1. Prepare Your Data")
-        download_csv_template()
-        
-        uploaded_file = st.file_uploader("Upload CSV File", type=["csv"], 
-                                        help="Upload your organizational data in CSV format")
+        chart_title = st.text_input("Chart Title", "Company Structure")
+        st.markdown("### CSV Instructions")
+        st.write("1. Download template below\n2. Edit in Excel/Sheets\n3. Upload your version")
         
-        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))
-        
-        edited_df = st.data_editor(df, num_rows="dynamic", use_container_width=True)
+        # CSV template download
+        template = "Parent,Child\nCEO,CTO\nCEO,CFO\nCTO,Engineering Manager"
+        b64 = base64.b64encode(template.encode()).decode()
+        href = f'<a href="data:file/csv;base64,{b64}" download="template.csv">📥 Download Template</a>'
+        st.markdown(href, unsafe_allow_html=True)
 
-    with tab2:
-        if st.button("Generate Organization Chart", type="primary"):
-            if not edited_df.empty:
+    uploaded_file = st.file_uploader("Upload CSV File", type=["csv"])
+    
+    if uploaded_file:
+        try:
+            df = pd.read_csv(uploaded_file)
+            if {'Parent', 'Child'}.issubset(df.columns):
                 with st.spinner("Generating professional chart..."):
-                    try:
-                        # Create and display chart
-                        chart = create_org_chart(edited_df, chart_title)
-                        st.graphviz_chart(chart, use_container_width=True)
-                        
-                        # PDF Generation
-                        pdf_bytes = chart.pipe(format='pdf')
-                        st.download_button(
-                            label="📥 Download PDF",
-                            data=pdf_bytes,
-                            file_name=f"{chart_title.replace(' ', '_')}.pdf",
-                            mime="application/pdf",
-                            help=f"Download {orientation} PDF ({paper_size} size)"
-                        )
-                    except Exception as e:
-                        st.error(f"Error generating chart: {str(e)}")
+                    chart = create_org_chart(df, chart_title)
+                    
+                    # Display chart
+                    st.graphviz_chart(chart, use_container_width=True)
+                    
+                    # PDF Generation
+                    pdf_bytes = chart.pipe(format='pdf')
+                    st.download_button(
+                        label="📥 Download PDF Report",
+                        data=pdf_bytes,
+                        file_name=f"{chart_title.replace(' ', '_')}.pdf",
+                        mime="application/pdf",
+                        help="Download high-quality PDF version"
+                    )
             else:
-                st.warning("Please input organizational data first!")
+                st.error("CSV must contain 'Parent' and 'Child' columns")
+        except Exception as e:
+            st.error(f"Error processing file: {str(e)}")
 
 if __name__ == "__main__":
     main()