Update bot.py

bot.py

@@ -1,48 +1,30 @@
-import os, io, tempfile, random, shutil, base64, subprocess, sys, zipfile
+# backend.py
+import os
+import shutil
+import tempfile
+import base64
 from pathlib import Path
-from fastapi import FastAPI, Request
-from telegram import Update
-from telegram.ext import Application, CommandHandler, MessageHandler, filters
-from motor.motor_asyncio import AsyncIOMotorClient
-import uvicorn
-from config import BOT_TOKEN, ADMIN_ID, MONGO_URI, DB_NAME
+from fastapi import FastAPI, File, Form, UploadFile
+from fastapi.responses import JSONResponse
+from encoder import encode_bytes  # helper encoding logic
 
-# --- MongoDB ---
-mongo_client = AsyncIOMotorClient(MONGO_URI)
-db = mongo_client[DB_NAME]
-stats_col = db["stats"]
-
-async def init_stats():
-    if not await stats_col.find_one({"_id":"encryption_count"}):
-        await stats_col.insert_one({"_id":"encryption_count","count":0})
-
-# --- Bot setup ---
 app = FastAPI()
-bot_app = Application.builder().token(BOT_TOKEN).build()
-
-# --- Handlers ---
-async def start(update, context):
-    await update.message.reply_text("🤖 Hello! Bot is running via Cloudflare Worker + FastAPI")
 
-async def echo_doc(update, context):
-    await update.message.reply_text("📄 Got your document!")
-
-bot_app.add_handler(CommandHandler("start", start))
-bot_app.add_handler(MessageHandler(filters.Document.ALL, echo_doc))
-
-# --- /poll route (used by Worker) ---
-@app.post("/poll")
-async def poll_endpoint(request: Request):
-    data = await request.json()
-    update = Update.de_json(data, bot_app.bot)
-    await bot_app.update_queue.put(update)
-    return {"ok": True}
+@app.post("/api/encode")
+async def api_encode(file: UploadFile = File(...), filename: str = Form(...), chat_id: str = Form(None), from_user: str = Form(None)):
+    """
+    Receives file bytes, runs encoding logic, returns JSON:
+    { ok: true, filename: "...", file_b64: "..." }
+    """
+    try:
+        data = await file.read()
+        # encode_bytes returns bytes for result file
+        result_bytes, out_name = encode_bytes(data, filename)
+        file_b64 = base64.b64encode(result_bytes).decode()
+        return JSONResponse({"ok": True, "filename": out_name, "file_b64": file_b64})
+    except Exception as e:
+        return JSONResponse({"ok": False, "error": str(e)}, status_code=500)
 
 @app.get("/")
-async def home():
-    return {"ok": True, "message": "Backend online!"}
-
-if __name__ == "__main__":
-    import asyncio
-    asyncio.run(init_stats())
-    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+async def index():
+    return {"ok": True, "status": "HF backend alive"}