Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import cv2
+from camera_input_live import camera_input_live
 from utils import object_detection_brainai as odb
 object_detection = odb.ObjectDetectionModel()
 
@@ -7,24 +9,22 @@ st.set_page_config(
     page_icon = ":black_cat:",
     layout = "wide")
 
+if "random_class" not in st.session_state:
+    st.session_state.random_class = object_detection.call_class()
+
 st.sidebar.header("메뉴")
-source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME"])
+source_radio = st.sidebar.radio("선택하세요", ["IMAGE", "VIDEO", "GAME", "WEBCAM"])
 
-col1, col2 = st.columns([2, 3]) 
+col1, col2 = st.columns([3, 2]) 
 with col1:
     st.title(":blue[Object Detection] :black_cat:")
 with col2:
     if source_radio == "GAME":
         if st.button('새로운 객체'):
             st.session_state.random_class = object_detection.call_class()
-    else:
-        pass
-
-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 추론이 시작됩니다.]" )
+    st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 이미지 파일을 선택하면 AI 추론이 시작됩니다.]")
     st.sidebar.header("이미지 파일 업로드")
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
 
@@ -46,7 +46,6 @@ elif source_radio == "VIDEO":
     st.write(":green[왼쪽 메뉴 'Browse files' 버튼을 클릭하여 비디오 파일을 선택하면 AI 추론이 시작됩니다.]" )
     st.sidebar.header("비디오 파일 업로드")
     input_video = st.sidebar.file_uploader("비디오 파일을 선택하세요..", type=("mp4"))
-    print(input_video)
     if input_video is not None:
         temp_file = object_detection.play_video(input_video)
         st.video(temp_file)
@@ -55,7 +54,7 @@ elif source_radio == "VIDEO":
 
 elif source_radio == "GAME":
     input_img = st.sidebar.file_uploader("이미지 파일을 선택하세요.", type=("jpg", "png"))
-    if input_img:
+    if input_img is not None:
         result_img, detected_object = object_detection.process_image(input_img)
         if st.session_state.random_class  in detected_object:
             st.title(f':{st.session_state.random_class}: 찾습니다!')
@@ -66,4 +65,12 @@ elif source_radio == "GAME":
 
     else:
             st.title(f':{st.session_state.random_class}: 있는 이미지를 업로드')  
+
+elif source_radio == "WEBCAM":
+    input_img = camera_input_live()
+    if input_img:
+        result = object_detection.process_image(input_img) 
+        img_plot = result[0].plot()
+        
+        st.image(img_plot, channels = "BGR")