Update app.py

app.py

@@ -310,12 +310,31 @@ with gr.Blocks(
         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.update(value=get_random_seed())
 
     # 이벤트 바인딩
     generate_btn.click(
-        fn=generate_image,
-        inputs=[height, width, steps, scales, prompt, seed],  # 딕셔너리 대신 순서대로 리스트
+        fn=process_and_save_image,  # 직접 process_and_save_image 함수 사용
+        inputs={
+            "height": height,
+            "width": width,
+            "steps": steps,
+            "scales": scales,
+            "prompt": prompt,
+            "seed": seed
+        },
         outputs=output
     )
 
@@ -326,4 +345,4 @@ with gr.Blocks(
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR])
+    demo.launch(allowed_paths=[PERSISTENT_DIR])