Update app.py

app.py

@@ -7,23 +7,31 @@ from pdf2image import convert_from_path
 
 logging.basicConfig(level=logging.INFO)
 
-# Name of the default CSV file (must be in your repository)
-DEFAULT_CSV = "default.csv"
+DEFAULT_CSV = "default.csv"  # Must be included in your repository
 
 def load_csv(file):
     """
-    Load the CSV file.
-    If a valid file path is provided and it does not equal the working directory,
-    use it. Otherwise, fall back to the default CSV.
+    If a valid file path is provided and it is not equal to the working directory,
+    load that CSV. Otherwise, load the default CSV.
     """
-    if file and isinstance(file, str) and os.path.isfile(file) and file != os.getcwd():
-        logging.info(f"Loading uploaded file: {file}")
+    # If file is empty, None, or equals the current working directory, use default.
+    cwd = os.getcwd()
+    if not file or file == "" or file.strip() == "" or file == cwd:
+        logging.info("No valid file provided; using default CSV.")
+        if os.path.isfile(DEFAULT_CSV):
+            return pd.read_csv(DEFAULT_CSV)
+        else:
+            raise FileNotFoundError("Default CSV not found.")
+    # Otherwise, if the file exists and is not a directory, use it.
+    if os.path.isfile(file):
+        logging.info(f"Loading uploaded CSV: {file}")
         return pd.read_csv(file)
-    elif os.path.isfile(DEFAULT_CSV):
-        logging.info("No valid file uploaded; using default CSV.")
-        return pd.read_csv(DEFAULT_CSV)
     else:
-        raise FileNotFoundError("No CSV file provided and default CSV not found.")
+        logging.warning(f"Provided file path '{file}' is not valid. Using default CSV.")
+        if os.path.isfile(DEFAULT_CSV):
+            return pd.read_csv(DEFAULT_CSV)
+        else:
+            raise FileNotFoundError("No valid CSV file and default CSV not found.")
 
 def build_tree(df):
     employees = {}
@@ -164,6 +172,7 @@ def generate_chart(file, title):
         HTML(string=html_content).write_pdf(pdf_path)
         logging.info("PDF generated successfully.")
     except Exception as e:
+        logging.error(f"Error generating PDF: {e}")
         return None, f"Error generating PDF: {e}"
     
     try:
@@ -174,13 +183,15 @@ def generate_chart(file, title):
         else:
             image_path = ""
     except Exception as e:
+        logging.error(f"Error converting PDF to image: {e}")
         image_path = ""
     
     return image_path, pdf_path
 
 with gr.Blocks() as demo:
     gr.Markdown("## Organization Chart Generator")
-    gr.Markdown("Upload a CSV file (optional). If no file is uploaded, the default CSV (default.csv) will be used.")
+    gr.Markdown("Upload a CSV file (optional). If no file is uploaded or the file is invalid, the default CSV will be used.")
+    
     file_input = gr.File(label="Upload CSV File (optional)", type="filepath")
     title_input = gr.Textbox(label="Enter PDF Title", placeholder="Company Org Chart")
     submit_button = gr.Button("Generate Chart")