Update app.py

app.py

@@ -1,18 +1,17 @@
 import gradio as gr
-from app_logic import build_space_from_image
+from app_logic import build_space_from_image, get_username_from_token
 
 def main_ui():
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        with gr.Row():
-            gr.Markdown(
-                """
-                # 🖼️ Image-to-Space Builder
-                ## Create a new Hugging Face Space directly from a repo-embedded image.
-                
-                This tool extracts the file structure and content from an encrypted image (created with a tool like KeyLock)
-                and uses it to build and deploy a new Hugging Face Space.
-                """
-            )
+    with gr.Blocks(theme=gr.themes.Soft(), title="Image-to-Space Builder") as demo:
+        gr.Markdown(
+            """
+            # 🖼️ Image-to-Space Builder
+            ## Create a new Hugging Face Space directly from a repo-embedded image.
+            
+            This tool extracts the file structure and content from an encrypted image (created with a tool like KeyLock)
+            and uses it to build and deploy a new Hugging Face Space.
+            """
+        )
 
         with gr.Row():
             with gr.Column(scale=1):
@@ -27,8 +26,11 @@ def main_ui():
                     type="password",
                     placeholder="Enter your write-permission token"
                 )
+                owner_input = gr.Textbox(
+                    label="Owner (Username/Org)", 
+                    placeholder="Autofills from token if left blank" # Updated placeholder
+                )
                 space_name_input = gr.Textbox(label="New Space Name", placeholder="e.g., my-new-app")
-                owner_input = gr.Textbox(label="Owner (Username/Org)", placeholder="Leave blank to use your username")
                 sdk_input = gr.Dropdown(
                     label="Space SDK",
                     choices=["gradio", "streamlit", "docker", "static"],
@@ -43,7 +45,26 @@ def main_ui():
                 gr.Markdown("### Build Status & Result")
                 output_status_md = gr.Markdown(label="Result")
         
-        # --- Event Handler ---
+        # --- Wrapper function for better UX ---
+        def autofill_owner_if_empty(api_token, current_owner):
+            """
+            Checks if the owner field is empty. If so, it tries to fetch the username.
+            This prevents overwriting a manually entered organization or username.
+            """
+            # Only run the API call if the owner field is currently empty
+            if not current_owner.strip():
+                return get_username_from_token(api_token)
+            # If the user has already typed something, leave it alone.
+            return current_owner
+
+        # --- NEW Event Handler for Autofill ---
+        api_token_input.blur(
+            fn=autofill_owner_if_empty,
+            inputs=[api_token_input, owner_input], # Pass both token and the current owner value
+            outputs=[owner_input]
+        )
+
+        # --- Main "Create" Button Event Handler (Unchanged) ---
         create_button.click(
             fn=build_space_from_image,
             inputs=[