Added multi-workspace support, image support and other improvements

Author: MichaelG0501

.env.example

@@ -6,31 +6,31 @@
 
 # --- Telegram (REQUIRED) ---
 # Get from @BotFather on Telegram (/newbot)
-TELEGRAM_BOT_TOKEN=your_bot_token_here
+TELEGRAM_BOT_TOKEN=000000000:AAAAAAAAAA # Replace with actual
 
 # Get from @userinfobot on Telegram
-ALLOWED_CHAT_ID=your_chat_id_here
+ALLOWED_CHAT_ID=123456 #，0002,0003 so on
 
 # --- Connection Mode ---
 # How the bot connects to the environment where the AI agent runs.
 # Options:
 #   ssh    (in WSL or macOS connecting to HPC using SSH)
 #   wsl    (I am in WSL and connect to WSL)
 #   local  (in local connect to local, e.g., macOS/Linux)
-# CONNECTION_MODE=ssh
+CONNECTION_MODE=wsl
 
 # --- SSH ---
 # Must match a Host entry in your ~/.ssh/config if CONNECTION_MODE=ssh
-SSH_HOST=hpc
+SSH_HOST=
 
 # --- Agent Paths ---
-# Full path to the AI agent binary on the target machine
-OPENCODE_PATH=/path/to/opencode
+# Full path to the AI agent binary on the HPC
+OPENCODE_PATH=~/.opencode/bin/opencode
 
-# Default working directory on the target machine
-WORKDIR=/path/to/your/project
+# Default working directory on the HPC
+HPC_WORKDIR=~/
 
-# Default AI model (see README for alias list)
+# Default AI model (Select depending on your subscription)
 DEFAULT_MODEL=opencode/minimax-m2.5-free
 
 # --- Environment Setup (REQUIRED if your target uses environment modules) ---
@@ -55,10 +55,10 @@ DEFAULT_MODEL=opencode/minimax-m2.5-free
 #   Node installed globally, no setup needed:
 #   SETUP_CMD=
 #
-SETUP_CMD=module purge && module load tools/dev && module load nodejs/20.13.1-GCCcore-13.3.0
+SETUP_CMD=
 
 # --- rclone Integration (optional) ---
-# Destination for the @@upload:@@ rclone transfer directive.
+# Destination for /upload rclone transfer.
 # Format: remote_name:path  (as configured in `rclone config` on target)
 #
 # Examples:
@@ -67,7 +67,7 @@ SETUP_CMD=module purge && module load tools/dev && module load nodejs/20.13.1-GC
 #   s3:mybucket/results         (S3-compatible)
 #   sharepoint:Documents/HPC    (SharePoint)
 #
-RCLONE_DEST=gdrive:HPC-Results
+RCLONE_DEST=
 
 # --- OpenCode Session Database (optional) ---
 # Path to OpenCode's SQLite database ON THE TARGET (used by the chat viewer tool).
@@ -78,3 +78,18 @@ RCLONE_DEST=gdrive:HPC-Results
 # --- Session Persistence (optional) ---
 # File on the RELAY MACHINE where active session IDs are stored between messages.
 # SESSIONS_FILE=~/.hpc_relay_sessions.json
+
+
+# --- Workspace Separation (optional but recommended) ---
+# Explicit mapping per Telegram chat_id -> isolated workspace/session/task files.
+# JSON format example:
+# CHANNEL_WORKSPACES={"123456":{"name":"AA","workdir":"~/workspace_AA","allowed_users":[123456]},"-000000":{"name":"group_shared","workdir":"~/workspace_shared"}}
+# If name is set, session/task files become:
+#   hpc_relay_sessions_<name>.json
+#   hpc_relay_tasks_<name>.json
+CHANNEL_WORKSPACES=
+# Auto-create isolated workspace namespace by chat_id when no explicit mapping exists.
+# 0 = disabled (default), 1 = enabled
+AUTO_WORKSPACE_PER_CHAT=0
+# Prefix for auto workspace names: <prefix>_<chat_id>
+AUTO_WORKSPACE_PREFIX=chat

.gitignore

@@ -3,6 +3,7 @@
 *.env
 .env.*
 !.env.example
+hpcrelay/
 
 # Python
 __pycache__/
@@ -39,5 +40,8 @@ Thumbs.db
 opencode_chat_viewer/
 
 # Session data
-oc_sessions.json
-hpc_relay_sessions_ssh.json
+*session*.json
+*task*.json
+
+# Workspace data
+workspace*/

README.md

@@ -2,7 +2,7 @@
 
 # HPC-Relay
 
-### Control AI Coding Agents on Your HPC Cluster, WSL, or macOS -- From Your Phone
+### Opencode AI agent on your phone - Access anywhere!
 
 [![Python 3.9+](https://img.shields.io/badge/Python-3.9+-blue?style=flat-square&logo=python)](https://python.org)
 [![Telegram Bot API](https://img.shields.io/badge/Telegram-Bot%20API-26A5E4?style=flat-square&logo=telegram)](https://core.telegram.org/bots/api)
@@ -70,9 +70,9 @@ You authenticate **once**. Everything else is automatic.
 |---|---|
 | **No VPN on phone** | Relay sits inside the authenticated network |
 | **AI session memory** | Parses & re-injects `sessionID` -- full context from cold start |
-| **File transfer** | Download files from the target machine to Telegram (`@@send:path@@`) |
-| **File upload** | Upload files from target to cloud (`@@upload:file@@`) |
-| **Wildcard fetch** | Send multiple files with glob patterns (e.g. `@@send:Auto*.png@@`) |
+| **File transfer** | Download files from the target machine to Telegram (`/send <path>`) |
+| **File upload** | Upload files from target to cloud (`/upload <path>`) |
+| **Wildcard fetch** | Send multiple files with glob patterns (e.g. `/send Auto*.png`) |
 | **Chat history viewer** | Generate interactive HTML viewer of all AI conversations |
 | **Agent-agnostic** | Works with OpenCode, Claude Code, Aider, or any headless CLI agent |
 
@@ -82,22 +82,28 @@ You authenticate **once**. Everything else is automatic.
 
 ### Talking to the AI Agent
 
+> Use slash commands only.
+
+
 Simply type a message. It gets forwarded to the AI agent on your target machine, for example:
 
 ```
 Generate an expression heatmap with random synthetic data.
 ```
 
-### Inline Directives
+### Slash Commands
 
 | Syntax | Action | Example |
 |---|---|---|
-| `@@model: <alias>@@` | Change the AI model for this message | `@@model: opus46@@ refactor my script` |
-| `@@session: <id>@@` | Switch to a specific AI session | `@@session: ses_abc123@@` / `@@session: new@@` |
-| `@@kill@@` or `!kill` | Kill the currently running AI process | `@@kill@@` |
-| `@@send: <path>@@` | Download a file from target machine to Telegram | `@@send: ~/results/plot.png@@` |
-| `@@send: Auto*.png@@` | Download matching files (wildcard) | `@@send: /project/Auto*.png@@` |
-| `@@upload: file@@` | Upload a file from target machine to cloud | `@@upload: ~/data/input.csv@@` |
+| `/model` | Open interactive provider -> model picker | `/model` |
+| `/new` | Start a fresh AI session | `/new` |
+| `/id <session_id>` | Switch to a specific session ID | `/id ses_abc123...` |
+| `/id` | Show recent sessions and pick one interactively | `/id` |
+| `/kill` or `/q` | Kill the currently running AI process | `/kill` |
+| `/send <path>` | Download a file from target machine to Telegram | `/send ~/results/plot.png` |
+| `/send Auto*.png` | Download matching files (wildcard) | `/send /project/Auto*.png` |
+| `/upload <path>` | Upload a file from target machine to cloud | `/upload ~/data/input.csv` |
+| `/scheduled` | Open scheduled tasks manager (edit/delete interactively) | `/scheduled` |
 
 ### Shell Commands
 
@@ -114,13 +120,14 @@ Prefix with `!` to run raw shell commands directly on the target machine (bypass
 Change models persistently (applies to all subsequent messages):
 
 ```
-@@model: opus46@@
+/model
 ```
 
 Or change just for one message:
 
 ```
-@@model: g5@@ explain this error in my GWAS script
+(after selecting model via /model)
+explain this error in my GWAS script
 ```
 
 **Available model aliases (For Github Copilot Pro):**
@@ -140,8 +147,9 @@ Or change just for one message:
 Sessions provide conversation memory. The bot auto-tracks sessions, but you can:
 
 ```
-@@session: ses_abc123@@          # Switch to a specific session
-@@session: new@@                 # Start a fresh session (clear context)
+/id ses_abc123                   # Switch to a specific session
+/new                             # Start a fresh session (clear context)
+/id                              # Show recent sessions to pick
 ```
 
 The AI retains full context within a session -- ask follow-ups without re-explaining.
@@ -155,16 +163,16 @@ The AI retains full context within a session -- ask follow-ups without re-explai
 The AI automatically sends files it creates. You can also request files manually:
 
 ```
-@@send: ~/results/summary.pdf@@
-@@send: /project/output/Auto*.png@@       # wildcard: sends all matching files
+/send ~/results/summary.pdf
+/send /project/output/Auto*.png       # wildcard: sends all matching files
 ```
 
 ### Upload to cloud
 
 Upload any file to your pre-configured cloud storage:
 
 ```
-@@upload :~/data/input.csv@@
+/upload ~/data/input.csv
 ```
 
 The file will be transferred to your preferred cloud storage such as google drive or onedrive.
@@ -280,7 +288,7 @@ python3 tools/chat_viewer.py
 
 ```bash
 !python3 /path/to/tools/chat_viewer.py
-@@send:~/opencode_chat_viewer/index.html@@
+/send ~/opencode_chat_viewer/index.html
 # Can be downloaded and viewed on Telegram
 ```
 
@@ -311,7 +319,7 @@ Many HPC clusters **prohibit heavy computation on login nodes**. HPC-Relay is de
 1. **Use the AI agent to generate job scripts** -- not to run heavy computation directly
 2. **Submit jobs** via `sbatch`, `qsub`, etc. directly from the HPC (or using `!` shell commands if supported by your workflow).
 3. **Monitor jobs** through your HPC's scheduling commands (e.g. `squeue`, `qstat`).
-4. **Retrieve results** with `@@send:path@@` or rclone
+4. **Retrieve results** with `/send <path>` or rclone
 
 ### Example
 
@@ -413,3 +421,27 @@ BSD-3 -- free for academic and personal use. See [LICENSE](LICENSE).
 If this helped your workflow, please star the repo -- it helps others find it.
 
 </div>
+
+
+### Schedule Management
+
+Use `/scheduled` to open an interactive task list. Tap a task to:
+- Delete it
+- Edit schedule presets (hourly/daily/once-after)
+
+Use `/scheduled` for all scheduled task management.
+
+
+### Workspace Separation (multi-chat isolation)
+
+HPC-Relay can isolate sessions/tasks/workdir per chat. Configure with env:
+
+```bash
+CHANNEL_WORKSPACES={"8670800334":{"name":"mg","workdir":"~/workspace_mg","allowed_users":[8670800334]}}
+AUTO_WORKSPACE_PER_CHAT=1
+AUTO_WORKSPACE_PREFIX=chat
+```
+
+- Explicit mapping (`CHANNEL_WORKSPACES`) has highest priority.
+- If unmapped and `AUTO_WORKSPACE_PER_CHAT=1`, bot uses namespace `<prefix>_<chat_id>`.
+- Session/task stores become `hpc_relay_sessions_<name>.json` and `hpc_relay_tasks_<name>.json`.

index.html

@@ -1084,8 +1084,8 @@ <h3>Stateful AI Sessions</h3>
           <div class="feature-icon" style="background:rgba(16,185,129,0.1)">&#x1F4E6;</div>
           <h3>File Transfers</h3>
           <p>Download files from the target machine to Telegram with <code
-              style="font-family:monospace;font-size:12px;color:var(--accent3)">@@send:path@@</code>. Upload from phone
-            with <code style="font-family:monospace;font-size:12px;color:var(--accent3)">@@upload:file@@</code>.
+              style="font-family:monospace;font-size:12px;color:var(--accent3)">/send <path></code>. Upload from phone
+            with <code style="font-family:monospace;font-size:12px;color:var(--accent3)">/upload <path></code>.
             Supports wildcards for batch transfer.</p>
           <span class="tag green">Send + Upload + Wildcards</span>
         </div>
@@ -1116,7 +1116,7 @@ <h3>Chat History Viewer</h3>
     <div class="container">
       <div class="section-label reveal">Control</div>
       <h2 class="section-title reveal">Commands Reference</h2>
-      <p class="section-sub reveal">Everything you can do from Telegram.</p>
+      <p class="section-sub reveal">Everything you can do from Telegram. Supports per-chat workspace/session isolation via env config.</p>
       <div class="cmd-grid">
 
         <div class="cmd-card reveal">
@@ -1133,41 +1133,47 @@ <h2 class="section-title reveal">Commands Reference</h2>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Switch Model</div>
-          <div class="cmd-syntax">@@model: opus46@@</div>
-          <div class="cmd-desc">Change AI model. Use alone (persistent) or inline with a prompt.</div>
+          <div class="cmd-syntax">/model</div>
+          <div class="cmd-desc">Open interactive model picker (provider -> model) and switch persistently.</div>
         </div>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Switch Session</div>
-          <div class="cmd-syntax">@@session: ses_abc123@@</div>
-          <div class="cmd-desc">Jump to a specific session for conversation context.</div>
+          <div class="cmd-syntax">/id ses_abc123</div>
+          <div class="cmd-desc">Switch sessions by ID (or run /id to pick from recent sessions).</div>
         </div>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Kill Process</div>
-          <div class="cmd-syntax">@@kill@@ or !kill</div>
-          <div class="cmd-desc">Terminate the currently running AI process immediately.</div>
+          <div class="cmd-syntax">/kill or /q</div>
+          <div class="cmd-desc">Terminate the currently running AI process immediately (/kill or /q).</div>
         </div>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Download File</div>
-          <div class="cmd-syntax">@@send:~/results/plot.png@@</div>
+          <div class="cmd-syntax">/send ~/results/plot.png</div>
           <div class="cmd-desc">Fetch a file from HPC and receive it in Telegram.</div>
         </div>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Wildcard Fetch</div>
-          <div class="cmd-syntax">@@send:Auto*.png@@</div>
+          <div class="cmd-syntax">/send Auto*.png</div>
           <div class="cmd-desc">Download all matching files. Supports glob patterns.</div>
         </div>
 
         <div class="cmd-card reveal">
           <div class="cmd-label">Upload File</div>
-          <div class="cmd-syntax">@@upload:~/data/input.csv@@</div>
+          <div class="cmd-syntax">/upload ~/data/input.csv</div>
           <div class="cmd-desc">Send a file from your phone to a path on HPC. Use as caption when sending a document.
           </div>
         </div>
 
+        <div class="cmd-card reveal">
+          <div class="cmd-label">Scheduled Tasks</div>
+          <div class="cmd-syntax">/scheduled</div>
+          <div class="cmd-desc">Open interactive scheduled task manager (view/edit/delete).</div>
+        </div>
+
       </div>
     </div>
   </section>
@@ -1328,7 +1334,7 @@ <h4>Example Workflow</h4>
             <li>You: <code style="font-family:monospace;font-size:12px;color:var(--accent3)">!squeue -u $USER</code>
             </li>
             <li>You: <code
-                style="font-family:monospace;font-size:12px;color:var(--accent3)">@@send:~/results/output.pdf@@</code>
+                style="font-family:monospace;font-size:12px;color:var(--accent3)">/send ~/results/output.pdf</code>
             </li>
           </ul>
         </div>