Update app.py

app.py

@@ -149,6 +149,17 @@ You can ask in natural language, e.g.:
 RE_TA = re.compile(r"\bTA\s+([A-Za-z0-9_\-]+)(?:\s+(\d+[mhHdD]))?(?:\s+lookback\s+(\d+))?", re.IGNORECASE)
 RE_LONG = re.compile(r"\bLONG\s+([A-Za-z0-9_\-]+)\s+([\d.]+)(?:\s+at\s+(market|mkt|[\d.]+))?(?:.*?\bSL\s+([\d.%]+))?(?:.*?\bTP\s+([\d.%]+))?", re.IGNORECASE)
 RE_SHORT = re.compile(r"\bSHORT\s+([A-Za-z0-9_\-]+)\s+([\d.]+)(?:\s+at\s+(market|mkt|[\d.]+))?(?:.*?\bSL\s+([\d.%]+))?(?:.*?\bTP\s+([\d.%]+))?", re.IGNORECASE)
+RE_CLOSE = re.compile(r"\b(close|exit|flatten)\s+(all|[A-Za-z0-9_\-]+)(?:\s+(\d+)%|\s+([\d.]+))?", re.IGNORECASE)
+
+def _close_desc(coin_or_all: str, pct: str | None, qty: str | None) -> str:
+    coin_or_all = coin_or_all.upper()
+    if coin_or_all == "ALL":
+        return "Close ALL open positions at market"
+    if pct:
+        return f"Close {pct}% of {coin_or_all} position at market"
+    if qty:
+        return f"Reduce {coin_or_all} position by {qty} units at market"
+    return f"Close {coin_or_all} position at market"
 
 def pct_or_price(s):
     if not s:
@@ -352,6 +363,61 @@ async def handle_message(message: str, history: list[tuple[str, str]]):
         + COMMAND_HELP
     )
 
+    mclose = RE_CLOSE.search(text)
+    if mclose:
+        # groups: verb, coin_or_all, pct (digits%), qty (number)
+        coin_or_all = mclose.group(2).strip()
+        pct = mclose.group(3)  # e.g., "50" meaning 50%
+        qty = mclose.group(4)  # absolute size to reduce
+        desc = _close_desc(coin_or_all, pct, qty)
+    
+        hyper_servers = []
+        news_servers = []
+        ta_servers = []
+        try:
+            # Tools for trader context (optional but helpful)
+            news_servers = make_crypto_news_mcp_servers()
+            ta_servers   = make_technical_analyst_mcp_servers()
+            hyper_servers = make_hyperliquid_trader_mcp_servers()
+    
+            await asyncio.gather(
+                connect_all(news_servers),
+                connect_all(ta_servers),
+                connect_all(hyper_servers),
+            )
+    
+            news_tool = await build_news_tool(news_servers)
+            ta_tool   = await build_ta_tool(ta_servers)
+            trader    = await build_trader(hyper_servers, [news_tool, ta_tool])
+    
+            # Natural-language prompt to place the close orders via Hyperliquid MCP
+            # (The trader agent already has the rules + account context)
+            trade_prompt = f"""
+            User request: {desc}.
+            Instructions:
+            - If 'ALL', close every open position at market.
+            - If a coin is specified:
+                - If a percent is provided, close that % of the CURRENT open position size.
+                - If a qty is provided, reduce by that absolute base-asset amount.
+                - If neither provided, fully close that coin position.
+            - Include SL/TP cleanup if needed (cancel/replace any attached orders).
+            - If the coin has no open position, report that clearly.
+            - Return a concise execution summary listing each order (coin, side, size, order type, price if applicable) and rationale.
+            """
+            with trace("close_positions"):
+                result = await Runner.run(trader, trade_prompt, max_turns=20)
+    
+            save_memory(f"[{now_sgt():%Y-%m-%d %H:%M}] Close: {desc}")
+            return f"### 🧹 Close — {desc}\n\n{result.final_output}"
+        except Exception as e:
+            return f"❌ Close error: `{e}`"
+        finally:
+            await asyncio.gather(
+                close_all(news_servers),
+                close_all(ta_servers),
+                close_all(hyper_servers),
+            )
+
 # ---------- Pretty printing for account/positions ----------
 
 from math import isnan