Update app.py

app.py

@@ -10,14 +10,21 @@ st.set_page_config(
 
 
 st.sidebar.header("메뉴")
-col1, col2 = st.columns([4, 1]) 
+
+
+source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
+
+col1, col2 = st.columns([2, 3]) 
 with col1:
     st.title(":blue[Object Detection] :black_cat:")
 with col2:
-    if st.button('새로운 객체'):
-        st.session_state.random_class = object_detection.call_class()
+    if source_radio == "GAME":
+        if st.button('새로운 객체'):
+            st.session_state.random_class = object_detection.call_class()
+    else:
+        pass
+
 
-source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
 if "random_class" not in st.session_state:
     st.session_state.random_class = object_detection.call_class()
 
@@ -53,16 +60,8 @@ 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)
-
-        with col1:
-            pass
-            #if st.button('새로운 객체'):
-            #    st.session_state.random_class = object_detection.call_class()
-            
-        with col2:
             if st.session_state.random_class  in detected_object:
                 st.title(f':{st.session_state.random_class}: 찾습니다!')
             else:
@@ -71,12 +70,6 @@ elif source_radio == "GAME":
         st.image(result_img)    
 
     else:
-        with col1:
-            pass
-            #if st.button('새로운 객체'):
-            #   st.session_state.random_class = object_detection.call_class()
-            
-        with col2:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')