Update app.py

app.py

@@ -30,25 +30,25 @@ image_paths = glob(os.path.join(data_path, '*.jpg'))
 app = FaceAnalysis(name="buffalo_l")  # ArcFace model
 app.prepare(ctx_id=0 if torch.cuda.is_available() else -1, det_size=(IMAGE_SHAPE, IMAGE_SHAPE))
 
-# Check if roll exists
-def roll_exists(rno):
-    with sqlite3.connect(STUDENT_DB) as conn:
-        cur = conn.execute("SELECT * FROM students WHERE rno=?", (rno,))
-        return cur.fetchone()
-
-# Register new student
-def register_student(rno, sname, sclass, image):
-    image_path = os.path.join(data_path, f"{rno}.jpg")
-    image.save(image_path)
-    with sqlite3.connect(STUDENT_DB) as conn:
-        conn.execute("INSERT INTO students (rno, sname, sclass, image_path) VALUES (?, ?, ?, ?)",
-                     (rno, sname, sclass, image_path))
-
-
-# Load all registered images
-def load_registered_images():
-    with sqlite3.connect(STUDENT_DB) as conn:
-        return [row[3] for row in conn.execute("SELECT * FROM students")]
+# # Check if roll exists
+# def roll_exists(rno):
+#     with sqlite3.connect(STUDENT_DB) as conn:
+#         cur = conn.execute("SELECT * FROM students WHERE rno=?", (rno,))
+#         return cur.fetchone()
+
+# # Register new student
+# def register_student(rno, sname, sclass, image):
+#     image_path = os.path.join(data_path, f"{rno}.jpg")
+#     image.save(image_path)
+#     with sqlite3.connect(STUDENT_DB) as conn:
+#         conn.execute("INSERT INTO students (rno, sname, sclass, image_path) VALUES (?, ?, ?, ?)",
+#                      (rno, sname, sclass, image_path))
+
+
+# # Load all registered images
+# def load_registered_images():
+#     with sqlite3.connect(STUDENT_DB) as conn:
+        # return [row[3] for row in conn.execute("SELECT * FROM students")]
         
 # Define function to match face embeddings
 def prod_function(app, prod_path, webcam_img_pil):
@@ -170,28 +170,28 @@ with register_tab:
         if st.button("Register"):
             if not (rno and sname and sclass):
                 st.warning("Fill all details.")
-            # Save locally in employees folder
-                local_path = os.path.join("employees", f"{sname}.jpg")
-                image_pil.save(local_path)
-
-                # Convert image to base64 for sending to Glitch
-                buffered = BytesIO()
-                image_pil.save(buffered, format="JPEG")
-                img_str = base64.b64encode(buffered.getvalue()).decode()
-
-                # Prepare data for POST request
-                glitch_url = "https://your-glitch-app.glitch.me/register"  # Replace this
-                data = {
-                    "rno": rno,
-                    "sname": sname,
-                    "sclass": sclass,
-                    "image_base64": img_str,
-                }
-
-                response = requests.post(glitch_url, json=data)
-                result = response.json()
-
-                if result["status"] == "success":
-                    st.success(result["message"])
-                else:
-                    st.error(result["message"])
+            # # Save locally in employees folder
+            #     local_path = os.path.join("employees", f"{sname}.jpg")
+            #     image_pil.save(local_path)
+
+            #     # Convert image to base64 for sending to Glitch
+            #     buffered = BytesIO()
+            #     image_pil.save(buffered, format="JPEG")
+            #     img_str = base64.b64encode(buffered.getvalue()).decode()
+
+            #     # Prepare data for POST request
+            #     glitch_url = "https://your-glitch-app.glitch.me/register"  # Replace this
+            #     data = {
+            #         "rno": rno,
+            #         "sname": sname,
+            #         "sclass": sclass,
+            #         "image_base64": img_str,
+            #     }
+
+            #     response = requests.post(glitch_url, json=data)
+            #     result = response.json()
+
+            #     if result["status"] == "success":
+            #         st.success(result["message"])
+            #     else:
+            #         st.error(result["message"])