Update app.py

app.py

@@ -47,9 +47,10 @@ elif source_radio == "VIDEO":
 
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
+    col1, col2 = st.columns([3, 1])  # Adjust the column widths as needed
     if input_img is None:
         
-        col1, col2 = st.columns([3, 1])  # Adjust the column widths as needed
+        
     
         with col1:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  # Display the header
@@ -59,13 +60,22 @@ elif source_radio == "GAME":
                 st.session_state.random_class = object_detection.call_class()
     else:
         result_img, detected_object = object_detection.process_image(input_img)
-        if st.session_state.random_class  in detected_object:
-            st.header(f':{st.session_state.random_class}: 찾습니다!')
-            
-        else:
-            st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
 
-        st.image(result_img)    
+
+        
+
+        with col1:
+               
+            if st.session_state.random_class  in detected_object:
+                st.header(f':{st.session_state.random_class}: 찾습니다!')
+            
+            else:
+                st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+            st.image(result_img) 
+        
+        with col2:
+            if st.button('새로운 객체'):
+                st.session_state.random_class = object_detection.call_class()