Update app.py

app.py

@@ -1,61 +1,30 @@
-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
+    
+    # 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()