Update app.py

app.py

@@ -43,7 +43,55 @@ symbol_volumes = {
     "EURUSD":  0.01,
     "_DEFAULT_FOREX": 0.01,  # catch-all for any forex pair not explicitly listed
 }
+
+# Per-symbol BUY/SELL toggle (controlled from dashboard, NOT TradingView)
+symbol_status = {
+    "NAS100":  {"buy": True, "sell": True},
+    "NAS100m": {"buy": True, "sell": True},
+    "USTEC":   {"buy": True, "sell": True},
+    "US500":   {"buy": True, "sell": True},
+    "US500m":  {"buy": True, "sell": True},
+    "US30":    {"buy": True, "sell": True},
+    "US30m":   {"buy": True, "sell": True},
+    "BTCUSD":  {"buy": True, "sell": True},
+    "EURUSD":  {"buy": True, "sell": True},
+    "_DEFAULT_FOREX": {"buy": True, "sell": True},
+}
+
+CONFIG_FILE = "bot_config.json"
+
+def load_config():
+    """Load persisted volumes + status toggles from disk."""
+    global symbol_volumes, symbol_status
+    if os.path.exists(CONFIG_FILE):
+        try:
+            with open(CONFIG_FILE, "r") as f_conf:
+                config = json.load(f_conf)
+                if "volumes" in config:
+                    symbol_volumes.update(config["volumes"])
+                if "status" in config:
+                    for k, v in config["status"].items():
+                        if k in symbol_status:
+                            symbol_status[k].update(v)
+                        else:
+                            symbol_status[k] = v
+            logger.info("Configuration loaded from bot_config.json")
+        except Exception as e:
+            logger.error(f"Error loading bot_config.json: {e}")
+
+def save_config():
+    """Persist current volumes + status toggles to disk."""
+    try:
+        config = {"volumes": symbol_volumes, "status": symbol_status}
+        with open(CONFIG_FILE, "w") as f_conf:
+            json.dump(config, f_conf, indent=4)
+        logger.info("Configuration saved to bot_config.json")
+    except Exception as e:
+        logger.error(f"Error saving bot_config.json: {e}")
+
+load_config()
 logger.info(f"Per-symbol volumes initialized: {symbol_volumes}")
+logger.info(f"Per-symbol statuses initialized: {symbol_status}")
 
 # Global caches to eliminate latency of symbol resolution and limits querying
 SYMBOL_ID_CACHE = {}      # symbol_name_upper -> symbol_id
@@ -782,20 +830,44 @@ async def dashboard(request: Request):
         if sym not in symbol_volumes:
             continue
         vol = symbol_volumes[sym]
+        status = symbol_status.get(sym, {"buy": True, "sell": True})
+        buy_on = status.get("buy", True)
+        sell_on = status.get("sell", True)
         display_name = "Default Forex" if sym == "_DEFAULT_FOREX" else sym
         row_bg = "#0f172a" if display_order.index(sym) % 2 == 0 else "#111827"
+        # BUY toggle button: green if ON, grey if OFF
+        buy_color = "#00E676" if buy_on else "#475569"
+        buy_text = "BUY ✓" if buy_on else "BUY ✗"
+        buy_val = "false" if buy_on else "true"  # clicking toggles to opposite
+        # SELL toggle button: red if ON, grey if OFF
+        sell_color = "#FF5252" if sell_on else "#475569"
+        sell_text = "SELL ✓" if sell_on else "SELL ✗"
+        sell_val = "false" if sell_on else "true"
         volume_rows_html += f'''
                     <tr style="border-bottom: 1px solid #1e293b; background: {row_bg};">
                         <td style="padding: 8px 12px; color: #e2e8f0; font-size: 14px; font-weight: 600;">{display_name}</td>
                         <td style="padding: 8px 12px;">
-                            <form action="/set-volume" method="post" style="display: flex; gap: 8px; align-items: center;">
+                            <form action="/set-volume" method="post" style="display: inline-flex; gap: 8px; align-items: center;">
                                 <input type="hidden" name="symbol" value="{sym}">
                                 <input type="text" name="volume" value="{vol:.2f}"
                                     style="background: #1e293b; border: 1px solid #334155; border-radius: 6px; padding: 6px 10px; color: #e2e8f0; font-size: 14px; width: 80px;">
                                 <button type="submit" class="btn" style="padding: 6px 14px; font-size: 12px;">Update</button>
                             </form>
                         </td>
-                        <td style="padding: 8px 12px; color: #64748b; font-size: 11px;">{vol:.2f} lots</td>
+                        <td style="padding: 8px 12px;">
+                            <form action="/toggle-status" method="post" style="display: inline-flex; gap: 6px;">
+                                <input type="hidden" name="symbol" value="{sym}">
+                                <input type="hidden" name="direction" value="buy">
+                                <input type="hidden" name="new_value" value="{buy_val}">
+                                <button type="submit" style="background: {buy_color}; color: white; border: none; border-radius: 6px; padding: 6px 12px; font-size: 12px; font-weight: 700; cursor: pointer;">{buy_text}</button>
+                            </form>
+                            <form action="/toggle-status" method="post" style="display: inline-flex; gap: 6px;">
+                                <input type="hidden" name="symbol" value="{sym}">
+                                <input type="hidden" name="direction" value="sell">
+                                <input type="hidden" name="new_value" value="{sell_val}">
+                                <button type="submit" style="background: {sell_color}; color: white; border: none; border-radius: 6px; padding: 6px 12px; font-size: 12px; font-weight: 700; cursor: pointer;">{sell_text}</button>
+                            </form>
+                        </td>
                     </tr>'''
 
     # 2. Construct OAuth status card HTML
@@ -834,7 +906,7 @@ async def dashboard(request: Request):
                     <tr style="border-bottom: 1px solid #334155;">
                         <th style="text-align: left; padding: 8px 12px; color: #94a3b8; font-size: 13px;">Symbol</th>
                         <th style="text-align: left; padding: 8px 12px; color: #94a3b8; font-size: 13px;">Lot Size</th>
-                        <th style="padding: 8px 12px;"></th>
+                        <th style="text-align: left; padding: 8px 12px; color: #94a3b8; font-size: 13px;">Allow Trade</th>
                     </tr>
                 </thead>
                 <tbody>
@@ -1151,6 +1223,7 @@ async def set_volume(request: Request, symbol: str = Form(...), volume: str = Fo
         vol_float = float(volume.strip())
         symbol_upper = symbol.strip().upper()
         symbol_volumes[symbol_upper] = vol_float
+        save_config()
         logger.info(f"Trade volume for symbol '{symbol_upper}' updated to {vol_float} lots.")
     except Exception as e:
         logger.error(f"Invalid volume update for symbol '{symbol}': {volume}. Error: {e}")
@@ -1164,6 +1237,31 @@ async def set_volume(request: Request, symbol: str = Form(...), volume: str = Fo
         redirect_url = f"{protocol}://{host}/"
     return RedirectResponse(redirect_url, status_code=303)
 
+@app.post("/toggle-status")
+async def toggle_status(request: Request, symbol: str = Form(...), direction: str = Form(...), new_value: str = Form(...)):
+    """Instantly toggle Allow BUY or Allow SELL for a symbol from the dashboard."""
+    global symbol_status
+    symbol_upper = symbol.strip().upper()
+    dir_lower = direction.strip().lower()
+    new_bool = new_value.strip().lower() == "true"
+    
+    if symbol_upper not in symbol_status:
+        symbol_status[symbol_upper] = {"buy": True, "sell": True}
+    symbol_status[symbol_upper][dir_lower] = new_bool
+    save_config()
+    
+    state_str = "ENABLED" if new_bool else "DISABLED"
+    logger.info(f"{dir_lower.upper()} {state_str} for {symbol_upper}")
+    
+    space_host = os.environ.get("SPACE_HOST")
+    if space_host:
+        redirect_url = f"https://{space_host}/"
+    else:
+        host = request.headers.get("host", "")
+        protocol = "https" if "hf.space" in host or request.headers.get("x-forwarded-proto") == "https" else "http"
+        redirect_url = f"{protocol}://{host}/"
+    return RedirectResponse(redirect_url, status_code=303)
+
 @app.post("/webhook")
 async def webhook(request: Request):
     # Parse Webhook JSON data
@@ -1178,6 +1276,19 @@ async def webhook(request: Request):
     
     if action not in ["BUY", "SELL", "STOP-TRADE", "CLOSE", "ANCHOR"]:
         raise HTTPException(status_code=400, detail="Invalid action. Must be BUY, SELL, STOP-TRADE, CLOSE, or ANCHOR")
+    
+    # ── Dashboard BUY/SELL toggle filter ──
+    # If the user disabled BUY or SELL for this symbol on the dashboard, ignore it instantly
+    if action in ["BUY", "SELL"]:
+        status = symbol_status.get(ticker, symbol_status.get("_DEFAULT_FOREX", {"buy": True, "sell": True}))
+        is_allowed = status.get(action.lower(), True)
+        if not is_allowed:
+            ignore_msg = f"{action} for {ticker} IGNORED — disabled on dashboard"
+            logger.info(ignore_msg)
+            WEBHOOK_LOGS.append({"time": datetime.now().strftime('%H:%M:%S'), "action": f"{action} BLOCKED", "color": "#94a3b8", "msg": ignore_msg})
+            if len(WEBHOOK_LOGS) > 50:
+                WEBHOOK_LOGS.pop(0)
+            return {"status": "ignored", "message": ignore_msg}
         
     # Load and refresh tokens
     tokens = get_valid_tokens()