Update app.py

app.py

@@ -55,22 +55,20 @@ def recognize_user(image):
     except Exception as e:
         return f"Error during recognition: {str(e)}"
         
-# Gradio interface for registering users
-register_interface = gr.Interface(
-    fn=register_user,
-    inputs=[gr.Image(), gr.Textbox(label="User ID")],
-    outputs="text",
-    live=True
-)
+with gr.Blocks() as demo:
+    with gr.Tab("Register"):
+        with gr.Row():
+            img = gr.Image()
+            user_id = gr.Textbox(label="User ID")
+            register_button = gr.Button("Register")
+        register_output = gr.Textbox()
+        register_button.click(register_user, inputs=[img, user_id], outputs=register_output)
 
-# Gradio interface for recognizing users
-recognize_interface = gr.Interface(
-    fn=recognize_user,
-    inputs=gr.Image(),
-    outputs="text",
-    live=True
-)
+    with gr.Tab("Recognize"):
+        with gr.Row():
+            img = gr.Image()
+            recognize_button = gr.Button("Recognize")
+        recognize_output = gr.Textbox()
+        recognize_button.click(recognize_user, inputs=img, outputs=recognize_output)
 
-if __name__ == "__main__":
-    register_interface.launch(share=True)
-    recognize_interface.launch(share=True)
+demo.launch(share=True)