Update app.py

app.py

@@ -1,61 +1,38 @@
-import sys
-import tkinter as tk
-from tkinter import messagebox
-from PIL import Image, ImageTk
-import threading
+import gradio as gr
 import cv2
 from anti_spoofing import AntiSpoofingSystem
 
-class AntiSpoofingGUI:
-    def __init__(self, anti_spoofing_system):
-        self.anti_spoofing_system = anti_spoofing_system
-        self.window = tk.Tk()
-        self.window.title("Anti-Spoofing System")
-
-        self.student_id_label = tk.Label(self.window, text="Student ID:")
-        self.student_id_label.pack()
-        self.student_id_entry = tk.Entry(self.window)
-        self.student_id_entry.pack()
-
-        self.student_name_label = tk.Label(self.window, text="Student Name:")
-        self.student_name_label.pack()
-        self.student_name_entry = tk.Entry(self.window)
-        self.student_name_entry.pack()
-
-        self.start_button = tk.Button(self.window, text="Start", command=self.start_anti_spoofing)
-        self.start_button.pack()
-
-        self.image_label = tk.Label(self.window)
-        self.image_label.pack()
-
-        # Create a PhotoImage object to use for the video feed
-        self.photo = ImageTk.PhotoImage("RGB", (640, 480))
-
-    def start_anti_spoofing(self):
-        self.student_id = self.student_id_entry.get()
-        self.student_name = self.student_name_entry.get()
-
-        if not self.student_id or not self.student_name:
-            messagebox.showwarning("Warning", "Please enter both Student ID and Name")
-            return
-
-        threading.Thread(target=self.run_anti_spoofing, daemon=True).start()
-
-    def run_anti_spoofing(self):
-        self.anti_spoofing_system.student_id = self.student_id
-        self.anti_spoofing_system.student_name = self.student_name
-        self.anti_spoofing_system.run(self.update_frame)
-
-    def update_frame(self, frame):
-        cv2image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)
-        self.photo.paste(Image.fromarray(cv2image))
-        self.image_label.config(image=self.photo)
-        self.image_label.update_idletasks()
-
-    def run(self):
-        self.window.mainloop()
-
-if __name__ == "__main__":
-    anti_spoofing_system = AntiSpoofingSystem()
-    gui = AntiSpoofingGUI(anti_spoofing_system)
-    gui.run()
+# Initialize the anti-spoofing system
+anti_spoofing_system = AntiSpoofingSystem()
+
+def process_frame(image, student_id, student_name):
+    # Set student details in the anti-spoofing system
+    anti_spoofing_system.student_id = student_id
+    anti_spoofing_system.student_name = student_name
+
+    # Convert the Gradio image format to OpenCV format
+    frame = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+    
+    # Process the frame using the anti-spoofing system
+    processed_frame = frame.copy()  # If needed, adapt the anti-spoofing system to take individual frames
+    anti_spoofing_system.detect_smartphone(processed_frame)
+    
+    # Display a message if a smartphone is detected or blinks are counted
+    message = "Mobile phone detected, can't record attendance" if anti_spoofing_system.smartphone_detected else "Processing attendance..."
+    
+    return cv2.cvtColor(processed_frame, cv2.COLOR_BGR2RGB), message
+
+iface = gr.Interface(
+    fn=process_frame,
+    inputs=[
+        gr.Image(source="webcam", tool=None, streaming=True, label="Webcam Feed"),
+        gr.Textbox(label="Student ID"), 
+        gr.Textbox(label="Student Name")
+    ],
+    outputs=[
+        gr.Image(label="Processed Frame"),
+        gr.Textbox(label="Status Message")
+    ]
+)
+
+iface.launch()