Update app.py

app.py

@@ -766,6 +766,7 @@ class IssueManager:
              if e.status == 404: error_msg = f"Error: Repository not found at {self.repo_url}."
              elif e.status == 401: error_msg = "Error: Invalid GitHub token or insufficient permissions for this repository."
              elif e.status == 403:
+                 rate_limit_remaining = e.headers.get('X-RateLimit-Remaining') # FIX: Access rate_limit_remaining from error headers
                  rate_limit_reset = e.headers.get('X-RateLimit-Reset')
                  reset_time_str = "unknown"
                  if rate_limit_reset:
@@ -1572,12 +1573,11 @@ Suggested Resolution Approach:
                          if self.main_loop.is_running():
                              self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
                          else:
-                              logger.error("Main loop not running, cannot schedule client removal.")
+                              logger.warning("Main loop not running, cannot schedule client removal.")
                          continue # Skip to next client
 
 
                 # No need to await tasks here if remove_ws_client is scheduled via call_soon_threadsafe
-                # The exceptions will be handled by the individual send calls.
 
             except asyncio.CancelledError:
                 logger.info("Broadcast loop cancelled.")
@@ -1633,6 +1633,7 @@ Suggested Resolution Approach:
                          tasks.append(client.send(update_payload))
                      except (ConnectionClosed, ConnectionAbortedError, ConnectionResetError, WebSocketException) as e:
                          logger.warning(f"Client {getattr(client, 'client_id', client.remote_address)} seems disconnected during broadcast: {type(e).__name__}. Attempting removal.")
+                         # Schedule removal in the main loop if not already in it
                          if self.main_loop.is_running():
                              self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
                          else:
@@ -1742,7 +1743,7 @@ Suggested Resolution Approach:
          tasks = []
          # No need for disconnected_clients list here, remove_ws_client handles it
 
-        for client in active_clients_snapshot:
+         for client in active_clients_snapshot:
             # Check if client is closed using the standard websockets property
             if client.closed:
                 continue
@@ -1763,9 +1764,9 @@ Suggested Resolution Approach:
                     logger.warning("Main loop not running, cannot schedule client removal.")
                 continue
 
-        if tasks:
-            logger.debug(f"Broadcasting single status update to {len(tasks)} clients.")
-            # No need to await tasks here if remove_ws_client is scheduled via call_soon_threadsafe
+         if tasks:
+             logger.debug(f"Broadcasting single status update to {len(tasks)} clients.")
+             # No need to await tasks here if remove_ws_client is scheduled via call_soon_threadsafe
 
 
     def _identify_stale_issues(self):
@@ -2593,7 +2594,7 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
                                     files_content[file_path_str] = f"# File path resolved outside repository: {file_path_str}"
                              else:
                                 logger.warning(f"On-demand file path {file_path_str} not found or not a file during load for issue {selected_id}.")
-                                files_content[file_path_str] = f"# File not found or not a file: {file_path_str}"
+                                 files_content[file_path_str] = f"# File not found or not a file: {file_path_str}"
                          except Exception as e:
                              logger.warning(f"Error reading on-demand file {full_path} for issue {selected_id}: {e}")
                              files_content[file_path_str] = f"# Error reading file: {e}"
@@ -2669,7 +2670,7 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
         # Expanded stop words list (lowercase)
         stop_words = set("a an the is are was were be been being has have had do does did will would shall should can could may might must it its this that these those there their then than and or but if because as at by for with without about against between into through during before after above below to from up down in out on off over under again further then once here there when where why how all any both each few more most other some such no nor not only own same so than too very s t can will just don should now d ll m o re ve yo ain aren couldn didn doesn isn it's mustn ouren shan shouldn wasn weren won wouldn".split())
         # Add common code/issue terms (lowercase)
-        stop_words.update({'issue', 'bug', 'error', 'fix', 'feat', 'chore', 'refactor', 'docs', 'test', 'file', 'line', 'code', 'when', 'user', 'report', 'problem', 'need', 'want', 'get', 'use', 'try', 'make', 'add', 'remove', 'change', 'update', 'create', 'build', 'run', 'start', 'stop', 'click', 'button', 'page', 'app', 'data', 'system', 'service', 'component', 'module', 'function', 'class', 'method', 'variable', 'value', 'string', 'number', 'list', 'dict', 'object', 'array', 'true', 'false', 'null', 'none', 'type', 'size', 'color', 'style', 'width', 'height', 'margin', 'padding', 'border', 'display', 'position', 'float', 'clear', 'overflow', 'index', 'key', 'value', 'id', 'name', 'text', 'html', 'css', 'js', 'py', 'java', 'c', 'cpp', 'go', 'rs', 'php', 'rb', 'json', 'yaml', 'xml', 'sql', 'log', 'config', 'setup', 'install', 'version', 'release', 'branch', 'commit', 'pull', 'request', 'merge', 'diff', 'patch', 'context', 'suggestion', 'analysis', 'missing', 'information', 'duplicate', 'potential', 'cluster', 'severity', 'label', 'assignee', 'state', 'title', 'body', 'url', 'created', 'updated', 'time', 'day', 'week', 'month', 'year', 'ago', 'now', 'new', 'old', 'different', 'same', 'similar', 'group', 'count', 'distribution', 'analytics', 'overview', 'studio', 'board', 'resolution', 'collaborative', 'editor', 'context', 'aware', 'assistance', 'tools', 'output', 'patch', 'steps', 'approach', 'plan', 'identify', 'propose', 'recommendation', 'workflow', 'standard', 'important', 'critical', 'high', 'medium', 'low', 'unknown', 'none', 'needed', 'analysis', 'hypothesis', 'preliminary', 'relevant', 'area', 'investigation', 'implementation', 'testing', 'next'})
+        stop_words.update({'issue', 'bug', 'error', 'fix', 'feat', 'chore', 'refactor', 'docs', 'test', 'file', 'line', 'code', 'when', 'user', 'report', 'problem', 'need', 'want', 'get', 'use', 'try', 'make', 'add', 'remove', 'change', 'update', 'create', 'build', 'run', 'start', 'stop', 'click', 'button', 'page', 'app', 'data', 'system', 'service', 'component', 'module', 'function', 'class', 'method', 'variable', 'value', 'string', 'number', 'list', 'dict', 'object', 'array', 'true', 'false', 'null', 'none', 'type', 'size', 'color', 'style', 'width', 'height', 'margin', 'padding', 'border', 'display', 'position', 'float', 'clear', 'overflow', 'index', 'key', 'value', 'id', 'name', 'text', 'html', 'css', 'js', 'py', 'java', 'c', 'cpp', 'h', 'hpp', 'go', 'rs', 'php', 'rb', 'json', 'yaml', 'xml', 'sql', 'log', 'config', 'setup', 'install', 'version', 'release', 'branch', 'commit', 'pull', 'request', 'merge', 'diff', 'patch', 'context', 'suggestion', 'analysis', 'missing', 'information', 'duplicate', 'potential', 'cluster', 'severity', 'label', 'assignee', 'state', 'title', 'body', 'url', 'created', 'updated', 'time', 'day', 'week', 'month', 'year', 'ago', 'now', 'new', 'old', 'different', 'same', 'similar', 'group', 'count', 'distribution', 'analytics', 'overview', 'studio', 'board', 'resolution', 'collaborative', 'editor', 'context', 'aware', 'assistance', 'tools', 'output', 'patch', 'steps', 'approach', 'plan', 'identify', 'propose', 'recommendation', 'workflow', 'standard', 'important', 'critical', 'high', 'medium', 'low', 'unknown', 'none', 'needed', 'analysis', 'hypothesis', 'preliminary', 'relevant', 'area', 'investigation', 'implementation', 'testing', 'next'})
 
 
         def get_top_keywords(indices):
@@ -3044,7 +3045,7 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
                                 }}
                          }} else {{
                              console.warn("Row for issue ID", issueId, "not found in the dataframe.");
-                             updateStatusBar(`Issue #${issueId} not found in the current list. Try re-scanning.`, true);
+                             updateStatusBar(`Issue #${{issueId}} not found in the current list. Try re-scanning.`, true);
                          }}
                     }};
 
@@ -3397,7 +3398,8 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
                                  // Get the text content of the AI output Markdown block
                                  const text = aiOutput.textContent || aiOutput.innerText;
                                  // Use regex to find the content within the ```diff ``` block
-                                 const diffMatch = text.match(/```diff\n(.*?)```/s);
+                                 // FIX: Escape the backslash in the regex for newline
+                                 const diffMatch = text.match(/```diff\\n(.*?)```/s);
                                  if (diffMatch && diffMatch[1]) {{
                                      navigator.clipboard.writeText(diffMatch[1].trim()).then(() => {{
                                          console.log("Patch copied to clipboard.");
@@ -3422,7 +3424,7 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
 
                 }})(); // End IIFE
             </script>
-            }"""
+            """
         # The _js parameter injects the JavaScript code into the Gradio page
         demo_app.load(_js=web_socket_js(WS_PORT, GRADIO_PORT), fn=None, inputs=None, outputs=None)
 
@@ -3436,7 +3438,7 @@ async def handle_ws_connection(websocket: WebSocketServerProtocol, path: str, ma
     setattr(websocket, 'client_id', client_id)
     remote_addr = websocket.remote_address
     logger.info(f"WebSocket client connected: {remote_addr} assigned ID {client_id}")
-    
+
     # Add the new client to the list of active clients
     manager.ws_clients.append(websocket)
     logger.info(f"Client list size: {len(manager.ws_clients)}")
@@ -3483,11 +3485,8 @@ async def handle_ws_connection(websocket: WebSocketServerProtocol, path: str, ma
                     delta_str = data.get("delta")
                     # Ensure data is valid and sender ID matches
                     if issue_num is not None and delta_str is not None and sender_id == client_id:
-                        # Schedule the update handling in the main loop
-                        if manager.main_loop.is_running():
-                             manager.main_loop.call_soon_threadsafe(asyncio.create_task, manager.handle_code_editor_update(int(issue_num), delta_str, sender_id))
-                        else:
-                             logger.warning("Main loop not running, cannot process code update.")
+                        # FIX: Corrected call - handle_ws_connection is already async, just await the async manager method
+                        await manager.handle_code_editor_update(int(issue_num), delta_str, sender_id)
                     else:
                          logger.warning(f"Invalid or unauthorized 'code_update' message from {sender_id}: Missing issue_num/delta or sender mismatch. Data: {str(data)[:200]}")
 
@@ -3796,4 +3795,4 @@ if __name__ == "__main__":
             logger.info("Closing asyncio loop.")
             loop.close()
 
-        logger.info("Application shutdown complete.")
+        logger.info("Application shutdown complete.")