Update app/main.py

app/main.py

@@ -1,3 +1,4 @@
+import json
 import os, json, traceback
 from pathlib import Path
 from fastapi import FastAPI, Request, BackgroundTasks
@@ -115,6 +116,60 @@ async def chat(request: Request):
 
 
 # ── Package installer ──────────────────────────────────────────
+
+# ── Cross-platform: Web SSE push from Telegram ────────────────
+import asyncio as _asyncio
+_sse_queues: dict = {}  # session_id -> asyncio.Queue
+
+@app.get("/api/sse/{session_id}")
+async def sse_endpoint(session_id: str, request: Request):
+    """SSE stream — pushes messages from Telegram to Web UI."""
+    queue = _asyncio.Queue()
+    _sse_queues[session_id] = queue
+
+    from agent_system import orchestrator
+    async def tg_notify(role, content):
+        await queue.put(json.dumps({"type":"tg_message","role":role,"content":content}))
+    orchestrator.register_tg_notify(session_id, tg_notify)
+
+    async def event_stream():
+        try:
+            yield "data: " + json.dumps({"type":"connected","session":session_id}) + "\n\n"
+            while True:
+                if await request.is_disconnected():
+                    break
+                try:
+                    msg = await _asyncio.wait_for(queue.get(), timeout=25)
+                    yield f"data: {msg}\n\n"
+                except _asyncio.TimeoutError:
+                    yield "data: " + json.dumps({"type":"ping"}) + "\n\n"
+        finally:
+            _sse_queues.pop(session_id, None)
+            orchestrator.unregister_tg_notify(session_id)
+
+    return StreamingResponse(event_stream(), media_type="text/event-stream",
+        headers={"X-Accel-Buffering":"no","Cache-Control":"no-cache"})
+
+
+@app.post("/api/send-to-telegram")
+async def send_to_telegram(request: Request):
+    """Send a web UI message to linked Telegram chat."""
+    try:
+        body = await request.json()
+        chat_id  = body.get("chat_id")
+        message  = body.get("message","")
+        api_key  = (body.get("api_key") or "").strip() or cfg.get_longcat_key()
+        model    = body.get("model","LongCat-Flash-Lite")
+        if not chat_id:
+            return JSONResponse({"ok":False,"detail":"No chat_id"})
+        from telegram_bot import handle_update
+        fake_update = {"message":{"chat":{"id":chat_id},"from":{"first_name":"WebUI"},"text":message}}
+        import asyncio as _aio
+        _aio.create_task(handle_update(fake_update, api_key, model))
+        return {"ok":True}
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"ok":False,"detail":str(e)})
+
 @app.post("/api/install-package")
 async def install_package(request: Request):
     try: