From 3216d1c6b05059bd2b49e31c3c6c31a6adbe2a40 Mon Sep 17 00:00:00 2001
From: Benjamin Berman <ben@hiddenswitch.com>
Date: Fri, 4 Aug 2023 15:43:41 -0700
Subject: [PATCH] fix image upload, running from main

---
 comfy/cmd/main.py   | 3 +++
 comfy/cmd/server.py | 4 ++--
 main.py             | 4 +++-
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/comfy/cmd/main.py b/comfy/cmd/main.py
index 5c7c21f77..88a4a6ade 100644
--- a/comfy/cmd/main.py
+++ b/comfy/cmd/main.py
@@ -170,3 +170,6 @@ def main():
         print("\nStopped server")
 
     cleanup_temp()
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file
diff --git a/comfy/cmd/server.py b/comfy/cmd/server.py
index 6cea43184..2de115142 100644
--- a/comfy/cmd/server.py
+++ b/comfy/cmd/server.py
@@ -144,7 +144,7 @@ class PromptServer():
         def get_dir_by_type(dir_type=None):
             type_dir = ""
             if dir_type is None:
-                dir_type = "../../input"
+                dir_type = "input"
 
             if dir_type == "input":
                 type_dir = folder_paths.get_input_directory()
@@ -201,7 +201,7 @@ class PromptServer():
         @routes.post("/upload/image")
         async def upload_image(request):
             post = await request.post()
-            return image_upload(post)
+            return await image_upload(post)
 
         @routes.post("/upload/mask")
         async def upload_mask(request):
diff --git a/main.py b/main.py
index 73082e1be..8649fd84c 100644
--- a/main.py
+++ b/main.py
@@ -1,2 +1,4 @@
 from comfy.cmd.main import main
-main()
\ No newline at end of file
+
+if __name__ == "__main__":
+    main()