Update app.py

app.py

@@ -11,6 +11,8 @@ st.set_page_config(
 st.title(":blue[Object Detection] :black_cat:")
 st.sidebar.header("메뉴")
 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()
 
 if source_radio == "IMAGE":
     st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 이미지 파일을 선택하면 AI 추론이 시작됩니다.]" )
@@ -44,27 +46,26 @@ 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() 
-        with col1:
-            st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  
-        with col2:
-            if st.button('새로운 객체'):
-                st.session_state.random_class = object_detection.call_class()
-    else:
+    
+    if input_img:
+
         result_img, detected_object = object_detection.process_image(input_img)
         detected_objects_list = detected_object.split(": ")[1].split(", ")
-        #detected_objects = detected_object.split(":")[1].strip()
-        #detected_objects_list = detected_object.split(", ")  
         st.write(detected_objects_list)
 
         if st.session_state.random_class  in detected_objects_list:
             st.header(f':{st.session_state.random_class}: 찾습니다!')
-            st.image(result_img)    
             
-        elif st.session_state.random_class not in detected_objects_list:
+        else:
             st.header(f':{st.session_state.random_class}: 찾을 수 없습니다')
-            st.image(result_img)    
+        st.image(result_img)    
 
+    else:
+        st.session_state.random_class = object_detection.call_class() 
+        col1, col2 = st.columns([3, 1])  
+        with col1:
+            st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  
+        with col2:
+            if st.button('새로운 객체'):
+                st.session_state.random_class = object_detection.call_class()