Update app.py

app.py

@@ -2,37 +2,43 @@ import uuid
 import gradio as gr
 from huggingface_hub import create_repo
 
-# Store created spaces' info
 state_store = {}
 
 with gr.Blocks(title="HF Space Creator") as demo:
-    # 1) OAuth button + login status
+    # --- Login UI ---
     login_btn = gr.LoginButton(variant="huggingface", size="lg")
     status_md = gr.Markdown("*Not logged in.*")
 
     def show_profile(profile: gr.OAuthProfile | None) -> str:
-        return "*Not logged in.*" if profile is None else f"Logged in as **{profile.username}**"
-
-    # Update status on page load and after click
-    demo.load(fn=show_profile, inputs=None, outputs=[status_md])
-    login_btn.click(fn=show_profile, inputs=None, outputs=[status_md])
-
-    # 2) 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)
+        return "*Not logged in.*" if profile is None \
+            else f"✅ Logged in as **{profile.username}**"
+
+    # Update status on load and on login/logout
+    demo.load(fn=show_profile, inputs=[login_btn], outputs=[status_md])
+    login_btn.click(fn=show_profile,  inputs=[login_btn], outputs=[status_md])
+
+    # Enable “Create Space” button only once logged in
+    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>")
 
-    # Enable create button only after login
     def enable_create(profile: gr.OAuthProfile | None):
         return gr.update(interactive=(profile is not None))
-    login_btn.click(fn=enable_create, inputs=None, outputs=[create_btn])
 
-    def create_space(name: str, oauth_token: gr.OAuthToken | None):
-        if oauth_token is None:
-            return "", "Please sign in first.", "<p>No Space created yet.</p>"
-        username = oauth_token.user_info["username"]
+    demo.load(fn=enable_create, inputs=[login_btn], outputs=[create_btn])
+    login_btn.click(fn=enable_create, inputs=[login_btn], outputs=[create_btn])
+
+    # Create Space callback
+    def create_space(
+        name: str,
+        profile: gr.OAuthProfile | None,
+        oauth_token: gr.OAuthToken | None
+    ) -> tuple[str,str,str]:
+        if oauth_token is None or profile is None:
+            return "", "⚠️ Please sign in first.", "<p>No Space created yet.</p>"
+        username = profile.username
         repo_id = f"{username}/{name}"
         create_repo(
             repo_id=repo_id,
@@ -41,7 +47,7 @@ with gr.Blocks(title="HF Space Creator") as demo:
             repo_type="space"
         )
         url = f"https://huggingface.co/spaces/{repo_id}"
-        logs = f"Created or found Space: {url}"
+        logs = f"✅ Created or found Space: {url}"
         iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
         return repo_id, logs, iframe