Update app.py

app.py

@@ -6,6 +6,7 @@ import pandas as pd
 from PIL import Image
 import io
 
+
 def create_database():
     """Creates the attendance database."""
     conn = sqlite3.connect('attendance.db')
@@ -59,7 +60,7 @@ def capture_photo():
 create_database()
 st.title("Attendance System")
 
-menu = st.sidebar.selectbox("Menu", ["Click Photo", "Database"])
+#upload_choice = st.sidebar.selectbox("Menu", ["Click Photo", "Database"])
 
 if menu == "Click Photo":
     st.header("Mark Attendance by Clicking Photo")
@@ -76,3 +77,31 @@ elif menu == "Database":
     records = get_attendance_records()
     df = pd.DataFrame(records, columns=["ID", "Name", "Timestamp"])
     st.dataframe(df)
+
+# Sidebar for input source
+upload_choice = st.sidebar.radio("Choose Input Source", ["Upload Image", "Upload Video", "Camera"])
+
+if upload_choice == "Camera":
+    image = st.camera_input("Take a picture")
+    if image:
+        frame = np.array(Image.open(image))
+        frame, result_text = process_frame(frame)
+        st.image(frame, caption='Processed Image', use_column_width=True)
+        st.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
+
+elif upload_choice == "Upload Image":
+    uploaded_image = st.file_uploader("Upload Image", type=["png", "jpg", "jpeg"])
+    if uploaded_image:
+        image = Image.open(uploaded_image)
+        frame = np.array(image)
+        frame, result_text = process_frame(frame)
+        st.image(frame, caption='Processed Image', use_column_width=True)
+        st.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
+
+elif upload_choice == "Upload Video":
+    uploaded_video = st.file_uploader("Upload Video", type=["mp4", "mov", "avi"])
+    if uploaded_video:
+        with tempfile.NamedTemporaryFile(delete=False) as tfile:
+            tfile.write(uploaded_video.read())
+            video_source = cv2.VideoCapture(tfile.name)
+            video_feed(video_source)