Update app.py

app.py

@@ -44,9 +44,8 @@ elif source_radio == "VIDEO":
 
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
+    col1, col2 = st.columns([3, 1])  
     if input_img is None:
-        st.session_state.random_class = object_detection.call_class() 
-        col1, col2 = st.columns([3, 1])  
         st.session_state.random_class = object_detection.call_class() 
         with col1:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  
@@ -56,12 +55,19 @@ elif source_radio == "GAME":
     else:
         result_img, detected_object = object_detection.process_image(input_img)
         detected_objects_list = detected_object.split(", ")  
-        if st.session_state.random_class  in detected_objects_list:
-            st.header(f':{st.session_state.random_class}: 찾습니다!')
-            
-        elif st.session_state.random_class not in detected_object:
-            st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+        
+
+
 
+        with col1:
+            if st.session_state.random_class  in detected_objects_list:
+                st.header(f':{st.session_state.random_class}: 찾습니다!')
+                
+            elif st.session_state.random_class not in detected_object:
+                st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+        with col2:
+            if st.button('새로운 객체'):
+                st.session_state.random_class = object_detection.call_class()
         st.image(result_img)