up

app.py

@@ -21,42 +21,58 @@ def push_to_hub(repo_id, filename, oauth_token: gr.OAuthToken):
         raise NotImplementedError("The filename must end with a `.pt2` extension.")
     
     # this will throw if token is invalid
-    _ = whoami(oauth_token.token)
-
-    # --- Ahead-of-time compilation ---
-    compiled_transformer = compile_transformer(pipe, prompt="prompt")
-
-    token = oauth_token.token 
-    out = _push_compiled_graph_to_hub(
-        compiled_transformer.archive_file,
-        repo_id=repo_id,
-        token=token,
-        path_in_repo=filename
-    )
-    if not isinstance(out, str) and hasattr(out, "commit_url"):
-        commit_url = out.commit_url
-        return f"[{commit_url}]({commit_url})"
-    else:
-        return out
-     
+    try:
+        _ = whoami(oauth_token.token)
 
-css = """
-#app {max-width: 840px; margin: 0 auto;}
-"""
+        # --- Ahead-of-time compilation ---
+        compiled_transformer = compile_transformer(pipe, prompt="prompt")
 
+        token = oauth_token.token 
+        out = _push_compiled_graph_to_hub(
+            compiled_transformer.archive_file,
+            repo_id=repo_id,
+            token=token,
+            path_in_repo=filename
+        )
+        if not isinstance(out, str) and hasattr(out, "commit_url"):
+            commit_url = out.commit_url
+            return f"[{commit_url}]({commit_url})"
+        else:
+            return out
+    except Exception as e:
+        raise gr.Error(f"""Oops, you forgot to login. Please use the loggin button on the top left to migrate your repo {e}""")
+     
+css="""
+#col-container {
+    margin: 0 auto;
+    max-width: 520px;
+}
+"""
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("## Compile a model graph ahead of time & push to the Hub")
-    gr.Markdown("Enter a **repo_id** and **filename**. This repo automatically compiles the [QwenImage](https://hf.co/Qwen/Qwen-Image) model on start.")
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("## Compile QwenImage graph ahead of time & push to the Hub")
+        gr.Markdown("Enter a **repo_id** and **filename**. This repo automatically compiles the [QwenImage](https://hf.co/Qwen/Qwen-Image) model on start.")
 
-    with gr.Row():
         repo_id = gr.Textbox(label="repo_id", placeholder="e.g. sayakpaul/qwen-aot")
         filename = gr.Textbox(label="filename", placeholder="e.g. compiled.pt2")
 
-    run = gr.Button("Push graph to Hub", variant="primary")
+        run = gr.Button("Push graph to Hub", variant="primary")
 
-    markdown_out = gr.Markdown()
+        markdown_out = gr.Markdown()
 
     run.click(push_to_hub, inputs=[repo_id, filename], outputs=[markdown_out])
 
-if __name__ == "__main__":
-    demo.launch(show_error=True)
+def swap_visibilty(profile: gr.OAuthProfile | None):
+    return gr.update(elem_classes=["main_ui_logged_in"]) if profile else gr.update(elem_classes=["main_ui_logged_out"])
+        
+css_login = '''
+.main_ui_logged_out{opacity: 0.3; pointer-events: none; margin: 0 auto; max-width: 520px}
+'''
+with gr.Blocks(css=css_login) as demo_login:
+    gr.LoginButton()
+    with gr.Column(elem_classes="main_ui_logged_out") as main_ui:
+        demo.render()
+    demo_login.load(fn=swap_visibilty, outputs=main_ui)
+    
+demo_login.queue()
+demo_login.launch()

requirements.txt

@@ -3,4 +3,5 @@ transformers
 accelerate
 safetensors
 sentencepiece
-kernels
+kernels
+gradio[oauth]