-
Notifications
You must be signed in to change notification settings - Fork 0
/
face_main.py
58 lines (49 loc) · 1.71 KB
/
face_main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import os
import sys
import streamlit as st
from pathlib import Path
# Add the current directory to the system path
sys.path.append(os.getcwd())
#Now try importing the modules
import image
import detection
import video
from streamlit_option_menu import option_menu
import os
os.environ['KMP_DUPLICATE_LIB_OK']='TRUE'
st.set_page_config(
page_title="YOLO",
layout="wide"
)
class MultiApp:
def __init__(self):
self.apps = []
def add_app(self, title, function):
self.apps.append({
"title": title,
"function": function
})
def run(self):
# st.sidebar.image('logo_.jpg', width=220)
with st.sidebar:
app = option_menu(
menu_title="page_names",
options=['IMAGE', 'VIDEO', 'URL LINKS'],
icons=['cloud-upload-fill', 'compass-fill', 'binoculars-fill'],
menu_icon='chat-text-fill',
default_index=1,
styles={
"container": {"padding": "5!important", "background-color": '#ff6666'},
"icon": {"color": "white", "font-size": "23px"},
"nav-link": {"color": "white", "font-size": "20px", "text-align": "left", "margin": "0px", "--hover-color": "blue"},
"nav-link-selected": {"background-color": "black"},
}
)
for app_info in self.apps:
if app == app_info["title"]:
app_info["function"]()
break
app = MultiApp()
app.add_app("IMAGE Detection", image.app())
app.add_app("VIDEO Detection", detection.app())
app.add_app("Capture Detection", video.app())