Update app.py

app.py

@@ -55,20 +55,25 @@ def recognize_user(image):
     except Exception as e:
         return f"Error during recognition: {str(e)}"
         
-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)
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("Facial Recognition System")
+        with gr.Tab("Register"):
+            with gr.Row():
+                img_register = 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_register, user_id], outputs=register_output)
 
-    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)
+        with gr.Tab("Recognize"):
+            with gr.Row():
+                img_recognize = gr.Image()
+                recognize_button = gr.Button("Recognize")
+            recognize_output = gr.Textbox()
+            recognize_button.click(recognize_user, inputs=[img_recognize], outputs=recognize_output)
 
-demo.launch(share=True)
+    demo.launch(share=True)
+
+if __name__ == "__main__":
+    main()