Update app.py

app.py

@@ -94,18 +94,20 @@ def apply_watermark(image_pil, wm_text="123", password_img=123, password_wm=456)
     return result_image
 
 # 전체 이미지 처리 함수
-def process_image(image, eps_value):
+def process_image(image, eps_value, wm_text, password_img, password_wm):
     # 적대적 이미지 생성
     adv_image = generate_adversarial_image(image, eps_value)
-
-    # 워터마크 적용
-    watermarked_image = apply_watermark(adv_image)
-
+    watermarked_image = apply_watermark(image, wm_text, int(password_img), int(password_wm))
     return watermarked_image
 
 # Gradio 인터페이스 정의
 gr.Interface(
     fn=process_image,
-    inputs=[gr.Image(type="pil"), gr.components.Slider(0.1, 1.0, step=0.1, value=0.3, label="Epsilon")],
-    outputs="image"
+    inputs=[gr.Image(type="pil", label="이미지를 업로드하세요"),  # 이미지 업로드 필드
+        gr.Slider(0.1, 1.0, step=0.1, value=0.3, label="Epsilon 값 설정 (노이즈 강도)"),  # epsilon 값 슬라이더
+        gr.Textbox(label="워터마크 텍스트 입력", value="123"),  # 워터마크 텍스트 입력 필드
+        gr.Number(label="이미지 비밀번호", value=123),  # 이미지 비밀번호 입력 필드
+        gr.Number(label="워터마크 비밀번호", value=456)  # 워터마크 비밀번호 입력 필드
+    ],
+    outputs=gr.Image(type="pil", label="워터마크가 삽입된 이미지")  # 결과 이미지 출력 필드
 ).launch()