Update runcomfy_gd.py

runcomfy_gd.py

@@ -1,4 +1,3 @@
-
 import os
 import subprocess
 import time
@@ -12,10 +11,16 @@ import select
 
 # --- 설정 변수 ---
 COMFYUI_PORT = 8188
-# 터널 스크립트 파일 경로 (ComfyUI 디렉토리 밖, 즉 부모 디렉토리)
-TUNNEL_SCRIPT_PATH = "../gradio-tunnel.py"
 
-# Gradio Tunnel Script 내용 (이전 단계에서 사용했던 내용 그대로)
+# 🚨 사용자 지정: 모든 .py 파일을 저장할 절대 경로
+# 이 경로에 gradio-tunnel.py가 저장되어야 합니다. (Colab 셀 실행 전에 /content/py 폴더가 존재해야 합니다.)
+PYTHON_SCRIPT_DIR = "/content/py"
+
+# 터널 스크립트의 최종 절대 경로
+TUNNEL_SCRIPT_PATH = os.path.join(PYTHON_SCRIPT_DIR, "gradio-tunnel.py") 
+# 예: "/content/py/gradio-tunnel.py"
+
+# Gradio Tunnel Script 내용 (터널링 기능을 수행하는 Python 코드)
 GRADIO_TUNNEL_SCRIPT_CONTENT = r"""
 import atexit
 import os
@@ -206,6 +211,14 @@ if __name__ == "__main__":
         print("오류: 포트 번호는 정수여야 합니다.", file=sys.stderr)
         sys.exit(1)
 
+    # requests 임포트 확인 (Colab 환경에서 필요)
+    try:
+        import requests
+    except ImportError:
+        # 이 코드는 runcomfy.py에서 실행하는 subprocess이므로, 
+        # runcomfy.py에서 requests 설치를 보장해야 함.
+        pass 
+
     try:
         address = setup_tunnel(
             "127.0.0.1",
@@ -217,12 +230,12 @@ if __name__ == "__main__":
         # 터널 프로세스가 종료되지 않도록 대기
         time.sleep(3600 * 24 * 3)
     except Exception as e:
-        print(f"터널 실행 중 치명적인 오류 발생: {{e}}", file=sys.stderr)
+        print(f"터널 실행 중 치명적인 오류 발생: {e}", file=sys.stderr)
         sys.exit(1)
 
 """
 
-# --- 1. main 함수 (인자 처리 간소화) ---
+# --- 1. main 함수 ---
 def main():
     # requests 라이브러리 설치를 보장합니다.
     try:
@@ -230,8 +243,8 @@ def main():
     except ImportError:
         print("필수 라이브러리 'requests'가 설치되어 있지 않습니다. 설치를 시도합니다.")
         try:
+            # 외부 라이브러리 설치를 시도
             subprocess.check_call([sys.executable, "-m", "pip", "install", "requests"])
-            # import requests # 재설치 후 재임포트가 필요하지만, main 함수 밖에서 이미 임포트 시도되었으므로 생략
         except Exception as e:
             print(f"경고: 'requests' 라이브러리 설치 실패. 터널링 오류가 발생할 수 있습니다. 오류: {e}")
 
@@ -239,25 +252,28 @@ def main():
     comfyui_args = sys.argv[1:]
 
     # --- 필수 경로 설정 및 이동 ---
+    # WS 변수는 이 스크립트를 실행하는 외부 Colab 셀에서 결정되거나, 
+    # 혹은 이 스크립트가 DRIVE 모드(/content/drive/MyDrive/ONECLICK) 전용으로 실행된다고 가정합니다.
     comfyui_dir = "/content/drive/MyDrive/ONECLICK/ComfyUI" 
-    # 또는 f-string 사용: comfyui_dir = f"{WS}/ComfyUI"
     
     print(f"ComfyUI 디렉토리로 이동 중: {comfyui_dir}")
     try:
         os.chdir(comfyui_dir)
     except FileNotFoundError:
         print(f"오류: 디렉토리가 없습니다: {comfyui_dir}. 경로를 확인해 주세요.")
+        sys.exit(1) # 디렉토리가 없으면 종료
 
-    # 터널 스크립트 파일을 저장
-    print("터널 스크립트 저장 중...")
+    # 🚨 수정: 터널 스크립트 파일을 지정된 경로에 저장 (폴더 생성 없음)
+    print(f"터널 스크립트 저장 중: {TUNNEL_SCRIPT_PATH}")
     try:
-        # TUNNEL_SCRIPT_PATH는 '../gradio-tunnel.py'이므로, CWD의 부모에 저장됨
+        # TUNNEL_SCRIPT_PATH에 저장 (절대 경로 사용)
         with open(TUNNEL_SCRIPT_PATH, "w") as f:
             f.write(GRADIO_TUNNEL_SCRIPT_CONTENT)
     except Exception as e:
-        print(f"오류: 터널 스크립트 파일 저장 실패: {e}")
+        print(f"오류: 터널 스크립트 파일 저장 실패. {PYTHON_SCRIPT_DIR} 경로가 존재하는지 확인해 주세요. 오류: {e}")
         sys.exit(1)
 
+
     # --- 2. ComfyUI 서버 실행 (백그라운드) ---
     print("\nComfyUI 서버를 백그라운드로 시작합니다...")
     comfyui_base_args = [
@@ -280,6 +296,7 @@ def main():
     time.sleep(15)
 
     print(f"\nGradio 터널 클라이언트 시작 중 (포트 {COMFYUI_PORT})...")
+    # 🚨 수정: TUNNEL_SCRIPT_PATH 변수(절대 경로) 사용
     tunnel_command = ["python", TUNNEL_SCRIPT_PATH, str(COMFYUI_PORT)]
     
     tunnel_proc = subprocess.Popen(
@@ -317,7 +334,7 @@ def main():
         print("="*50)
 
     else:
-        # --- 터널링 실패 시 오류 진단 로직 (communicate() 제거) ---
+        # --- 터널링 실패 시 오류 진단 로직 ---
         print("\n" + "="*50)
         print("🚨 **터널링 클라이언트 실행에 실패했습니다!**")
         print("아래 로그에서 오류 원인을 확인하세요.")
@@ -325,22 +342,21 @@ def main():
         
         # 1. 터널 프로세스 종료 시 남은 로그 확인 및 출력
         print("\n--- 터널 클라이언트 최종 로그 ---")
-        tunnel_proc.stdout.seek(0) # 터널 stdout의 처음으로 이동
-        tunnel_logs = tunnel_proc.stdout.read()
-        if tunnel_logs:
-             print(tunnel_logs.strip())
-        else:
-            print("터널 로그가 발견되지 않았습니다. 즉시 종료된 것으로 보입니다.")
-
-        # 2. ComfyUI 프로세스 로그 확인 (communicate 대신 read())
-        # ComfyUI의 stdout이 닫히지 않도록, 비차단 방식으로 현재까지의 로그를 읽습니다.
+        # 터널 stdout의 처음으로 이동 (이미 읽은 내용 포함)
+        try:
+            tunnel_logs = tunnel_proc.stdout.read()
+            if tunnel_logs:
+                 print(tunnel_logs.strip())
+            else:
+                print("터널 로그가 발견되지 않았습니다. 즉시 종료된 것으로 보입니다.")
+        except Exception as e:
+            print(f"터널 로그를 읽는 중 예외 발생: {e}")
+
+        # 2. ComfyUI 프로세스 로그 확인
         print("\n--- ComfyUI 초기 로그 확인 (ComfyUI가 포트를 열었는지 확인) ---")
-        # subprocess.PIPE는 기본적으로 차단 모드이므로, read()를 쓰기 전에
-        # 프로세스가 종료되지 않았다면, 현재까지 버퍼에 쌓인 로그를 최대한 출력해 봅니다.
-        # 이 시점에서는 15초 동안 대기했으므로, ComfyUI가 시작 시 출력하는 로그가 있어야 합니다.
         try:
-            # select를 사용하여 비차단 방식으로 읽을 수 있도록 시도 (Colab 환경에서 항상 작동하지는 않을 수 있음)
             if comfyui_proc.stdout is not None:
+                # select를 사용하여 비차단 방식으로 읽을 수 있도록 시도
                 ready_to_read, _, _ = select.select([comfyui_proc.stdout], [], [], 0.5)
                 if ready_to_read:
                     comfy_logs = comfyui_proc.stdout.read()
@@ -352,6 +368,12 @@ def main():
         except Exception as e:
             print(f"ComfyUI 로그 읽기 중 예외 발생: {e}")
 
+        # 모든 프로세스 종료 후 오류 발생
+        if comfyui_proc.poll() is None:
+            comfyui_proc.terminate()
+        if tunnel_proc.poll() is None:
+            tunnel_proc.terminate()
+
         raise RuntimeError("Gradio 터널링에 실패했습니다. 위의 상세 로그를 확인해 주십시오.")
 
 
@@ -395,4 +417,4 @@ def main():
         print("백그라운드 프로세스가 종료되었습니다.")
 
 if __name__ == '__main__':
-    main()
+    main()