Fix IsADirectoryError in main_no_modal.py

Fixes smol-ai#26
MaskyS · Jun 6, 2023 · 8d50a73 · 8d50a73
1 parent e0a11ad
commit 8d50a73
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/main_no_modal.py b/main_no_modal.py
@@ -193,8 +193,14 @@ def write_file(filename, filecode, directory):
     print("\033[94m" + filename + "\033[0m")
     print(filecode)
 
-    file_path = directory + "/" + filename
+    file_path = os.path.join(directory, filename)
     dir = os.path.dirname(file_path)
+
+    # Check if the filename is actually a directory
+    if os.path.isdir(file_path):
+        print(f"Error: {filename} is a directory, not a file.")
+        return
+
     os.makedirs(dir, exist_ok=True)
 
     # Open the file in write mode