Upload app_wrapper.py

app_wrapper.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+
+import os
+import gradio as gr
+
+from app import demo as main_demo
+
+
+allowed_usernames = os.environ["ALLOWED_USERNAMES"].split(",") 
+
+def render(profile: gr.OAuthProfile | None) -> dict:
+    if profile is None:
+        visible = None
+    else:
+        visible = profile.username in allowed_usernames
+
+    return gr.Column(visible=visible)
+
+
+with gr.Blocks() as demo:
+    gr.LoginButton()
+    with gr.Column(visible=False) as col:
+        main_demo.render()
+    demo.load(fn=render, outputs=col)
+
+if __name__ == "__main__":
+    demo.launch(show_api=False)