Update app.py

app.py

@@ -147,8 +147,7 @@ def create_diff_view(original: str, modified: str) -> str:
 
 # --- Gradio UI Configuration ---
 
-# 1. THEME FIX: Removed invalid 'radius_size' from .set()
-# We use standard color overrides only.
+# 1. THEME: Cleaned up to avoid errors
 theme = gr.themes.Default(
     font=[gr.themes.GoogleFont('JetBrains Mono'), 'monospace'],
     font_mono=[gr.themes.GoogleFont('JetBrains Mono'), 'monospace'],
@@ -159,10 +158,9 @@ theme = gr.themes.Default(
     button_primary_background_fill_hover="*primary_600",
     body_background_fill="*neutral_900",
     body_text_color="*neutral_50",
-    # radius_size="none"  <-- REMOVED to prevent TypeError
 )
 
-# 2. CSS FIX: Added strict border-radius rule for the "square" look
+# 2. CSS: Styling
 css = """
     /* Force square corners globally */
     * { border-radius: 0 !important; } 
@@ -238,7 +236,8 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
                                 send_btn = gr.Button("Generate", variant="primary", scale=1)
         
         with gr.Tab("Diff View", id="diff-tab"):
-            diff_view = gr.Code(label="AI Changes", language="diff", elem_classes="diff-view code-wrap", interactive=False)
+            # FIX: Changed language="diff" to language="javascript" to prevent crash
+            diff_view = gr.Code(label="AI Changes", language="javascript", elem_classes="diff-view code-wrap", interactive=False)
             with gr.Row():
                 apply_btn = gr.Button("Apply Changes", variant="primary")
                 discard_btn = gr.Button("Discard Changes", variant="secondary")
@@ -338,4 +337,6 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
     )
 
 if __name__ == "__main__":
+    # If the warning about theme/css being in launch() persists, you can move them here.
+    # But for now, we keep them in Blocks to support older/standard versions as well.
     demo.launch(server_name="0.0.0.0", server_port=7860)