Upload 12 files

app.py

@@ -1,30 +1,61 @@
-import gradio as gr
+import sys
+import tkinter as tk
+from tkinter import messagebox
+from PIL import Image, ImageTk
+import threading
 import cv2
 from anti_spoofing import AntiSpoofingSystem
 
-# 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
-    
-    # Process the frame using the anti-spoofing system
-    frame = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
-    anti_spoofing_system.run(lambda frame: frame)  # Modify 'run' to accept a single frame if necessary
-    
-    # Here you could return whether spoofing is detected, a message, or the modified frame
-    return frame
-
-iface = gr.Interface(
-    fn=process_frame,
-    inputs=[
-        gr.Image(source="webcam", tool=None, streaming=True), 
-        gr.Textbox(label="Student ID"), 
-        gr.Textbox(label="Student Name")
-    ],
-    outputs=gr.Image()
-)
-
-iface.launch()
+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()