Update app.py

app.py

@@ -1,66 +1,51 @@
 import gradio as gr
 from huggingface_hub import create_repo
 
-state_store = {}
-
 with gr.Blocks(title="HF Space Creator") as demo:
-    # — Login UI —
     login_btn  = gr.LoginButton(variant="huggingface", size="lg")
     status_md  = gr.Markdown("*Not logged in.*")
 
-    def show_profile(
-        profile: gr.OAuthProfile | None,
-        token:   gr.OAuthToken  | None
-    ) -> str:
-        if profile is None:
-            return "*Not logged in.*"
-        return f"✅ Logged in as **{profile.username}**"
+    def show_profile(profile: gr.OAuthProfile | None) -> str:
+        return "*Not logged in.*" if profile is None \
+            else f"✅ Logged in as **{profile.username}**"
 
-    demo.load(fn=show_profile, inputs=[login_btn], outputs=[status_md])
-    login_btn.click(fn=show_profile, inputs=[login_btn], outputs=[status_md])
+    demo.load(show_profile, inputs=None, outputs=status_md)
+    login_btn.click(show_profile, inputs=None, outputs=status_md)
 
-    # — Repo creation UI —
     repo_name      = gr.Textbox(label="New Space name", placeholder="your-repo-name")
     create_btn     = gr.Button("Create Space", interactive=False)
     session_id     = gr.Textbox(value="", visible=False)
     logs           = gr.Textbox(label="Logs", interactive=False, lines=5)
     preview_iframe = gr.HTML("<p>No Space created yet.</p>")
 
-    def enable_create(
-        profile: gr.OAuthProfile | None,
-        token:   gr.OAuthToken  | None
-    ):
+    def enable_create(profile: gr.OAuthProfile | None):
         return gr.update(interactive=(profile is not None))
 
-    demo.load(fn=enable_create, inputs=[login_btn], outputs=[create_btn])
-    login_btn.click(fn=enable_create, inputs=[login_btn], outputs=[create_btn])
+    demo.load(enable_create, inputs=None, outputs=create_btn)
+    login_btn.click(enable_create, inputs=None, outputs=create_btn)
 
     def create_space(
         name: str,
-        profile: gr.OAuthProfile | None,
-        token:   gr.OAuthToken  | None
+        oauth_token: gr.OAuthToken | None
     ) -> tuple[str,str,str]:
-        if profile is None or token is None:
-            return (
-                "", 
-                "⚠️ Please sign in first.", 
-                "<p>No Space created yet.</p>"
-            )
-        repo_id = f"{profile.username}/{name}"
+        if oauth_token is None:
+            return "", "⚠️ Please sign in first.", "<p>No Space created yet.</p>"
+        username = oauth_token.user_info["username"]
+        repo_id = f"{username}/{name}"
         create_repo(
             repo_id=repo_id,
-            token=token.token,
+            token=oauth_token.token,
             exist_ok=True,
             repo_type="space"
         )
         url    = f"https://huggingface.co/spaces/{repo_id}"
         logmsg = f"✅ Created or found Space: {url}"
-        iframe= f'<iframe src="{url}" width="100%" height="500px"></iframe>'
+        iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
         return repo_id, logmsg, iframe
 
     create_btn.click(
         fn=create_space,
-        inputs=[repo_name, login_btn],
+        inputs=[repo_name],
         outputs=[session_id, logs, preview_iframe]
     )