Update app.py

app.py

@@ -5,54 +5,73 @@ import os
 import base64
 from io import StringIO
 
-# Add Graphviz to system PATH (critical for Hugging Face Spaces)
+# Fix Graphviz PATH error for Hugging Face
 os.environ["PATH"] += os.pathsep + '/usr/bin/graphviz'
 
-# Custom CSS for professional UI
+# Enhanced UI CSS
 st.markdown("""
 <style>
-    .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;
+    .stApp {
+        background: #f8f9fa;
+        font-family: 'Segoe UI', system-ui;
     }
-    .stDownloadButton button:hover {
+    .st-emotion-cache-1kyxreq {
+        justify-content: center;
+    }
+    .stButton>button {
+        background: linear-gradient(135deg, #4CAF50, #45a049);
+        color: white;
+        border-radius: 8px;
+        padding: 12px 24px;
+        border: none;
+        font-weight: 600;
+        transition: transform 0.3s;
+    }
+    .stButton>button:hover {
         transform: scale(1.05);
-        box-shadow: 0 4px 8px rgba(0,0,0,0.2);
+        box-shadow: 0 4px 12px rgba(76,175,80,0.3);
+    }
+    .stDownloadButton>button {
+        background: linear-gradient(135deg, #2196F3, #1976D2);
+    }
+    .st-eb {
+        border-radius: 10px;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.1);
     }
 </style>
 """, unsafe_allow_html=True)
 
 def create_org_chart(df, title):
-    """Create professional organization chart with Graphviz"""
+    """Create professional org chart with straight lines"""
     dot = Digraph(comment=title)
     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')
-    dot.attr('edge', color='#666666', arrowsize='0.8')
+            fontsize='24', fontname='Arial', margin='0.5',
+            splines='ortho')  # Force straight lines
+    
+    # Node styling
+    dot.attr('node', shape='box', style='filled',
+            fillcolor='linear-gradient(#4CAF50, #45a049)', 
+            fontname='Arial', fontcolor='white',
+            margin='0.4', width='2', height='0.9',
+            fixedsize='false', fontsize='14')
+    
+    # Edge styling for straight lines
+    dot.attr('edge', color='#666666', arrowsize='0.8', 
+            penwidth='1.2', dir='forward')
 
     # Process relationships
-    relationships = set()
+    added_nodes = set()
     for _, row in df.iterrows():
         parent = row['Parent'].strip()
         child = row['Child'].strip()
         if parent and child:
-            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)
+            if parent not in added_nodes:
+                dot.node(parent)
+                added_nodes.add(parent)
+            if child not in added_nodes:
+                dot.node(child)
+                added_nodes.add(child)
+            dot.edge(parent, child)
     
     return dot
 
@@ -60,42 +79,57 @@ def main():
     st.title("🏢 Professional Organization Chart Generator")
     
     with st.sidebar:
-        st.header("Configuration")
+        st.header("⚙️ Configuration")
         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")
+        st.markdown("---")
+        st.markdown("### 📋 CSV Instructions")
+        st.write("1. Download template\n2. Edit in Excel/Sheets\n3. Upload your file")
         
         # 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>'
+        href = f'<a href="data:file/csv;base64,{b64}" download="template.csv" style="text-decoration: none; color: white; background: #4CAF50; padding: 10px 20px; border-radius: 8px; display: inline-block; margin-top: 10px;">📥 Download Template</a>'
         st.markdown(href, unsafe_allow_html=True)
 
-    uploaded_file = st.file_uploader("Upload CSV File", type=["csv"])
+    uploaded_file = st.file_uploader("📤 Upload CSV File", type=["csv"], 
+                                   help="Upload your organizational hierarchy CSV file")
     
     if uploaded_file:
         try:
             df = pd.read_csv(uploaded_file)
             if {'Parent', 'Child'}.issubset(df.columns):
-                with st.spinner("Generating professional chart..."):
+                with st.spinner("🔍 Analyzing structure..."):
                     chart = create_org_chart(df, chart_title)
                     
                     # Display chart
-                    st.graphviz_chart(chart, use_container_width=True)
+                    col1, col2 = st.columns([3, 1])
+                    with col1:
+                        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"
-                    )
+                    with col2:
+                        st.markdown("### 📤 Export Options")
+                        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="Download high-quality PDF version"
+                        )
+                        
+                        png_bytes = chart.pipe(format='png')
+                        st.download_button(
+                            label="🖼️ Download PNG",
+                            data=png_bytes,
+                            file_name=f"{chart_title.replace(' ', '_')}.png",
+                            mime="image/png",
+                            help="Download image version"
+                        )
             else:
-                st.error("CSV must contain 'Parent' and 'Child' columns")
+                st.error("❌ CSV must contain 'Parent' and 'Child' columns")
         except Exception as e:
-            st.error(f"Error processing file: {str(e)}")
+            st.error(f"🚨 Error processing file: {str(e)}")
 
 if __name__ == "__main__":
     main()