Update app.py

app.py

@@ -74,36 +74,42 @@ except Exception as e:
 @torch.inference_mode()
 def generate_image(
     prompt: str,
-    id_image = None,
-    width: int = 512,
-    height: int = 512,
-    num_steps: int = 20,
-    start_step: int = 0,
-    guidance: float = 4.0,
-    seed: int = -1,
-    id_weight: float = 1.0,
-    neg_prompt: str = "",
-    true_cfg: float = 1.0,
-    timestep_to_start_cfg: int = 1,
-    max_sequence_length: int = 128,
-    gamma: float = 0.5,
-    eta: float = 0.7,
-    s: float = 0,
-    tau: float = 5,
+    id_image,
+    num_steps: int,
+    guidance: float,
+    seed,
+    id_weight: float,
+    neg_prompt: str,
+    true_cfg: float,
+    gamma: float,
+    eta: float,
 ):
     # 모델이 초기화되지 않았으면 오류 메시지 반환
     if not model_initialized:
-        return None, "GPU 오류: CUDA GPU를 찾을 수 없어 모델을 초기화할 수 없습니다.", None
+        return None, "GPU 오류: CUDA GPU를 찾을 수 없어 모델을 초기화할 수 없습니다."
     
     # ID 이미지가 없으면 실행 불가
     if id_image is None:
-        return None, "오류: ID 이미지가 필요합니다.", None
+        return None, "오류: ID 이미지가 필요합니다."
     
     try:
+        # 고정 매개변수
+        width = 512
+        height = 512
+        start_step = 0
+        timestep_to_start_cfg = 1
+        max_sequence_length = 128
+        s = 0
+        tau = 5
+        
         flux_generator.t5.max_length = max_sequence_length
 
         # 시드 설정
-        seed = int(seed)
+        try:
+            seed = int(seed)
+        except:
+            seed = -1
+            
         if seed == -1:
             seed = None
 
@@ -231,17 +237,17 @@ def generate_image(
         edited = rearrange(edited[0], "c h w -> h w c")
         edited = Image.fromarray((127.5 * (edited + 1.0)).cpu().byte().numpy())
 
-        return edited, str(opts.seed), flux_generator.pulid_model.debug_img_list
+        return edited, str(opts.seed)
     
     except Exception as e:
         import traceback
         error_msg = f"이미지 생성 중 오류 발생: {str(e)}\n{traceback.format_exc()}"
         print(error_msg)
-        return None, error_msg, None
+        return None, error_msg
 
 
 def create_demo():
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme="apriel") as demo:
         gr.Markdown("# PuLID: 인물 이미지 변환 도구")
         
         if not model_initialized:
@@ -260,7 +266,7 @@ def create_demo():
                 with gr.Accordion("고급 옵션", open=False):
                     neg_prompt = gr.Textbox(label="네거티브 프롬프트", value="")
                     true_cfg = gr.Slider(1.0, 10.0, 3.5, step=0.1, label="CFG 스케일")
-                    seed = gr.Textbox(-1, label="시드 (-1: 랜덤)")
+                    seed = gr.Textbox(value="-1", label="시드 (-1: 랜덤)")
                     gr.Markdown("### 기타 옵션")
                     gamma = gr.Slider(0.0, 1.0, 0.5, step=0.1, label="감마")
                     eta = gr.Slider(0.0, 1.0, 0.8, step=0.1, label="에타")
@@ -272,10 +278,13 @@ def create_demo():
                 seed_output = gr.Textbox(label="결과/오류 메시지")
                 gr.Markdown(_CITE_)
 
+        # Gradio 이벤트 연결 수정 - 직접 정수 값을 전달하지 않고 UI 컴포넌트만 전달
         generate_btn.click(
             fn=generate_image,
-            inputs=[prompt, id_image, 512, 512, num_steps, 0, guidance, seed, id_weight, neg_prompt,
-                    true_cfg, 1, 128, gamma, eta, 0, 5],
+            inputs=[
+                prompt, id_image, num_steps, guidance, seed, 
+                id_weight, neg_prompt, true_cfg, gamma, eta
+            ],
             outputs=[output_image, seed_output],
         )