Update app.py

app.py

@@ -46,19 +46,23 @@ elif source_radio == "VIDEO":
 
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
-    
+    col1, col2 = st.columns([1, 4]) 
     if input_img:
-
         result_img, detected_object = object_detection.process_image(input_img)
-        detected_objects_list = detected_object.split(": ")[1].split(", ")
-        if st.session_state.random_class  in detected_objects_list:
-            st.header(f':{st.session_state.random_class}: 찾습니다!')
-        else:
-            st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
+
+        with col1:
+            if st.button('새로운 객체'):
+                st.session_state.random_class = object_detection.call_class()
+            
+        with col2:
+            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)    
 
     else:
-        col1, col2 = st.columns([1, 4])  
         with col1:
             if st.button('새로운 객체'):
                 st.session_state.random_class = object_detection.call_class()