Update app.py

app.py

@@ -1,10 +1,8 @@
-import streamlit as st
 import os
-import cv2
-import numpy as np
+import shutil
+import streamlit as st
 import tempfile
 import zipfile
-import shutil
 import embeddings
 
 def extract_database(zip_file):
@@ -14,25 +12,25 @@ def extract_database(zip_file):
         zip_ref.extractall(temp_dir)
     return temp_dir
 
-
-#main function
 def main():
-    st.title("FACE FUSE")
-    tab_selection = st.sidebar.selectbox("Select Funtionality",["Embeddings","Immage Recognition","Video Recognition","Webcam Recognition"])
+    st.title("FACE ANALYZER")
+    tab_selection = st.sidebar.selectbox("Select Functionality", ["Embeddings", "Image Recognition", "Video Recognition", "Webcam Recognition"])
     if tab_selection == "Embeddings":
         st.header("Embeddings")
-        zip_file = st.file_uploader("Upload DataBase(ZIP)",type="zip")
+        zip_file = st.file_uploader("Upload DataBase(ZIP)", type="zip")
 
         if zip_file:
             db_dir = extract_database(zip_file)
             if st.button("Get Embeddings"):
-                temp_dir=embeddings.get_embeddings(db_dir)
+                temp_dir = embeddings.get_embeddings(db_dir)
                 embeddings_path = os.path.join(temp_dir, "embeddings.npy")
                 names_path = os.path.join(temp_dir, "names.npy")
+
                 st.write("Download numpy files:")
                 st.download_button(label="Download embeddings.npy", data=open(embeddings_path, "rb"), file_name="embeddings.npy")
                 st.download_button(label="Download names.npy", data=open(names_path, "rb"), file_name="names.npy")
+
             shutil.rmtree(db_dir, ignore_errors=True)
 
-if __name__=="__main__":
+if __name__ == "__main__":
     main()