Update app.py

app.py

@@ -14,8 +14,8 @@ st.markdown("<span style='font-size: 18px;'>안녕하세요! 저희는 딥페이
 st.markdown("<span style='font-size: 18px;'>저희의 목표는 온라인에 게시된 개인의 사진이 악성 딥페이크 영상에 사용되지 않도록 하는 것입니다. 정식 론칭에 앞서, 여러분의 의견을 듣기 위해 간단한 테스트를 진행하고 있습니다.</span>", unsafe_allow_html=True)
 st.markdown("<span style='font-size: 18px;'>최근 SNS에 업로드된 이미지가 딥페이크에 악용되는 사례가 매일 보고되고 있습니다. 따라서 해결책을 강구하기 위해, 여러분의 소중한 의견이 필요합니다.</span>", unsafe_allow_html=True)
 st.markdown("")
-st.markdown("<span style='font-size: 18px;'>진행 방법 1. 이미지를 업로드하면, 사전 방지 필터가 적용된 이미지를 보여드립니다.</span>", unsafe_allow_html=True)
-st.markdown("<span style='font-size: 18px;'>진행 방법 2. 흰 버튼을 클릭하면 딥페이크 결과를 확인할 수 있습니다.</span>", unsafe_allow_html=True)
+st.markdown("<span style='font-size: 18px;'>진행 과정 1. 사람 이미지를 업로드하면, 사전 방지 필터가 적용된 이미지를 보여드립니다.</span>", unsafe_allow_html=True)
+st.markdown("<span style='font-size: 18px;'>진행 과정 2. 딥페이크 실행 버튼을 클릭하면, 딥페이크 결과를 확인할 수 있습니다.</span>", unsafe_allow_html=True)
 st.markdown("<span style='font-size: 18px;'>여러분의 참여에 감사드립니다!</span>", unsafe_allow_html=True)
 st.markdown("<span style='font-size: 14px;'> *사전 방지 필터란: 사진에 미세한 변화를 주어 딥페이크 모델에 학습되지 못하도록 방해하는 필터.</span>", unsafe_allow_html=True)
 st.markdown(
@@ -219,47 +219,28 @@ if uploaded_file is not None:
     
     st.markdown('<div class="custom-caption-1">필터를 입힌 이미지는 육안으로 봤을때,큰 차이가 없음</div>', unsafe_allow_html=True)
 
-    button_clicked = st.button("상단의 두 사진을 딥페이크 모델에 학습시키기")
+    button_clicked = st.button("딥페이크 적용하기")
     
-    # if button_clicked:
-    #     with col1:
-    #         processed_image = change_hair_to_blonde(image)
-    #         st.image(processed_image, use_column_width=True)
-    #         st.markdown('<div class="custom-caption-2">원본 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
-    #         st.markdown('<div class="custom-caption-3">이해를 돕기 위해 사진에 노란색을 입히는 딥페이크 알고리즘 적용. 원본 이미지는 딥페이크 알고리즘의 영향을 받음.</div>', unsafe_allow_html=True)
-        
-    #     with col2:
-    #         deepfake_image = add_noise(image)
-    #         st.image(deepfake_image, use_column_width=True)
-    #         st.markdown('<div class="custom-caption-2">방지 필터 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
-    #         st.markdown('<div class="custom-caption-3">방지 필터를 입힌 이미지는 딥페이크의 영향을 받지 않고 알아보기 힘든 사진을 출력함으로써 이미지를 보호함.</div>', unsafe_allow_html=True)
-
     col3, col4 = st.columns(2)
     
     if button_clicked:
         processed_image = change_hair_to_blonde(image)
         deepfake_image = add_noise(image)
     else:
-        processed_image = image
-        deepfake_image = image
-
-    with col3:
-        st.image(processed_image, use_column_width=True)
-        st.markdown('<div class="custom-caption-2">원본 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
-        st.markdown('<div class="custom-caption-3">이해를 돕기 위해 사진에 노란색을 입히는 딥페이크 알고리즘 적용. 원본 이미지는 딥페이크 알고리즘의 영향을 받음.</div>', unsafe_allow_html=True)
-    
-    with col4:
-        st.image(deepfake_image, use_column_width=True)
-        st.markdown('<div class="custom-caption-2">방지 필터 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
-        st.markdown('<div class="custom-caption-3">방지 필터를 입힌 이미지는 딥페이크의 영향을 받지 않고 알아보기 힘든 사진을 출력함으로써 이미지를 보호함.</div>', unsafe_allow_html=True)
-    
-    st.markdown('<p class="survey">위 서비스를 사용해 보셨거나, 저희 기술적 원리에 관심이 있으신 분들께선 아래의 간단한 인터뷰에 참여해 주시면 진심으로 감사드리겠습니다.</p>', unsafe_allow_html=True)
-    st.markdown('<p class="survey-1"><a href="https://docs.google.com/forms/d/e/1FAIpQLSdzRtuvQyp3CQDhlxEag40v2yDM7u9NYpJ2gv5kgwuNbo1gUA/viewform?usp=sf_link" target="_blank" class="a-tag">여기를 클릭하여 인터뷰에 응해 주신다면 큰 도움이 될 것 같습니다!!</a></p>', unsafe_allow_html=True)
-    st.markdown('<p class="survey-2">서비스를 이용해 주셔서 감사합니다! 좋은 하루 보내세요!</p>', unsafe_allow_html=True)
-
-
-    
-        # st.markdown('<p class="survey">위 서비스를 사용해 보셨거나, 저희 기술적 원리에 관심이 있으신 분들께선 아래의 간단한 인터뷰에 참여해 주시면 진심으로 감사드리겠습니다.</p>', unsafe_allow_html=True)
-        # st.markdown('<p class="survey-1"><a href="https://docs.google.com/forms/d/e/1FAIpQLSdzRtuvQyp3CQDhlxEag40v2yDM7u9NYpJ2gv5kgwuNbo1gUA/viewform?usp=sf_link" target="_blank" class="a-tag">여기를 클릭하여 인터뷰에 응해 주신다면 큰 도움이 될 것 같습니다!!</a></p>', unsafe_allow_html=True)
-        # st.markdown('<p class="survey-2">서비스를 이용해 주셔서 감사합니다! 좋은 하루 보내세요!</p>', unsafe_allow_html=True)
-
+        processed_image = None
+        deepfake_image = None
+
+    if processed_image is not None and deepfake_image is not None:
+        with col3:
+            st.image(processed_image, use_column_width=True)
+            st.markdown('<div class="custom-caption-2">원본 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
+            st.markdown('<div class="custom-caption-3">이해를 돕기 위해 사진에 노란색을 입히는 딥페이크 알고리즘 적용. 원본 이미지는 딥페이크 알고리즘의 영향을 받음.</div>', unsafe_allow_html=True)
+        
+        with col4:
+            st.image(deepfake_image, use_column_width=True)
+            st.markdown('<div class="custom-caption-2">사전 방지 필터 이미지로 딥페이크를 제작한 경우</div>', unsafe_allow_html=True)
+            st.markdown('<div class="custom-caption-3">방지 필터를 입힌 이미지는 딥페이크의 영향을 받지 않고 알아보기 힘든 사진을 출력함으로써 이미지를 보호함.</div>', unsafe_allow_html=True)
+        
+        st.markdown('<p class="survey"> 유사한 서비스를 사용해 보셨거나, 저희 기술적 원리에 관심이 있으신 분들께선 아래의 간단한 인터뷰에 참여해 주시면 진심으로 감사드리겠습니다.</p>', unsafe_allow_html=True)
+        st.markdown('<p class="survey-1"><a href="https://docs.google.com/forms/d/e/1FAIpQLSdzRtuvQyp3CQDhlxEag40v2yDM7u9NYpJ2gv5kgwuNbo1gUA/viewform?usp=sf_link" target="_blank" class="a-tag">인터뷰로 딥페이크 박멸에 참여!!</a></p>', unsafe_allow_html=True)
+        st.markdown('<p class="survey-2">서비스를 이용해 주셔서 감사합니다! 좋은 하루 보내세요!</p>', unsafe_allow_html=True)