Update app.py

app.py

@@ -472,13 +472,7 @@ def create_ui():
     cached = get_cached_models()
     total_models = len(MODELS)
     
-    css = """
-    .main-header { text-align: center; margin-bottom: 20px; }
-    .tool-description { color: #666; font-size: 0.9em; margin-bottom: 10px; }
-    .status-bar { padding: 10px; border-radius: 8px; margin-bottom: 15px; }
-    """
-    
-    with gr.Blocks(title="Axon v26", css=css) as demo:
+    with gr.Blocks(title="Axon v26") as demo:
         # Header
         gr.HTML("""
         <div class="main-header">
@@ -647,7 +641,7 @@ def create_ui():
                 shell_desc = gr.Textbox(lines=3, placeholder="Describe what you want to do...", label="Description")
                 shell_type = gr.Dropdown(choices=["Bash", "PowerShell", "Zsh", "Fish"], value="Bash", label="Shell")
                 shell_btn = gr.Button("Generate", variant="primary")
-                shell_output = gr.Code(label="Generated Command", language="bash")
+                shell_output = gr.Code(label="Generated Command", language="shell")
             
             # === CRON TAB ===
             with gr.Tab("⏰ Cron"):
@@ -689,7 +683,7 @@ def create_ui():
                     format_from = gr.Dropdown(choices=["JSON", "YAML", "XML", "CSV", "TOML"], value="JSON", label="From")
                     format_to = gr.Dropdown(choices=["JSON", "YAML", "XML", "CSV", "TOML"], value="YAML", label="To")
                 format_btn = gr.Button("Convert", variant="primary")
-                format_output = gr.Code(label="Converted Data", language="yaml")
+                format_output = gr.Code(label="Converted Data", language="python")
             
             # === REFACTOR TAB (NEW) ===
             with gr.Tab("🎨 Refactor"):
@@ -809,4 +803,8 @@ if __name__ == "__main__":
     
     print("\n🚀 Starting Axon...")
     demo = create_ui()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, css="""
+    .main-header { text-align: center; margin-bottom: 20px; }
+    .tool-description { color: #666; font-size: 0.9em; margin-bottom: 10px; }
+    .status-bar { padding: 10px; border-radius: 8px; margin-bottom: 15px; }
+    """)