Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import json
 import time
 import os
+from datetime import datetime
 from pathlib import Path
 from llama_cpp import Llama
 from faster_whisper import WhisperModel
@@ -49,7 +50,6 @@ def load_model(model_name):
     if model_name == current_model_name and model_name in loaded_models:
         return loaded_models[model_name]
     
-    # Unload previous model to save RAM
     if current_model_name and current_model_name != model_name:
         if current_model_name in loaded_models:
             del loaded_models[current_model_name]
@@ -78,7 +78,7 @@ def load_model(model_name):
         print(f"✅ {model_name} loaded!")
         return llm
     except Exception as e:
-        print(f"❌ Failed to load {model_name}: {e}")
+        print(f"❌ Failed to load: {e}")
         return None
 
 # ===== WHISPER =====
@@ -100,8 +100,8 @@ init_whisper()
 def get_status():
     available = [name for name, file in MODELS.items() if os.path.exists(os.path.join(MODELS_DIR, file))]
     if current_model_name:
-        short_name = current_model_name.split()[1] if len(current_model_name.split()) > 1 else current_model_name
-        return f"🟢 Ready • {len(available)} models • Active: {short_name}"
+        short = current_model_name.split('(')[0].strip().split()[-1]
+        return f"🟢 Ready • {len(available)} models • Active: {short}"
     return f"🟡 {len(available)} models available"
 
 def get_model_info(model_name):
@@ -131,11 +131,10 @@ def generate_response(model_name, prompt, temperature=0.7, max_tokens=2048):
         return "❌ **Model not available.**"
     
     try:
-        # Different formats for different models
         if "deepseek" in model_name.lower():
             formatted = f"### Instruction:\n{prompt}\n\n### Response:\n"
             stop_tokens = ["### Instruction:", "### Response:"]
-        else:  # Qwen format
+        else:
             formatted = f"<|im_start|>system\nYou are an expert coding assistant.<|im_end|>\n<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
             stop_tokens = ["<|im_end|>", "<|im_start|>"]
         
@@ -154,6 +153,7 @@ def generate_response(model_name, prompt, temperature=0.7, max_tokens=2048):
         return response if response else "⚠️ Empty response."
     except Exception as e:
         return f"❌ **Error:** {str(e)[:100]}"
+
 def extract_code(text):
     if not text or "```" not in text:
         return text
@@ -168,6 +168,47 @@ def extract_code(text):
         pass
     return text
 
+# ===== HISTORY FUNCTIONS =====
+
+def export_chat_history(history):
+    if not history:
+        return None, "⚠️ No chat history to export."
+    
+    export = {
+        "exported_at": datetime.now().isoformat(),
+        "tool": "Axon v6 Chat",
+        "messages": []
+    }
+    
+    for user_msg, assistant_msg in history:
+        export["messages"].append({"role": "user", "content": user_msg})
+        export["messages"].append({"role": "assistant", "content": assistant_msg})
+    
+    filename = f"/tmp/axon_chat_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
+    with open(filename, "w") as f:
+        json.dump(export, f, indent=2)
+    
+    return filename, f"✅ Exported {len(history)} messages!"
+
+def export_code(code, language):
+    if not code or not code.strip():
+        return None, "⚠️ No code to export."
+    
+    ext_map = {
+        "Python": "py", "JavaScript": "js", "TypeScript": "ts", "Go": "go",
+        "Rust": "rs", "Java": "java", "C++": "cpp", "C#": "cs", "C": "c",
+        "PHP": "php", "Ruby": "rb", "Swift": "swift", "Kotlin": "kt",
+        "HTML/CSS": "html", "SQL": "sql", "Bash": "sh", "PowerShell": "ps1", "Lua": "lua"
+    }
+    
+    ext = ext_map.get(language, "txt")
+    filename = f"/tmp/axon_code_{datetime.now().strftime('%Y%m%d_%H%M%S')}.{ext}"
+    
+    with open(filename, "w") as f:
+        f.write(code)
+    
+    return filename, f"✅ Exported as .{ext}!"
+
 # ===== STREAMING =====
 
 def chat_stream(message, history, model_name, temperature, max_tokens):
@@ -181,25 +222,25 @@ def chat_stream(message, history, model_name, temperature, max_tokens):
         yield history + [[message, "❌ Model not available."]]
         return
     
-    conv = "<|im_start|>system\nYou are an expert coding assistant. Use markdown code blocks.<|im_end|>\n"
-    for u, a in history:
-        conv += f"<|im_start|>user\n{u}<|im_end|>\n"
-        if a:
-            conv += f"<|im_start|>assistant\n{a}<|im_end|>\n"
-    conv += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
+    if "deepseek" in model_name.lower():
+        conv = "### Instruction:\nYou are an expert coding assistant. Use markdown code blocks.\n\n"
+        for u, a in history:
+            conv += f"User: {u}\nAssistant: {a}\n\n"
+        conv += f"User: {message}\n\n### Response:\n"
+        stop_tokens = ["### Instruction:", "User:"]
+    else:
+        conv = "<|im_start|>system\nYou are an expert coding assistant. Use markdown code blocks.<|im_end|>\n"
+        for u, a in history:
+            conv += f"<|im_start|>user\n{u}<|im_end|>\n"
+            if a:
+                conv += f"<|im_start|>assistant\n{a}<|im_end|>\n"
+        conv += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
+        stop_tokens = ["<|im_end|>", "<|im_start|>"]
     
     try:
         full = ""
-        for chunk in llm(
-            conv,
-            max_tokens=max_tokens,
-            temperature=temperature,
-            top_p=0.9,
-            stop=["<|im_end|>", "<|im_start|>"],
-            stream=True
-        ):
-            token = chunk["choices"][0]["text"]
-            full += token
+        for chunk in llm(conv, max_tokens=max_tokens, temperature=temperature, top_p=0.9, stop=stop_tokens, stream=True):
+            full += chunk["choices"][0]["text"]
             yield history + [[message, full]]
     except Exception as e:
         yield history + [[message, f"❌ {str(e)[:50]}"]]
@@ -215,11 +256,16 @@ def generate_stream(prompt, language, model_name, temperature, max_tokens):
         yield "❌ Model not available."
         return
     
-    formatted = f"<|im_start|>system\nYou are an expert coder.<|im_end|>\n<|im_start|>user\nWrite clean {language} code with comments:\n{prompt}\n\nOutput only code:<|im_end|>\n<|im_start|>assistant\n"
+    if "deepseek" in model_name.lower():
+        formatted = f"### Instruction:\nWrite clean {language} code with comments:\n{prompt}\n\nOutput only code:\n\n### Response:\n"
+        stop_tokens = ["### Instruction:"]
+    else:
+        formatted = f"<|im_start|>system\nYou are an expert coder.<|im_end|>\n<|im_start|>user\nWrite clean {language} code with comments:\n{prompt}\n\nOutput only code:<|im_end|>\n<|im_start|>assistant\n"
+        stop_tokens = ["<|im_end|>"]
     
     try:
         full = ""
-        for chunk in llm(formatted, max_tokens=max_tokens, temperature=temperature, stop=["<|im_end|>"], stream=True):
+        for chunk in llm(formatted, max_tokens=max_tokens, temperature=temperature, stop=stop_tokens, stream=True):
             full += chunk["choices"][0]["text"]
             yield extract_code(full)
     except Exception as e:
@@ -281,8 +327,6 @@ def optimize_code(code, language, focus, model_name, max_tokens):
         return err
     return generate_response(model_name, f"Optimize {language} for {focus.lower()}. Explain:\n{code}", 0.3, max_tokens)
 
-# ===== NEW FEATURES =====
-
 def security_scan(code, model_name, max_tokens):
     valid, err = validate_input(code, "Code")
     if not valid:
@@ -408,36 +452,126 @@ def convert_data_format(data, from_fmt, to_fmt, model_name, max_tokens):
     result = generate_response(model_name, f"Convert {from_fmt} to {to_fmt}:\n{data}\n\nOutput only:", 0.1, max_tokens)
     return result if result.startswith("❌") else extract_code(result)
 
+# ===== THEME =====
+
+light_theme = gr.themes.Soft(
+    primary_hue="indigo",
+    secondary_hue="blue",
+)
+
+dark_theme = gr.themes.Soft(
+    primary_hue="indigo",
+    secondary_hue="blue",
+).set(
+    body_background_fill="#0f172a",
+    body_background_fill_dark="#0f172a",
+    block_background_fill="#1e293b",
+    block_background_fill_dark="#1e293b",
+    border_color_primary="#334155",
+    border_color_primary_dark="#334155",
+)
+
 # ===== UI =====
 
-with gr.Blocks(title="Axon v6") as demo:
+with gr.Blocks(title="Axon v6", theme=dark_theme) as demo:
     
+    # State for theme
+    is_dark = gr.State(True)
+    
+    # Header
     gr.HTML("""
     <div style="background: linear-gradient(135deg, #6366f1, #8b5cf6, #06b6d4); border-radius: 16px; padding: 24px; margin-bottom: 16px;">
-        <h1 style="color: white; margin: 0; font-size: 2rem;">🔥 Axon v6</h1>
-        <p style="color: rgba(255,255,255,0.9); margin: 4px 0 0 0;">llama.cpp Edition • 19 Tools • Community Wheels! 🛞</p>
-        <div style="display: flex; gap: 8px; margin-top: 12px; flex-wrap: wrap;">
-            <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">🤖 3 Models</span>
-            <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">🛠️ 19 Tools</span>
-            <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">⚡ llama.cpp</span>
-            <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">🔒 Local</span>
+        <div style="display: flex; justify-content: space-between; align-items: center; flex-wrap: wrap; gap: 16px;">
+            <div>
+                <h1 style="color: white; margin: 0; font-size: 2rem;">🔥 Axon v6</h1>
+                <p style="color: rgba(255,255,255,0.9); margin: 4px 0 0 0;">AI Coding Assistant • 8 Models • 19 Tools • 100% Local</p>
+            </div>
+            <div style="display: flex; gap: 8px; flex-wrap: wrap;">
+                <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">🤖 8 Models</span>
+                <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">🛠️ 19 Tools</span>
+                <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 0.8rem; color: white;">⚡ llama.cpp</span>
+            </div>
         </div>
     </div>
     """)
     
-    status = gr.Markdown(value=get_status, every=5)
+    # Status row
+    with gr.Row():
+        status = gr.Markdown(value=get_status, every=5)
     
+    # Settings
     with gr.Row():
-        model_dropdown = gr.Dropdown(choices=list(MODELS.keys()), value="Qwen2.5 Coder 3B (Fast)", label="🤖 Model", scale=3)
+        model_dropdown = gr.Dropdown(choices=list(MODELS.keys()), value="🚀 Qwen2.5 Coder 3B (Fast)", label="🤖 Model", scale=3)
         temperature = gr.Slider(0, 1, value=0.7, step=0.1, label="🌡️ Creativity", scale=2)
         max_tokens = gr.Slider(256, 4096, value=2048, step=256, label="📏 Max Tokens", scale=2)
     
-    model_info = gr.Markdown(value="⚖️ Balanced • ~2GB • Recommended")
+    model_info = gr.Markdown(value="🚀 Fast • ~2GB • Great all-rounder")
     model_dropdown.change(get_model_info, model_dropdown, model_info)
 
     with gr.Tabs():
         
-        # Chat
+        # ===== HOME =====
+        with gr.TabItem("🏠 Home"):
+            gr.HTML("""
+            <div style="padding: 20px;">
+                <h2 style="margin-top: 0;">Welcome to Axon v6! 🔥</h2>
+                <p>The ultimate free AI coding assistant - running 100% locally on your browser.</p>
+                
+                <h3>🚀 Quick Start</h3>
+                <ol>
+                    <li><strong>Select a model</strong> from the dropdown above</li>
+                    <li><strong>Choose a tool</strong> from the tabs</li>
+                    <li><strong>Start coding!</strong></li>
+                </ol>
+                
+                <h3>🤖 Models</h3>
+                <table style="width: 100%; border-collapse: collapse;">
+                    <tr style="background: rgba(99, 102, 241, 0.2);">
+                        <th style="padding: 8px; text-align: left;">Model</th>
+                        <th style="padding: 8px; text-align: left;">Size</th>
+                        <th style="padding: 8px; text-align: left;">Best For</th>
+                    </tr>
+                    <tr><td style="padding: 8px;">⭐ Qwen3 30B-A3B</td><td>~10GB</td><td>Best quality (MoE)</td></tr>
+                    <tr><td style="padding: 8px;">🏆 Qwen2.5 14B</td><td>~8GB</td><td>Premium tasks</td></tr>
+                    <tr><td style="padding: 8px;">🧠 DeepSeek V2 Lite</td><td>~9GB</td><td>Complex logic</td></tr>
+                    <tr><td style="padding: 8px;">⚖️ Qwen2.5 7B</td><td>~4.5GB</td><td>Balanced</td></tr>
+                    <tr><td style="padding: 8px;">🚀 Qwen2.5 3B</td><td>~2GB</td><td>Fast & capable</td></tr>
+                    <tr><td style="padding: 8px;">⚡ DeepSeek 6.7B</td><td>~4GB</td><td>Algorithms</td></tr>
+                    <tr><td style="padding: 8px;">💨 Qwen2.5 1.5B</td><td>~1GB</td><td>Quick tasks</td></tr>
+                    <tr><td style="padding: 8px;">🔬 Qwen2.5 0.5B</td><td>~0.3GB</td><td>Instant</td></tr>
+                </table>
+                
+                <h3>🛠️ 19 Tools Available</h3>
+                <p>
+                    <strong>Core:</strong> Chat, Generate, Explain, Debug, Review<br>
+                    <strong>Advanced:</strong> Security, Complexity, Convert, Test, Document, Optimize, Diff, Pseudo, Interview<br>
+                    <strong>Builders:</strong> SQL, Shell, Cron, Regex, API<br>
+                    <strong>Data:</strong> Mock Data, Format Converter
+                </p>
+                
+                <h3>🔗 Links</h3>
+                <p>
+                    <a href="https://huggingface.co/datasets/AIencoder/llama-cpp-wheels" target="_blank">🛞 Pre-built Wheels</a> •
+                    <a href="https://github.com/ggerganov/llama.cpp" target="_blank">📦 llama.cpp</a> •
+                    <a href="https://huggingface.co/Qwen" target="_blank">🤖 Qwen Models</a>
+                </p>
+            </div>
+            """)
+            
+            # Share buttons
+            gr.HTML("""
+            <div style="padding: 20px; border-top: 1px solid #334155;">
+                <h3>📤 Share Axon</h3>
+                <div style="display: flex; gap: 10px; flex-wrap: wrap;">
+                    <a href="https://twitter.com/intent/tweet?text=Check%20out%20Axon%20v6%20-%20Free%20AI%20Coding%20Assistant!%20🔥%208%20Models,%2019%20Tools,%20100%25%20Local&url=https://huggingface.co/spaces/AIencoder/Axon" target="_blank" style="background: #1DA1F2; color: white; padding: 8px 16px; border-radius: 8px; text-decoration: none;">🐦 Twitter</a>
+                    <a href="https://www.reddit.com/submit?url=https://huggingface.co/spaces/AIencoder/Axon&title=Axon%20v6%20-%20Free%20AI%20Coding%20Assistant%20with%208%20Models%20and%2019%20Tools" target="_blank" style="background: #FF4500; color: white; padding: 8px 16px; border-radius: 8px; text-decoration: none;">🤖 Reddit</a>
+                    <a href="https://www.linkedin.com/sharing/share-offsite/?url=https://huggingface.co/spaces/AIencoder/Axon" target="_blank" style="background: #0A66C2; color: white; padding: 8px 16px; border-radius: 8px; text-decoration: none;">💼 LinkedIn</a>
+                    <a href="https://huggingface.co/spaces/AIencoder/Axon" target="_blank" style="background: #FFD21E; color: black; padding: 8px 16px; border-radius: 8px; text-decoration: none;">🤗 HuggingFace</a>
+                </div>
+            </div>
+            """)
+        
+        # ===== CHAT =====
         with gr.TabItem("💬 Chat"):
             chatbot = gr.Chatbot(height=400)
             with gr.Row():
@@ -447,8 +581,12 @@ with gr.Blocks(title="Axon v6") as demo:
                 audio = gr.Audio(sources=["microphone"], type="filepath", label="🎤", scale=2)
                 transcribe = gr.Button("🎤 Transcribe", scale=1)
                 clear = gr.Button("🗑️ Clear", scale=1)
+                export_chat_btn = gr.Button("💾 Export", scale=1)
+            
+            chat_export_file = gr.File(label="Download", visible=False)
+            chat_export_status = gr.Markdown("")
         
-        # Generate
+        # ===== GENERATE =====
         with gr.TabItem("⚡ Generate"):
             with gr.Row():
                 with gr.Column():
@@ -459,8 +597,12 @@ with gr.Blocks(title="Axon v6") as demo:
                     gen_btn = gr.Button("⚡ Generate", variant="primary")
                 with gr.Column():
                     gen_output = gr.Code(label="Code", language="python", lines=14)
+                    with gr.Row():
+                        gen_export_btn = gr.Button("💾 Export Code")
+                        gen_export_status = gr.Markdown("")
+                    gen_export_file = gr.File(label="Download", visible=False)
         
-        # Explain
+        # ===== EXPLAIN =====
         with gr.TabItem("🔍 Explain"):
             with gr.Row():
                 with gr.Column():
@@ -470,7 +612,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     explain_output = gr.Markdown()
         
-        # Debug
+        # ===== DEBUG =====
         with gr.TabItem("🔧 Debug"):
             with gr.Row():
                 with gr.Column():
@@ -480,7 +622,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     fix_output = gr.Markdown()
         
-        # Review
+        # ===== REVIEW =====
         with gr.TabItem("📋 Review"):
             with gr.Row():
                 with gr.Column():
@@ -489,7 +631,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     review_output = gr.Markdown()
         
-        # Security
+        # ===== SECURITY =====
         with gr.TabItem("🔐 Security"):
             with gr.Row():
                 with gr.Column():
@@ -498,7 +640,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     security_output = gr.Markdown()
         
-        # Complexity
+        # ===== COMPLEXITY =====
         with gr.TabItem("📊 Complexity"):
             with gr.Row():
                 with gr.Column():
@@ -507,7 +649,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     complexity_output = gr.Markdown()
         
-        # Convert
+        # ===== CONVERT =====
         with gr.TabItem("🔄 Convert"):
             with gr.Row():
                 with gr.Column():
@@ -519,7 +661,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     convert_output = gr.Code(label="Result", lines=10)
         
-        # Test
+        # ===== TEST =====
         with gr.TabItem("🧪 Test"):
             with gr.Row():
                 with gr.Column():
@@ -531,7 +673,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     test_output = gr.Code(label="Tests", lines=10)
         
-        # Document
+        # ===== DOCUMENT =====
         with gr.TabItem("📝 Document"):
             with gr.Row():
                 with gr.Column():
@@ -543,7 +685,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     doc_output = gr.Code(label="Documented", lines=10)
         
-        # Optimize
+        # ===== OPTIMIZE =====
         with gr.TabItem("🚀 Optimize"):
             with gr.Row():
                 with gr.Column():
@@ -555,7 +697,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     opt_output = gr.Markdown()
         
-        # Diff
+        # ===== DIFF =====
         with gr.TabItem("🔀 Diff"):
             with gr.Row():
                 with gr.Column():
@@ -565,7 +707,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     diff_output = gr.Markdown()
         
-        # Pseudocode
+        # ===== PSEUDOCODE =====
         with gr.TabItem("📐 Pseudo"):
             with gr.Row():
                 with gr.Column():
@@ -575,7 +717,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     pseudo_output = gr.Markdown()
         
-        # Interview
+        # ===== INTERVIEW =====
         with gr.TabItem("🎓 Interview"):
             with gr.Row():
                 with gr.Column():
@@ -587,7 +729,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     interview_output = gr.Markdown()
         
-        # Builders
+        # ===== BUILDERS =====
         with gr.TabItem("🛠️ Builders"):
             gr.Markdown("### 🗄️ SQL")
             with gr.Row():
@@ -632,7 +774,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     api_output = gr.Code(lines=8)
         
-        # Data
+        # ===== DATA =====
         with gr.TabItem("📦 Data"):
             gr.Markdown("### 📦 Mock Data")
             with gr.Row():
@@ -656,13 +798,19 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     format_output = gr.Code(label="Output", lines=6)
 
+    # Footer
     gr.HTML("""
-    <div style="text-align:center;padding:16px;opacity:0.6;">
-        🔥 Axon v6 llama.cpp • Wheels: <a href="https://huggingface.co/datasets/AIencoder/llama-cpp-wheels" target="_blank" style="color:#8b5cf6;">AIencoder/llama-cpp-wheels</a>
+    <div style="text-align:center;padding:16px;border-top:1px solid #334155;margin-top:20px;">
+        <p style="opacity:0.7;">🔥 Axon v6 • Built with ❤️ by <a href="https://huggingface.co/AIencoder" target="_blank" style="color:#8b5cf6;">AIencoder</a></p>
+        <p style="opacity:0.5;font-size:0.8rem;">
+            Wheels: <a href="https://huggingface.co/datasets/AIencoder/llama-cpp-wheels" target="_blank" style="color:#8b5cf6;">AIencoder/llama-cpp-wheels</a> •
+            Powered by <a href="https://github.com/ggerganov/llama.cpp" target="_blank" style="color:#8b5cf6;">llama.cpp</a>
+        </p>
     </div>
     """)
 
     # ===== EVENTS =====
+    
     def respond(message, history, model, temp, tokens):
         history = history or []
         for updated in chat_stream(message, history, model, temp, tokens):
@@ -673,6 +821,18 @@ with gr.Blocks(title="Axon v6") as demo:
     clear.click(lambda: [], None, chatbot)
     transcribe.click(transcribe_audio, audio, msg)
     
+    # Export handlers
+    def handle_chat_export(history):
+        file, status = export_chat_history(history)
+        return gr.update(value=file, visible=file is not None), status
+    
+    def handle_code_export(code, lang):
+        file, status = export_code(code, lang)
+        return gr.update(value=file, visible=file is not None), status
+    
+    export_chat_btn.click(handle_chat_export, chatbot, [chat_export_file, chat_export_status])
+    gen_export_btn.click(handle_code_export, [gen_output, gen_lang], [gen_export_file, gen_export_status])
+    
     gen_btn.click(generate_stream, [gen_prompt, gen_lang, model_dropdown, gen_temp, max_tokens], gen_output)
     explain_btn.click(explain_code, [explain_input, model_dropdown, explain_detail, max_tokens], explain_output)
     fix_btn.click(fix_code, [fix_input, fix_error, model_dropdown, max_tokens], fix_output)
@@ -697,8 +857,8 @@ with gr.Blocks(title="Axon v6") as demo:
     mock_btn.click(generate_mock_data, [mock_schema, mock_count, mock_format, model_dropdown, max_tokens], mock_output)
     format_btn.click(convert_data_format, [format_input, format_from, format_to, model_dropdown, max_tokens], format_output)
 
-# Preload default model
-print("🔥 Preloading default model...")
-load_model("Qwen2.5 Coder 3B (Fast)")
+# Preload
+print("🔥 Preloading model...")
+load_model("🚀 Qwen2.5 Coder 3B (Fast)")
 
 demo.launch(server_name="0.0.0.0", server_port=7860)