Upload folder using huggingface_hub

config.py

@@ -17,12 +17,11 @@ class AppConfig:
     
     # Model Configuration
     # Mutable: User can change this in the UI
-    MODEL_NAME: str = '../hf/270m'
+    MODEL_NAME: str = 'gg-hf-gm/functiongemma-270m-it'
 
     AVAILABLE_MODELS: List[str] = field(default_factory=lambda: [
-        '../hf/270m',
-        '../hf/gemma-3-270m-it',
-        'google/gemma-3-270m-it'
+        'gg-hf-gm/functiongemma-270m-it',
+        'google/functiongemma-270m-it'
     ])
 
     DEFAULT_DATASET: Final[str] = 'bebechien/SimpleToolCalling'

engine.py

@@ -80,10 +80,11 @@ class FunctionGemmaEngine:
         self.imported_dataset = []
         self.stop_event = threading.Event()
         self.current_tools = DEFAULT_TOOLS
+        self.has_model_tuned = False
 
         authenticate_hf(self.config.HF_TOKEN)
         try:
-            self.refresh_data_and_model()
+            self.refresh_model()
         except Exception as e:
             print(f"Initial load warning: {e}")
 
@@ -110,8 +111,8 @@ class FunctionGemmaEngine:
         self.model, self.tokenizer = load_model_and_tokenizer(self.config.MODEL_NAME)
         self.loaded_model_name = self.config.MODEL_NAME
 
-    def refresh_data_and_model(self) -> str:
-        self.imported_dataset = []
+    def refresh_model(self) -> str:
+        self.has_model_tuned = False
         try:
             self._load_model_weights()
             return f"Model loaded: {self.loaded_model_name}\nData cleared.\nReady (Session {self.session_id})."
@@ -230,6 +231,8 @@ class FunctionGemmaEngine:
         
         train_thread.join()
         
+        self.has_model_tuned = True
+        
         while not log_queue.empty():
             payload = log_queue.get()
             if isinstance(payload, tuple):

ui.py

@@ -1,19 +1,25 @@
 import gradio as gr
 from config import AppConfig
 from engine import FunctionGemmaEngine
+from typing import Optional
 
 def build_interface() -> gr.Blocks:
     
     # --- State Management Wrappers ---
     
-    def init_session():
+    def init_session(profile: Optional[gr.OAuthProfile] = None):
         config = AppConfig()
         new_engine = FunctionGemmaEngine(config)
+
+        username = profile.username if profile else None
+        repo_update, push_update = update_hub_interactive(new_engine, username)
+
         return (
             new_engine,
             new_engine.get_tools_json(),
             new_engine.config.MODEL_NAME,
-            f"Ready. (Session {new_engine.session_id})"
+            f"Ready. (Session {new_engine.session_id})",
+            repo_update, push_update, username
         )
 
     def run_training_wrapper(engine, epochs, lr, test_size, shuffle, model_name):
@@ -22,7 +28,7 @@ def build_interface() -> gr.Blocks:
 
     def handle_reset(engine, model_name):
         engine.config.MODEL_NAME = model_name.strip()
-        return engine.refresh_data_and_model()
+        return engine.refresh_model()
 
     def update_tools_wrapper(engine, json_val):
         return engine.update_tools(json_val)
@@ -51,14 +57,33 @@ def build_interface() -> gr.Blocks:
             oauth_token=oauth_token.token, 
         )
 
+    def update_repo_preview(username, repo_name):
+        """Updates the markdown preview to show 'username/repo_name'."""
+        if not username:
+            return "⚠️ Sign in to see the target repository path."
+    
+        clean_repo = repo_name.strip() if repo_name else "..."
+        return f"Target Repository: **`{username}/{clean_repo}`**"
+        
+    def update_hub_interactive(engine, username: Optional[str] = None):
+        is_logged_in = username is not None
+        has_model_tuned = engine is not None and engine.has_model_tuned
+        
+        return gr.update(interactive=is_logged_in), gr.update(interactive=is_logged_in and has_model_tuned)
+
     # --- UI Layout ---
     with gr.Blocks(title="FunctionGemma Modkit") as demo:
         engine_state = gr.State()
+        username_state = gr.State()
 
         with gr.Column():
             gr.Markdown("# 🤖 FunctionGemma Modkit: Fine-Tuning")
             gr.Markdown("Fine-tune FunctionGemma to understand your custom functions.<br>See [README](https://huggingface.co/spaces/google/functiongemma-modkit/blob/main/README.md) for more details.")
-            gr.LoginButton(value="(Optional) Sign in to Hugging Face, if you want to push fine-tuned model to your repo.")
+            gr.Markdown("(Optional) Sign in to Hugging Face if you plan to push your fine-tuned model to the Hub later (3. Export).")
+            with gr.Row():
+                gr.LoginButton(value="Sign in with Hugging Face")
+                with gr.Column(scale=3):
+                    gr.Markdown("")
         
         with gr.Tabs():
             
@@ -99,7 +124,7 @@ def build_interface() -> gr.Blocks:
                             choices=default_models,
                             allow_custom_value=True, 
                             label="Base Model",
-                            info="Select a preset OR type a custom Hugging Face model ID (e.g. 'google/gemma-3-1b-it')",
+                            info="Select a preset OR type a custom Hugging Face model ID (e.g. 'google/functiongemma-270m-it')",
                             interactive=True
                         )
                         param_epochs = gr.Slider(
@@ -145,29 +170,47 @@ def build_interface() -> gr.Blocks:
                     with gr.Column():
                         gr.Markdown("#### Option A: Download ZIP")
                         gr.Markdown("Download the model weights locally.")
-                        zip_btn = gr.Button("⬇️ Prepare Model ZIP", variant="secondary")
+                        zip_btn = gr.Button("⬇️ Prepare Model ZIP", variant="secondary", interactive=False)
                         download_file = gr.File(label="Download Archive", interactive=False)
                     
                     with gr.Column():
-                        gr.Markdown("#### Option B: Upload to Hugging Face Hub")
-                        gr.Markdown("Publish to your HF profile. **Requires Login**.")
+                        gr.Markdown("#### Option B: Save to Hugging Face Hub")
+                        gr.Markdown("Publish your fine-tuned model to your personal Hugging Face account.")
                         
-                        with gr.Group():
-                            repo_id_input = gr.Textbox(
-                                label="Repository Name", 
-                                placeholder="my-function-gemma-v1",
-                                info="Will be created under your username (e.g. user/repo)"
-                            )
-                            upload_hub_btn = gr.Button("☁️ Upload to Hub", variant="primary")
+                        repo_name_input = gr.Textbox(
+                            label="TargetRepository Name", 
+                            value="my-functiongemma-v1",
+                            placeholder="e.g., my-functiongemma-v1",
+                            interactive=False
+                        )
+                        push_to_hub_btn = gr.Button("Save to Hugging Face Hub", variant="secondary", interactive=False)
+                        repo_id_preview = gr.Markdown("Target Repository: (Waiting for input...)")
                         
                         upload_status = gr.Markdown("")
 
         # --- EVENT WIRING ---
+        
+        action_buttons = [
+            clear_reload_btn,
+            run_training_btn,
+            zip_btn
+        ]
+
+        def set_interactivity(interactive: bool):
+            return [gr.update(interactive=interactive) for _ in action_buttons]
 
         demo.load(
+            fn=lambda: set_interactivity(False), outputs=action_buttons
+        ).then(
             fn=init_session,
             inputs=None,
-            outputs=[engine_state, tools_editor, param_model, output_display]
+            outputs=[engine_state, tools_editor, param_model, output_display, repo_name_input, push_to_hub_btn, username_state]
+        ).then(
+            fn=update_repo_preview,
+            inputs=[username_state, repo_name_input],
+            outputs=[repo_id_preview]
+        ).then(
+            fn=lambda: [gr.update(interactive=True)]*2, outputs=[clear_reload_btn, run_training_btn]
         )
 
         update_tools_btn.click(
@@ -181,14 +224,15 @@ def build_interface() -> gr.Blocks:
             inputs=[engine_state, import_file], 
             outputs=[import_status]
         )
-
+        
         run_training_btn.click(
             fn=lambda: (
                 gr.update(visible=False), 
                 gr.update(interactive=False), 
+                gr.update(interactive=False), 
                 gr.update(visible=True)
             ),
-            outputs=[run_training_btn, clear_reload_btn, stop_training_btn]
+            outputs=[run_training_btn, clear_reload_btn, zip_btn, stop_training_btn]
         ).then(
             fn=run_training_wrapper,
             inputs=[engine_state, param_epochs, param_lr, param_test_size, param_shuffle, param_model],
@@ -197,9 +241,14 @@ def build_interface() -> gr.Blocks:
             fn=lambda: (
                 gr.update(visible=True), 
                 gr.update(interactive=True), 
+                gr.update(interactive=True), 
                 gr.update(visible=False)
             ),
-            outputs=[run_training_btn, clear_reload_btn, stop_training_btn]
+            outputs=[run_training_btn, clear_reload_btn, zip_btn, stop_training_btn]
+        ).then(
+            fn=update_hub_interactive,
+            inputs=[engine_state, username_state],
+            outputs=[repo_name_input, push_to_hub_btn]
         )
         
         stop_training_btn.click(
@@ -209,21 +258,51 @@ def build_interface() -> gr.Blocks:
         )
 
         clear_reload_btn.click(
+            fn=lambda: set_interactivity(False), outputs=action_buttons
+        ).then(
+            fn=lambda: gr.update(interactive=False), outputs=push_to_hub_btn
+        ).then(
             fn=handle_reset, 
             inputs=[engine_state, param_model], 
             outputs=[output_display]
+        ).then(
+            fn=lambda: [gr.update(interactive=True)]*2, outputs=[clear_reload_btn, run_training_btn]
+        ).then(
+            fn=update_hub_interactive,
+            inputs=[engine_state, username_state],
+            outputs=[repo_name_input, push_to_hub_btn]
         )
         
         zip_btn.click(
+            fn=lambda: set_interactivity(False), outputs=action_buttons
+        ).then(
             fn=zip_wrapper,
             inputs=[engine_state],
             outputs=[download_file]
+        ).then(
+            fn=lambda: set_interactivity(True), outputs=action_buttons
         )
         
-        upload_hub_btn.click(
+        repo_name_input.change(
+            fn=update_repo_preview,
+            inputs=[username_state, repo_name_input],
+            outputs=[repo_id_preview]
+        )
+
+        push_to_hub_btn.click(
+            fn=lambda: set_interactivity(False), outputs=action_buttons
+        ).then(
+            fn=lambda: gr.update(interactive=False), outputs=push_to_hub_btn
+        ).then(
             fn=upload_wrapper,
-            inputs=[engine_state, repo_id_input],
+            inputs=[engine_state, repo_name_input],
             outputs=[upload_status]
+        ).then(
+            fn=lambda: set_interactivity(True), outputs=action_buttons
+        ).then(
+            fn=update_hub_interactive,
+            inputs=[engine_state, username_state],
+            outputs=[repo_name_input, push_to_hub_btn]
         )
 
     return demo