Update app.py

app.py

@@ -1,12 +1,16 @@
-import gradio as gr
-import pandas as pd
 import os
 import logging
+import gradio as gr
+import pandas as pd
 import base64
 import pdfkit
 import shutil
 from graphviz import Digraph
 
+# Debug prints for environment variables
+print("PATH =", os.environ.get("PATH"))
+print("GRAPHVIZ_DOT =", os.environ.get("GRAPHVIZ_DOT"))
+
 # Set up logging for debugging
 logging.basicConfig(level=logging.INFO)
 
@@ -20,11 +24,6 @@ def get_wkhtmltopdf_path():
     return None
 
 def process_uploaded_file(file):
-    """
-    Process the file input from Gradio.
-    If the file is a dict, write its "data" content to a temporary file.
-    If it's a string, ensure it is not a directory.
-    """
     logging.info(f"Processing uploaded file: {file}")
     if isinstance(file, str):
         if os.path.abspath(file) == os.getcwd():
@@ -77,39 +76,37 @@ def generate_chart(file, title):
         logging.error(f"Error reading CSV: {e}")
         return None, f"Error reading CSV: {e}"
     
-    # Validate required columns
     expected_columns = {"Name", "Role", "Reporting To"}
     if not expected_columns.issubset(set(df.columns)):
         return None, "CSV must contain Name, Role, and Reporting To columns."
     
-    # Create a Graphviz Digraph for the organization chart.
+    # Create a Graphviz Digraph for the org chart.
     dot = Digraph(comment=title, format='png')
-    dot.attr(rankdir='TB')  # Top-to-Bottom layout
+    dot.attr(rankdir='TB')
     dot.attr('node', shape='box', style='rounded,filled', fillcolor='lightblue',
              fontsize='10', fontname='Helvetica', margin='0.1')
     
-    # Add nodes: using "Name" as the unique identifier.
+    # Add nodes.
     for idx, row in df.iterrows():
-        node_id = row['Name']  # Assumes names are unique
+        node_id = row['Name']  # Assumes names are unique.
         label = f"{row['Name']}\n({row['Role']})"
         dot.node(node_id, label=label)
     
-    # Add edges: from manager (Reporting To) to subordinate.
+    # Add edges.
     for idx, row in df.iterrows():
         if pd.notna(row['Reporting To']) and row['Reporting To'] != "":
             manager = row['Reporting To']
             subordinate = row['Name']
-            # If the manager is not in the CSV, add them as a node with just their name.
             if manager not in df['Name'].values:
                 dot.node(manager, label=manager)
             dot.edge(manager, subordinate)
     
-    # Render the diagram to a PNG file.
+    # Render diagram to PNG.
     output_path = "/tmp/chart"
     dot.render(output_path, cleanup=True)
     chart_path = output_path + ".png"
     
-    # Create an HTML template embedding the chart image.
+    # Build HTML content with embedded chart image.
     with open(chart_path, "rb") as img_file:
         encoded = base64.b64encode(img_file.read()).decode()
     img_html = f"<img src='data:image/png;base64,{encoded}' style='width:100%;'/>"
@@ -126,7 +123,7 @@ def generate_chart(file, title):
     </html>
     """
     
-    # Generate PDF output using wkhtmltopdf (if available).
+    # Generate PDF.
     wkhtml_path = get_wkhtmltopdf_path()
     if wkhtml_path is None:
         logging.error("wkhtmltopdf not found, creating dummy PDF file.")