Update app.py

app.py

@@ -130,6 +130,23 @@ def main():
         if value:
             url = st.text_input("url을 입력하세요",value=st.query_params['q'])
             title, content, comments = get_comments(url)
+
+            if st.button("스크랩 시작"):
+                if url:
+                    title, content, comments = get_comments(url)
+                    
+                    # 결과 표시
+                    st.subheader("기사 제목")
+                    st.write(title)
+                    
+                    st.subheader("본문 내용")
+                    st.write(content)
+                    
+                    st.subheader("댓글")
+                    for comment in comments:
+                        if evaluation_predict(comment) == 1:
+                            st.write(comment)
+            
     
             # 결과 표시
             st.subheader("기사 제목")
@@ -142,27 +159,42 @@ def main():
             for comment in comments:
                 if evaluation_predict(comment) == 1:
                     st.write(comment)
+                    
         else:
             url = st.text_input("url을 입력하세요")
+            if st.button("스크랩 시작"):
+                if url:
+                    title, content, comments = get_comments(url)
+                    
+                    # 결과 표시
+                    st.subheader("기사 제목")
+                    st.write(title)
+                    
+                    st.subheader("본문 내용")
+                    st.write(content)
+                    
+                    st.subheader("댓글")
+                    for comment in comments:
+                        if evaluation_predict(comment) == 1:
+                            st.write(comment)
     else:
         url = st.text_input("url을 입력하세요")
-
     
-    if st.button("스크랩 시작"):
-        if url:
-            title, content, comments = get_comments(url)
-            
-            # 결과 표시
-            st.subheader("기사 제목")
-            st.write(title)
-            
-            st.subheader("본문 내용")
-            st.write(content)
-            
-            st.subheader("댓글")
-            for comment in comments:
-                if evaluation_predict(comment) == 1:
-                    st.write(comment)
+        if st.button("스크랩 시작"):
+            if url:
+                title, content, comments = get_comments(url)
+                
+                # 결과 표시
+                st.subheader("기사 제목")
+                st.write(title)
+                
+                st.subheader("본문 내용")
+                st.write(content)
+                
+                st.subheader("댓글")
+                for comment in comments:
+                    if evaluation_predict(comment) == 1:
+                        st.write(comment)
     return 0
 
 if __name__ == "__main__":