Update app.py

app.py

@@ -93,7 +93,6 @@ class PTYTerminal:
                 # Child process — become the shell
                 os.close(pid)
                 os.setsid()
-                # Set slave as controlling terminal
                 slave_fd = fd
                 fcntl.ioctl(slave_fd, termios.TIOCSCTTY, 0)
                 os.dup2(slave_fd, 0)
@@ -102,21 +101,18 @@ class PTYTerminal:
                 if slave_fd > 2:
                     os.close(slave_fd)
                 env = os.environ.copy()
-                env["TERM"] = "dumb"  # Prevent color codes / fancy prompts
-                env["PS1"] = "$ "     # Simple prompt
+                env["TERM"] = "dumb"
+                env["PS1"] = "$ "
                 env["DEBIAN_FRONTEND"] = "noninteractive"
                 os.execvpe("/bin/bash", ["/bin/bash", "--norc", "--noprofile", "-i"], env)
             else:
                 # Parent process
                 os.close(fd)
                 self.master_fd = pid
-                # Set non-blocking reads
                 flags = fcntl.fcntl(self.master_fd, fcntl.F_GETFL)
                 fcntl.fcntl(self.master_fd, fcntl.F_SETFL, flags | os.O_NONBLOCK)
-                # Set terminal size (cols x rows)
                 winsize = struct.pack("HHHH", 40, 120, 0, 0)
                 fcntl.ioctl(self.master_fd, termios.TIOCSWINSZ, winsize)
-                # Drain initial bash output
                 time.sleep(0.3)
                 self._read_output()
         except Exception as e:
@@ -139,7 +135,7 @@ class PTYTerminal:
                     chunk = os.read(self.master_fd, 4096)
                     if chunk:
                         output.append(chunk.decode("utf-8", errors="replace"))
-                        deadline = time.time() + 0.15  # Extend deadline while data flows
+                        deadline = time.time() + 0.15
                     else:
                         break
                 else:
@@ -152,7 +148,6 @@ class PTYTerminal:
     def _clean_output(self, text: str) -> str:
         """Strip ANSI escape codes and control characters."""
         cleaned = self.STRIP_ANSI.sub("", text)
-        # Remove null bytes and other control chars (keep newlines/tabs)
         cleaned = "".join(ch for ch in cleaned if ch == "\n" or ch == "\t" or (ord(ch) >= 32))
         return cleaned
 
@@ -185,53 +180,46 @@ class PTYTerminal:
 
             self._append_line(f"$ {cmd}")
 
-            # Write the command to bash
             try:
                 os.write(self.master_fd, (cmd + "\n").encode())
             except OSError as e:
                 self._append_line(f"[Write Error] {e}")
                 return self.get_log()
 
-            # Determine how long to wait based on command type
             parts = cmd.split()
             base = parts[0].lower() if parts else ""
             long_commands = ["pip", "pip3", "npm", "npx", "apt-get", "apt", "git", "wget",
                              "curl", "make", "cmake", "cargo", "yarn", "conda"]
 
             if base in long_commands:
-                wait_time = 180  # 3 minutes for installs
+                wait_time = 180
             elif base in ("python", "python3", "node"):
                 wait_time = 60
             else:
                 wait_time = 15
 
-            # Read output with smart timeout — stop early when command finishes
             all_output = []
             start = time.time()
             idle_count = 0
-            
+
             while time.time() - start < wait_time:
                 chunk = self._read_output(0.3)
                 if chunk:
                     idle_count = 0
                     all_output.append(chunk)
-                    # Check if we see our prompt back (command finished)
                     combined = "".join(all_output)
                     if combined.rstrip().endswith("$"):
                         break
                 else:
                     idle_count += 1
-                    # For short commands, bail after a brief idle
                     if base not in long_commands and idle_count >= 3:
                         break
-                    # For long commands, bail after longer idle
                     if base in long_commands and idle_count >= 10:
                         break
 
             raw = "".join(all_output)
             cleaned = self._clean_output(raw)
 
-            # Remove the echoed command from output (bash echoes what you type)
             lines = cleaned.split("\n")
             filtered = []
             skip_echo = True
@@ -240,7 +228,6 @@ class PTYTerminal:
                 if skip_echo and stripped == cmd.strip():
                     skip_echo = False
                     continue
-                # Skip bare prompt lines
                 if stripped == "$" or stripped == "$ ":
                     continue
                 filtered.append(line)
@@ -252,11 +239,10 @@ class PTYTerminal:
             return self.get_log()
 
     def run_editor_code(self, code: str) -> str:
-        """Execute code from the editor by writing it to a temp file and running it."""
+        """Execute code from the editor by writing to a temp file."""
         with self.lock:
             self._append_line("$ python [editor]")
-        
-        # Write code to a temp file and run it through the PTY
+
         tmp_path = "/tmp/_axon_editor_run.py"
         try:
             with open(tmp_path, "w") as f:
@@ -266,20 +252,17 @@ class PTYTerminal:
                 self._append_line(f"[Error] Could not write temp file: {e}")
             return self.get_log()
 
-        # Run through the shell (but bypass the echo/prompt logic — use subprocess for cleaner output)
         try:
             result = subprocess.run(
                 [sys.executable, tmp_path],
                 capture_output=True, text=True, timeout=30
             )
-            if result.stdout.strip():
-                with self.lock:
+            with self.lock:
+                if result.stdout.strip():
                     self._append_line(result.stdout.rstrip())
-            if result.stderr.strip():
-                with self.lock:
+                if result.stderr.strip():
                     self._append_line(result.stderr.rstrip())
-            if not result.stdout.strip() and not result.stderr.strip():
-                with self.lock:
+                if not result.stdout.strip() and not result.stderr.strip():
                     self._append_line("(No output)")
         except subprocess.TimeoutExpired:
             with self.lock:
@@ -430,7 +413,6 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
     with gr.Tabs() as main_tabs:
         with gr.Tab("Editor", id="editor-tab"):
             with gr.Row(equal_height=True):
-                # Sidebar
                 with gr.Column(scale=1, min_width=200):
                     gr.Markdown("### 📁 Explorer")
                     file_list = gr.Dropdown(
@@ -447,7 +429,6 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
                         placeholder="script.py", label="New File", container=False
                     )
 
-                # Editor
                 with gr.Column(scale=4):
                     editor = gr.Code(
                         value=fs.get_current_file_content(),
@@ -491,7 +472,7 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
 
                         with gr.Tab("AI Chat", id="chat-tab"):
                             chat_history = gr.Chatbot(
-                                label="Axon AI", height=300, type="messages"
+                                label="Axon AI", height=300
                             )
                             with gr.Row():
                                 chat_input = gr.Textbox(
@@ -525,7 +506,7 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
     diff_modified_state = gr.State("")
     diff_mode_state = gr.State(False)
 
-    # --- Handler Functions ---
+    # --- Handlers ---
     def update_file(content):
         fs.save_file(content)
         return fs.get_current_file_content()
@@ -586,7 +567,6 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
     file_list.change(load_file, file_list, [editor, current_file_state])
     new_file_btn.click(create_file, new_file_name, [file_list, editor])
 
-    # Terminal
     terminal_input.submit(
         terminal_command, terminal_input, [terminal_output, terminal_input]
     )
@@ -595,11 +575,9 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
     )
     clear_btn.click(clear_terminal_fn, None, terminal_output)
 
-    # Run button
     run_btn.click(run_editor_wrapper, editor, terminal_output)
     complete_btn.click(complete_wrapper, editor, editor)
 
-    # AI features → Diff tab
     explain_btn.click(
         explain_wrapper,
         editor,
@@ -615,30 +593,15 @@ with gr.Blocks(title="Axon Pro - Python IDE") as demo:
     chat_input.submit(
         generate_wrapper,
         [chat_input, chat_history],
-        [
-            diff_view,
-            diff_original_state,
-            diff_modified_state,
-            diff_mode_state,
-            chat_history,
-            chat_input,
-        ],
+        [diff_view, diff_original_state, diff_modified_state, diff_mode_state, chat_history, chat_input],
     ).then(lambda: gr.Tabs(selected="diff-tab"), None, main_tabs)
 
     send_btn.click(
         generate_wrapper,
         [chat_input, chat_history],
-        [
-            diff_view,
-            diff_original_state,
-            diff_modified_state,
-            diff_mode_state,
-            chat_history,
-            chat_input,
-        ],
+        [diff_view, diff_original_state, diff_modified_state, diff_mode_state, chat_history, chat_input],
     ).then(lambda: gr.Tabs(selected="diff-tab"), None, main_tabs)
 
-    # Diff actions
     apply_btn.click(
         lambda mod: (mod, False), diff_modified_state, [editor, diff_mode_state]
     ).then(lambda: gr.Tabs(selected="editor-tab"), None, main_tabs)