Update app.py

app.py

@@ -27,9 +27,10 @@ pipe.to(device)
 # 이미지 생성 함수 정의
 @spaces.GPU(duration=160)
 def generate_image(prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt, progress=gr.Progress(track_tqdm=True)):
-    if seed == 0:
+    # 시드가 0이거나 None이면 랜덤 시드 사용
+    if seed is None or seed == 0:
         seed = random.randint(1, MAX_SEED)
-
+    
     generator = torch.Generator().manual_seed(seed)
     
     with torch.inference_mode():
@@ -45,6 +46,13 @@ def generate_image(prompt, num_inference_steps, height, width, guidance_scale, s
     
     return output
 
+# 랜덤 시드 생성 함수
+def random_seed():
+    return random.randint(1, MAX_SEED)
+
+def create_random_seed():
+    return gr.Number.update(value=random_seed())
+
 # 예제 프롬프트
 examples = [
     ["A cat holding a sign that says hello world"],
@@ -139,6 +147,17 @@ h1 {
     overflow: hidden;
     border: 2px solid #F3F4F6;
 }
+/* 랜덤 시드 버튼 스타일 */
+button.gr-button.gr-button-lg {
+    border-radius: 8px;
+    font-weight: 500;
+    transition: all 0.2s;
+}
+button.gr-button.gr-button-lg:hover {
+    transform: translateY(-1px);
+    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+}
+'''
 '''
 
 # Gradio 인터페이스 생성
@@ -195,7 +214,11 @@ with gr.Blocks(
                 minimum=1,
                 maximum=50,
                 value=25,
-                step=1
+                step=1,
+                interactive=True,
+                show_label=True,
+                container=True,
+                randomize=False
             )
             guidance_scale = gr.Slider(
                 label="가이던스 스케일",
@@ -203,7 +226,11 @@ with gr.Blocks(
                 minimum=0.0,
                 maximum=7.0,
                 value=3.5,
-                step=0.1
+                step=0.1,
+                interactive=True,
+                show_label=True,
+                container=True,
+                randomize=False
             )
         
         with gr.Row():
@@ -213,7 +240,11 @@ with gr.Blocks(
                 minimum=256,
                 maximum=1024,
                 step=32,
-                value=1024
+                value=1024,
+                interactive=True,
+                show_label=True,
+                container=True,
+                randomize=False
             )
             height = gr.Slider(
                 label="높이",
@@ -221,19 +252,23 @@ with gr.Blocks(
                 minimum=256,
                 maximum=1024,
                 step=32,
-                value=1024
+                value=1024,
+                interactive=True,
+                show_label=True,
+                container=True,
+                randomize=False
             )
         
         with gr.Row():
-            seed = gr.Slider(
-                value=random.randint(1, MAX_SEED),  # 초기값을 랜덤으로 설정
+            seed = gr.Number(
+                value=random_seed(),
                 minimum=0,
                 maximum=MAX_SEED,
                 step=1,
                 label="시드",
-                info="생성 과정의 시작점입니다. 0을 입력하면 랜덤한 시드가 사용됩니다. 🔄 버튼을 클릭하면 랜덤한 값으로 변경됩니다.",
-                randomize=True  # 이 옵션을 추가하면 🔄 버튼이 작동합니다
+                info="생성 과정의 시작점입니다. 0을 입력하면 랜덤한 시드가 사용됩니다."
             )
+            seed_button = gr.Button("🔄 랜덤 시드", size="sm")
             num_images_per_prompt = gr.Slider(
                 label="프롬프트당 이미지 수",
                 info="설정된 값으로 생성할 이미지의 수",
@@ -251,6 +286,11 @@ with gr.Blocks(
         cache_examples=CACHE_EXAMPLES
     )
 
+    seed_button.click(
+        fn=create_random_seed,
+        outputs=[seed]
+    )
+
     gr.on(
         triggers=[
             prompt.submit,