Update app.py

app.py

@@ -20,6 +20,7 @@ from forgekit.config_generator import (
     MERGE_METHODS, PRESETS,
 )
 from forgekit.notebook_generator import generate_merge_notebook, save_notebook
+from forgekit.ai_advisor import merge_advisor, model_describer, config_explainer
 
 # ===== THEME =====
 theme = gr.themes.Base(
@@ -622,7 +623,87 @@ license: apache-2.0
             d_btn.click(gen_deploy, [d_model, d_type], [d_output, d_readme])
 
         # =====================================================
-        # TAB 5: LEADERBOARD
+        # TAB 5: AI ADVISOR
+        # =====================================================
+        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.""")
+
+            ai_token = gr.Textbox(
+                label="HF Token (recommended for faster responses)",
+                type="password",
+                placeholder="hf_... (optional but reduces rate limits)",
+            )
+
+            with gr.Tabs():
+                # --- Merge Advisor ---
+                with gr.Tab("💡 Merge Advisor"):
+                    gr.Markdown("**Tell the AI what models you want to merge and it'll recommend the best strategy.**")
+                    ai_models = gr.Textbox(
+                        label="Models (one per line)",
+                        placeholder="Qwen/Qwen2.5-Coder-7B-Instruct\nQwen/Qwen2.5-Math-7B-Instruct",
+                        lines=4,
+                    )
+                    ai_goal = gr.Textbox(
+                        label="What do you want the merged model to do?",
+                        placeholder="I want a model that's great at both coding and math reasoning",
+                    )
+                    ai_advise_btn = gr.Button("💡 Get Recommendation", variant="primary")
+                    ai_advise_out = gr.Markdown()
+
+                    ai_advise_btn.click(
+                        merge_advisor, [ai_models, ai_goal, ai_token], ai_advise_out
+                    )
+
+                # --- Model Describer ---
+                with gr.Tab("🔮 Capability Predictor"):
+                    gr.Markdown("**Predict what your merged model will be good (and bad) at.**")
+                    desc_models = gr.Textbox(
+                        label="Models (one per line)",
+                        placeholder="Qwen/Qwen2.5-Coder-7B-Instruct\nQwen/Qwen2.5-Math-7B-Instruct",
+                        lines=4,
+                    )
+                    desc_method = gr.Textbox(label="Merge Method", placeholder="dare_ties")
+                    desc_weights = gr.Textbox(label="Weights", placeholder="0.5, 0.5")
+                    desc_btn = gr.Button("🔮 Predict Capabilities", variant="primary")
+                    desc_out = gr.Markdown()
+
+                    desc_btn.click(
+                        model_describer, [desc_models, desc_method, desc_weights, ai_token], desc_out
+                    )
+
+                # --- Config Explainer ---
+                with gr.Tab("📖 Config Explainer"):
+                    gr.Markdown("**Paste any mergekit YAML config and get a plain-English explanation.**")
+                    explain_yaml = gr.Code(
+                        label="Paste YAML Config",
+                        language="yaml",
+                        lines=12,
+                        value="""merge_method: dare_ties
+base_model: Qwen/Qwen2.5-7B-Instruct
+models:
+  - model: Qwen/Qwen2.5-Coder-7B-Instruct
+    parameters:
+      weight: 0.5
+      density: 0.7
+  - model: Qwen/Qwen2.5-Math-7B-Instruct
+    parameters:
+      weight: 0.5
+      density: 0.6
+parameters:
+  int8_mask: true
+  normalize: true
+dtype: bfloat16""",
+                    )
+                    explain_btn = gr.Button("📖 Explain This Config", variant="primary")
+                    explain_out = gr.Markdown()
+
+                    explain_btn.click(
+                        config_explainer, [explain_yaml, ai_token], explain_out
+                    )
+
+        # =====================================================
+        # TAB 6: LEADERBOARD
         # =====================================================
         with gr.Tab("🏆 Leaderboard", id="leaderboard"):
             gr.Markdown("""### Community Merge Leaderboard
@@ -667,4 +748,4 @@ license: apache-2.0
 
 
 if __name__ == "__main__":
-    demo.launch(theme=theme, css=CSS)
+    demo.launch(server_name="0.0.0.0", server_port=7860, theme=theme, css=CSS)