Update app.py

app.py

@@ -90,7 +90,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:
-        return f"🟢 Ready • {len(available)} models • Active: {current_model_name.split()[1] if len(current_model_name.split()) > 1 else 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}"
     return f"🟡 {len(available)} models available"
 
 def get_model_info(model_name):
@@ -120,7 +121,6 @@ def generate_response(model_name, prompt, temperature=0.7, max_tokens=2048):
         return "❌ **Model not available.** Check if downloaded."
     
     try:
-        # Qwen2.5 chat format
         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"
         
         output = llm(
@@ -166,7 +166,6 @@ def chat_stream(message, history, model_name, temperature, max_tokens):
         yield history + [[message, "❌ Model not available."]]
         return
     
-    # Build conversation
     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"
@@ -401,7 +400,7 @@ with gr.Blocks(title="Axon v6") as demo:
     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 • Your Wheels! 🛞</p>
+        <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>
@@ -423,6 +422,7 @@ with gr.Blocks(title="Axon v6") as demo:
 
     with gr.Tabs():
         
+        # Chat
         with gr.TabItem("💬 Chat"):
             chatbot = gr.Chatbot(height=400)
             with gr.Row():
@@ -433,6 +433,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 transcribe = gr.Button("🎤 Transcribe", scale=1)
                 clear = gr.Button("🗑️ Clear", scale=1)
         
+        # Generate
         with gr.TabItem("⚡ Generate"):
             with gr.Row():
                 with gr.Column():
@@ -444,6 +445,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     gen_output = gr.Code(label="Code", language="python", lines=14)
         
+        # Explain
         with gr.TabItem("🔍 Explain"):
             with gr.Row():
                 with gr.Column():
@@ -453,6 +455,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     explain_output = gr.Markdown()
         
+        # Debug
         with gr.TabItem("🔧 Debug"):
             with gr.Row():
                 with gr.Column():
@@ -462,6 +465,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     fix_output = gr.Markdown()
         
+        # Review
         with gr.TabItem("📋 Review"):
             with gr.Row():
                 with gr.Column():
@@ -470,6 +474,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     review_output = gr.Markdown()
         
+        # Security
         with gr.TabItem("🔐 Security"):
             with gr.Row():
                 with gr.Column():
@@ -478,6 +483,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     security_output = gr.Markdown()
         
+        # Complexity
         with gr.TabItem("📊 Complexity"):
             with gr.Row():
                 with gr.Column():
@@ -486,6 +492,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     complexity_output = gr.Markdown()
         
+        # Convert
         with gr.TabItem("🔄 Convert"):
             with gr.Row():
                 with gr.Column():
@@ -497,6 +504,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     convert_output = gr.Code(label="Result", lines=10)
         
+        # Test
         with gr.TabItem("🧪 Test"):
             with gr.Row():
                 with gr.Column():
@@ -508,6 +516,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     test_output = gr.Code(label="Tests", lines=10)
         
+        # Document
         with gr.TabItem("📝 Document"):
             with gr.Row():
                 with gr.Column():
@@ -519,6 +528,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     doc_output = gr.Code(label="Documented", lines=10)
         
+        # Optimize
         with gr.TabItem("🚀 Optimize"):
             with gr.Row():
                 with gr.Column():
@@ -530,6 +540,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     opt_output = gr.Markdown()
         
+        # Diff
         with gr.TabItem("🔀 Diff"):
             with gr.Row():
                 with gr.Column():
@@ -539,6 +550,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     diff_output = gr.Markdown()
         
+        # Pseudocode
         with gr.TabItem("📐 Pseudo"):
             with gr.Row():
                 with gr.Column():
@@ -548,6 +560,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     pseudo_output = gr.Markdown()
         
+        # Interview
         with gr.TabItem("🎓 Interview"):
             with gr.Row():
                 with gr.Column():
@@ -559,6 +572,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     interview_output = gr.Markdown()
         
+        # Builders
         with gr.TabItem("🛠️ Builders"):
             gr.Markdown("### 🗄️ SQL")
             with gr.Row():
@@ -567,7 +581,7 @@ with gr.Blocks(title="Axon v6") as demo:
                     sql_type = gr.Dropdown(["PostgreSQL", "MySQL", "SQLite"], value="PostgreSQL")
                     sql_btn = gr.Button("🗄️ Build", variant="primary")
                 with gr.Column():
-                    sql_output = gr.Code(language="sql", lines=6)
+                    sql_output = gr.Code(lines=6)
             
             gr.Markdown("---\n### 🐚 Shell")
             with gr.Row():
@@ -576,7 +590,7 @@ with gr.Blocks(title="Axon v6") as demo:
                     shell_type = gr.Dropdown(["Bash", "PowerShell", "Zsh"], value="Bash")
                     shell_btn = gr.Button("🐚 Build", variant="primary")
                 with gr.Column():
-                    shell_output = gr.Code(language="bash", lines=6)
+                    shell_output = gr.Code(lines=6)
             
             gr.Markdown("---\n### ⏰ Cron")
             with gr.Row():
@@ -603,6 +617,7 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     api_output = gr.Code(lines=8)
         
+        # Data
         with gr.TabItem("📦 Data"):
             gr.Markdown("### 📦 Mock Data")
             with gr.Row():
@@ -626,7 +641,11 @@ with gr.Blocks(title="Axon v6") as demo:
                 with gr.Column():
                     format_output = gr.Code(label="Output", lines=6)
 
-    gr.HTML('<div style="text-align:center;padding:16px;opacity:0.5;">🔥 Axon v6 llama.cpp • Your Wheels Power This! 🛞</div>')
+    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>
+    """)
 
     # ===== EVENTS =====
     def respond(message, history, model, temp, tokens):