Update app.py

app.py

@@ -1,10 +1,6 @@
-from fastapi import FastAPI, Request, UploadFile, File
-from fastapi.responses import StreamingResponse
+from fastapi import FastAPI, Request, Form
 import threading
-import os
-import zipfile
 import time
-from io import BytesIO
 
 app = FastAPI()
 lock = threading.Lock()
@@ -13,101 +9,50 @@ lock = threading.Lock()
 channels = {}
 update_ids = {}
 
-file_channels = {}
 
-UPLOAD_DIR = "uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
-
-
-# -------- SEND TEXT --------
+# -------- SEND MESSAGE --------
 @app.post("/sendMessage/{channel}")
-async def send_message(channel: str, req: Request):
-    data = await req.body()
-    text = data.decode("utf-8", errors="ignore").strip()
-
-    with lock:
-        if channel not in channels:
-            channels[channel] = []
-            update_ids[channel] = 0
-
-        update_ids[channel] += 1
-
-        channels[channel].append({
-            "update_id": update_ids[channel],
-            "text": text
-        })
-
-    return {"ok": True}
-
-
-# -------- SEND FILE --------
-@app.post("/sendFile/{channel}")
-async def send_file(channel: str, files: list[UploadFile] = File(...)):
-    saved_files = []
+async def send_message(channel: str, text: str = Form(None), req: Request = None):
+    try:
+        # support both form-data & raw body
+        if not text:
+            data = await req.body()
+            text = data.decode("utf-8", errors="ignore").strip()
 
-    with lock:
-        if channel not in file_channels:
-            file_channels[channel] = []
+        if not text:
+            return {"ok": False, "error": "Empty message"}
 
-    for file in files:
-        filename = f"{int(time.time()*1000)}_{file.filename}"
-        filepath = os.path.join(UPLOAD_DIR, filename)
+        with lock:
+            if channel not in channels:
+                channels[channel] = []
+                update_ids[channel] = 0
 
-        try:
-            content = await file.read()
-            with open(filepath, "wb") as f:
-                f.write(content)
+            update_ids[channel] += 1
 
-            with lock:
-                file_channels[channel].append(filepath)
+            channels[channel].append({
+                "update_id": update_ids[channel],
+                "text": text,
+                "time": int(time.time())
+            })
 
-            saved_files.append(filename)
+        return {"ok": True}
 
-        except Exception as e:
-            saved_files.append(f"error:{file.filename}")
+    except Exception as e:
+        return {"ok": False, "error": str(e)}
 
-    return {"ok": True, "files": saved_files}
 
-
-# -------- GET UPDATES (TEXT + FILE) --------
+# -------- GET UPDATES (consume-on-read) --------
 @app.get("/getUpdates/{channel}")
 def get_updates(channel: str):
+    try:
+        with lock:
+            msgs = channels.get(channel, []).copy()
+            channels[channel] = []
 
-    with lock:
-        msgs = channels.get(channel, []).copy()
-        channels[channel] = []
-
-        files = file_channels.get(channel, []).copy()
-        file_channels[channel] = []
-
-    # -------- CASE 1: FILE EXISTS --------
-    if files:
-        mem_zip = BytesIO()
-
-        with zipfile.ZipFile(mem_zip, "w", zipfile.ZIP_DEFLATED) as zipf:
-
-            # add files
-            for file_path in files:
-                if os.path.exists(file_path):
-                    zipf.write(file_path, os.path.basename(file_path))
-
-            # add text হিসেবে messages.txt
-            if msgs:
-                text_content = "\n".join([m.get("text", "") for m in msgs])
-                zipf.writestr("messages.txt", text_content)
-
-        mem_zip.seek(0)
-
-        return StreamingResponse(
-            mem_zip,
-            media_type="application/zip",
-            headers={
-                "Content-Disposition": f"attachment; filename={channel}_data.zip"
-            }
-        )
+        return {
+            "ok": True,
+            "result": msgs if msgs else []
+        }
 
-    # -------- CASE 2: ONLY TEXT --------
-    return {
-        "ok": True,
-        "result": msgs if msgs else []
-    }
+    except Exception as e:
+        return {"ok": False, "error": str(e)}