Update app.py

app.py

@@ -103,20 +103,8 @@ def save_image(image):
         print(f"Error in save_image: {str(e)}")
         return None
 
-# 예시 프롬프트 정의 부분을 수정
-example_prompts = [
-    "A 3D Star Wars Darth Vader helmet, highly detailed metallic finish",
-    "A 3D Iron Man mask with glowing eyes and metallic red-gold finish",
-    "A 3D Pokemon Pikachu figure with glossy surface",
-    "A 3D geometric abstract cube transforming into a sphere, metallic finish",
-    "A 3D steampunk mechanical heart with brass and copper details",
-    "A 3D crystal dragon with transparent iridescent scales",
-    "A 3D futuristic hovering drone with neon light accents",
-    "A 3D ancient Greek warrior helmet with ornate details",
-    "A 3D robotic butterfly with mechanical wings and metallic finish",
-    "A 3D floating magical crystal orb with internal energy swirls"
-]
-
+def get_random_seed():
+    return torch.randint(0, 1000000, (1,)).item()
 
 @spaces.GPU
 def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt="", seed=None):
@@ -157,20 +145,19 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
             print(f"Error in image generation: {str(e)}")
             return None
 
-def get_random_seed():
-    return torch.randint(0, 1000000, (1,)).item()
-
-def process_example(prompt):
+def generate_image(h, w, s, sc, p, sd):
+    """Gradio에서 넘겨받은 파라미터 순서에 맞춰 이미지 생성 함수 호출"""
     return process_and_save_image(
-        height=1024,
-        width=1024,
-        steps=8,
-        scales=3.5,
-        prompt=prompt,
-        seed=get_random_seed()
+        height=h,
+        width=w,
+        steps=s,
+        scales=sc,
+        prompt=p,
+        seed=sd
     )
 
-# example_prompts 변수와 process_example 함수 제거
+def update_random_seed():
+    return gr.Number.update(value=get_random_seed())
 
 # Gradio 인터페이스
 with gr.Blocks(
@@ -286,7 +273,7 @@ with gr.Blocks(
         gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
         gallery_html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>"
         
-        for img_file, prompt in SAMPLE_IMAGES.items():
+        for img_file, sample_prompt in SAMPLE_IMAGES.items():
             if os.path.exists(img_file):
                 gallery_html += f"""
                 <div style='
@@ -306,46 +293,25 @@ with gr.Blocks(
                         font-weight: bold;
                         color: #333;
                         padding: 10px;
-                    '>Prompt: {prompt}</p>
+                    '>Prompt: {sample_prompt}</p>
                 </div>
                 """
         
         gallery_html += "</div>"
         gr.HTML(gallery_html)
 
-# 이벤트 핸들러 부분만 수정
-    def generate_image(h, w, s, sc, p, sd):
-        return process_and_save_image(
-            height=h,
-            width=w,
-            steps=s,
-            scales=sc,
-            prompt=p,
-            seed=sd
-        )
-
-    def update_random_seed():
-        return gr.Number.update(value=get_random_seed())
-
     # 이벤트 바인딩
     generate_btn.click(
-        generate_image,
-        inputs={
-            "h": height,
-            "w": width,
-            "s": steps,
-            "sc": scales,
-            "p": prompt,
-            "sd": seed
-        },
+        fn=generate_image,
+        inputs=[height, width, steps, scales, prompt, seed],  # 딕셔너리 대신 순서대로 리스트
         outputs=output
     )
 
     randomize_seed.click(
-        update_random_seed,
+        fn=update_random_seed,
         inputs=None,
         outputs=seed
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR])
+    demo.launch(allowed_paths=[PERSISTENT_DIR])