Update app.py

app.py

@@ -15,7 +15,7 @@ MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 DEVICE = "cpu"
 DTYPE = torch.float32 
 
-# --- File System Class (Same as before) ---
+# --- File System Class ---
 class FileSystem:
     def __init__(self):
         self.files = {
@@ -89,6 +89,7 @@ def load_model():
 def run_code(code: str) -> Tuple[str, str]:
     start_time = time.time()
     try:
+        # Placeholder for actual execution logic
         output = f"✓ Code executed successfully\nExecution time: {time.time() - start_time:.4f}s"
         return output, ""
     except Exception as e:
@@ -139,29 +140,33 @@ def create_diff_view(original: str, modified: str) -> str:
     lines_original = original.split('\n')
     lines_modified = modified.split('\n')
     diff = []
-    # Very basic diff visualizer
     import difflib
     for line in difflib.unified_diff(lines_original, lines_modified, lineterm=""):
         diff.append(line)
     return "\n".join(diff)
 
-# --- Gradio UI ---
-# 1. FIXED THEME DEFINITION
+# --- Gradio UI Configuration ---
+
+# 1. THEME FIX: Removed invalid 'radius_size' from .set()
+# We use standard color overrides only.
 theme = gr.themes.Default(
     font=[gr.themes.GoogleFont('JetBrains Mono'), 'monospace'],
     font_mono=[gr.themes.GoogleFont('JetBrains Mono'), 'monospace'],
     primary_hue="blue",
     neutral_hue="gray",
-    # radius=0  <-- REMOVED THIS CAUSE OF ERROR
 ).set(
     button_primary_background_fill="*primary_500",
     button_primary_background_fill_hover="*primary_600",
     body_background_fill="*neutral_900",
     body_text_color="*neutral_50",
-    radius_size="none" # <-- Added this to achieve square corners
+    # radius_size="none"  <-- REMOVED to prevent TypeError
 )
 
+# 2. CSS FIX: Added strict border-radius rule for the "square" look
 css = """
+    /* Force square corners globally */
+    * { border-radius: 0 !important; } 
+    
     .editor-container { height: 50vh; }
     .terminal { height: 15vh; background-color: #1e1e1e; color: #d4d4d4; overflow: auto; }
     .diff-view { height: 40vh; overflow: auto; }
@@ -181,7 +186,6 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
     
     gr.Markdown("# ⚡ Axon Pro — Free AI-Powered Code IDE")
     
-    # 2. FIXED TABS VARIABLE ASSIGNMENT
     with gr.Tabs() as main_tabs:
         with gr.Tab("Editor", id="editor-tab"):
             with gr.Row(equal_height=True):
@@ -241,13 +245,13 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
     
     status_bar = gr.Markdown(f"**AXON PRO v1.0** | CPU Mode | TinyLlama-1.1B", elem_classes="status-bar")
     
-    # State
+    # State management
     current_file_state = gr.State(fs.current_file)
     diff_original_state = gr.State("")
     diff_modified_state = gr.State("")
     diff_mode_state = gr.State(False)
     
-    # Wrappers
+    # Event Handlers
     def update_file_content(content):
         fs.save_file(content)
         return fs.get_current_file_content()
@@ -290,7 +294,7 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
         new_hist = history + [[prompt, "Generated code available in Diff View"]]
         return diff, "", generated, True, new_hist, ""
 
-    # Event Wiring
+    # Binding Events
     editor.change(update_file_content, editor, None)
     file_list.change(load_file, file_list, [editor, current_file_state])
     new_file_btn.click(create_new_file, new_file_name, [file_list, editor])
@@ -299,7 +303,7 @@ with gr.Blocks(title="Axon Pro - Free AI IDE", theme=theme, css=css) as demo:
     complete_btn.click(complete_code_wrapper, editor, editor)
     clear_btn.click(lambda: "$ _", None, terminal)
     
-    # 3. FIXED TAB SWITCHING (Using main_tabs variable)
+    # Tab switching actions
     explain_btn.click(
         explain_code_wrapper, editor, 
         [diff_view, diff_original_state, diff_modified_state, diff_mode_state]