Update app.py

app.py

@@ -21,6 +21,7 @@ from forgekit.config_generator import (
 )
 from forgekit.notebook_generator import generate_merge_notebook, save_notebook
 from forgekit.ai_advisor import merge_advisor, model_describer, config_explainer
+from forgekit.kaggle_runner import push_and_run_kernel, check_kernel_status, generate_kaggle_notebook
 
 # ===== THEME =====
 theme = gr.themes.Base(
@@ -627,12 +628,12 @@ license: apache-2.0
         # =====================================================
         with gr.Tab("🤖 AI Advisor", id="ai"):
             gr.Markdown("""### AI-Powered Merge Intelligence
-            Get smart recommendations, capability predictions, and plain-English explanations — powered by open-source LLMs via HF Inference API.""")
+            Get smart recommendations, capability predictions, and plain-English explanations — powered by **Llama 3.3 70B** on Groq (free, blazing fast).""")
 
-            ai_token = gr.Textbox(
-                label="HF Token (recommended for faster responses)",
+            groq_key = gr.Textbox(
+                label="Groq API Key (free at console.groq.com)",
                 type="password",
-                placeholder="hf_... (optional but reduces rate limits)",
+                placeholder="gsk_... (free, no credit card needed)",
             )
 
             with gr.Tabs():
@@ -652,7 +653,7 @@ license: apache-2.0
                     ai_advise_out = gr.Markdown()
 
                     ai_advise_btn.click(
-                        merge_advisor, [ai_models, ai_goal, ai_token], ai_advise_out
+                        merge_advisor, [ai_models, ai_goal, groq_key], ai_advise_out
                     )
 
                 # --- Model Describer ---
@@ -669,7 +670,7 @@ license: apache-2.0
                     desc_out = gr.Markdown()
 
                     desc_btn.click(
-                        model_describer, [desc_models, desc_method, desc_weights, ai_token], desc_out
+                        model_describer, [desc_models, desc_method, desc_weights, groq_key], desc_out
                     )
 
                 # --- Config Explainer ---
@@ -699,11 +700,139 @@ dtype: bfloat16""",
                     explain_out = gr.Markdown()
 
                     explain_btn.click(
-                        config_explainer, [explain_yaml, ai_token], explain_out
+                        config_explainer, [explain_yaml, groq_key], explain_out
                     )
 
         # =====================================================
-        # TAB 6: LEADERBOARD
+        # TAB 6: KAGGLE RUNNER
+        # =====================================================
+        with gr.Tab("🚀 Run on Kaggle", id="kaggle"):
+            gr.Markdown("""### Run Your Merge on Kaggle's Free GPU
+            Push your merge notebook directly to Kaggle and run it on a free T4 GPU — no local setup needed.
+
+            **You need:** A [Kaggle account](https://www.kaggle.com) with an API token. Go to *Settings > API > Create New Token*.""")
+
+            with gr.Row():
+                with gr.Column():
+                    kg_username = gr.Textbox(label="Kaggle Username", placeholder="your_kaggle_username")
+                    kg_key = gr.Textbox(label="Kaggle API Key", type="password", placeholder="From kaggle.json")
+                with gr.Column():
+                    kg_hf_note = gr.Markdown("""**Important:** Add your HF token as a Kaggle Secret:
+1. Go to your kernel's **Settings** tab
+2. Under **Secrets**, add `HF_TOKEN` with your HuggingFace token
+3. This lets the kernel download gated models and upload results""")
+
+            gr.Markdown("---")
+            gr.Markdown("#### Configure Merge (or use settings from Merge Builder tab)")
+
+            with gr.Row():
+                kg_models = gr.Textbox(
+                    label="Models (one per line)", lines=4,
+                    placeholder="Qwen/Qwen2.5-Coder-7B-Instruct\nQwen/Qwen2.5-Math-7B-Instruct",
+                )
+                with gr.Column():
+                    kg_method = gr.Dropdown(choices=list(MERGE_METHODS.keys()), value="dare_ties", label="Method")
+                    kg_base = gr.Textbox(label="Base Model", placeholder="Qwen/Qwen2.5-7B-Instruct")
+                    kg_weights = gr.Textbox(label="Weights", placeholder="0.5, 0.5")
+                    kg_densities = gr.Textbox(label="Densities", placeholder="0.7, 0.6")
+
+            with gr.Row():
+                kg_output_name = gr.Textbox(label="Output Model Name", placeholder="My-Merged-7B", value="ForgeKit-Merge")
+                kg_hf_user = gr.Textbox(label="HF Username (for upload)", placeholder="AIencoder")
+
+            kg_run_btn = gr.Button("🚀 Push & Run on Kaggle", variant="primary", size="lg")
+            kg_status = gr.Markdown()
+
+            def run_on_kaggle(
+                username, key, models_text, method, base, weights_text, densities_text,
+                output_name, hf_user,
+            ):
+                # Build config
+                models = [m.strip() for m in models_text.strip().split("\n") if m.strip()]
+                if len(models) < 2:
+                    return "Add at least 2 models."
+
+                weights = []
+                if weights_text.strip():
+                    try:
+                        weights = [float(w.strip()) for w in weights_text.split(",")]
+                    except ValueError:
+                        return "Invalid weights."
+
+                densities = []
+                if densities_text.strip():
+                    try:
+                        densities = [float(d.strip()) for d in densities_text.split(",")]
+                    except ValueError:
+                        return "Invalid densities."
+
+                config = MergeConfig(
+                    method=method,
+                    models=models,
+                    base_model=base.strip(),
+                    weights=weights,
+                    densities=densities,
+                )
+
+                name = output_name.strip() or "ForgeKit-Merge"
+
+                # Generate notebook
+                nb = generate_merge_notebook(
+                    config,
+                    output_model_name=name,
+                    hf_username=hf_user.strip(),
+                    include_quantize=True,
+                    include_deploy=False,
+                    quant_types=["Q5_K_M", "Q4_K_M"],
+                )
+
+                # Adapt for Kaggle
+                kaggle_nb_json = generate_kaggle_notebook(nb)
+
+                # Push to Kaggle
+                result = push_and_run_kernel(
+                    notebook_json=kaggle_nb_json,
+                    kernel_title=f"ForgeKit-{name}",
+                    kaggle_username=username.strip(),
+                    kaggle_key=key.strip(),
+                    enable_gpu=True,
+                    enable_internet=True,
+                )
+
+                if result["success"]:
+                    return result["message"]
+                else:
+                    return result["error"]
+
+            kg_run_btn.click(
+                run_on_kaggle,
+                [kg_username, kg_key, kg_models, kg_method, kg_base, kg_weights, kg_densities,
+                 kg_output_name, kg_hf_user],
+                kg_status,
+            )
+
+            gr.Markdown("---")
+            gr.Markdown("#### Check Kernel Status")
+            with gr.Row():
+                kg_check_slug = gr.Textbox(label="Kernel Slug", placeholder="username/forgekit-my-merged-7b")
+                kg_check_btn = gr.Button("🔍 Check Status", variant="secondary")
+            kg_check_out = gr.Markdown()
+
+            def check_status(slug, username, key):
+                if not slug.strip():
+                    return "Enter a kernel slug (username/kernel-name)"
+                result = check_kernel_status(slug.strip(), username.strip(), key.strip())
+                if result["success"]:
+                    msg = result["display"]
+                    if result.get("failure_message"):
+                        msg += f"\n\nError: {result['failure_message']}"
+                    return msg
+                return result["error"]
+
+            kg_check_btn.click(check_status, [kg_check_slug, kg_username, kg_key], kg_check_out)
+
+        # =====================================================
+        # TAB 7: LEADERBOARD
         # =====================================================
         with gr.Tab("🏆 Leaderboard", id="leaderboard"):
             gr.Markdown("""### Community Merge Leaderboard