Update bot.py

bot.py

@@ -1,31 +1,97 @@
-# backend.py
 import os
-import shutil
-import tempfile
+import io
+import zipfile
 import base64
-from pathlib import Path
-from fastapi import FastAPI, File, Form, UploadFile
-from fastapi.responses import JSONResponse
-from encoder import encode_bytes
-  # helper encoding logic
+import random
+import tempfile
+from fastapi import FastAPI, Request
+import aiohttp
+import uvicorn
 
 app = FastAPI()
 
+# -------------------------------
+# 🔐 Simple Python File Encoder
+# -------------------------------
+
+def make_zip_bytes(filename: str, content: bytes) -> bytes:
+    """Compress file bytes into a ZIP archive (in memory)."""
+    bio = io.BytesIO()
+    with zipfile.ZipFile(bio, "w", compression=zipfile.ZIP_DEFLATED) as zf:
+        zf.writestr(filename, content)
+    return bio.getvalue()
+
+def rand_bytes(n): 
+    """Generate random bytes."""
+    return os.urandom(n)
+
+def xor_bytes(data: bytes, key: bytes) -> bytes:
+    """Simple XOR encryption."""
+    return bytes([b ^ key[i % len(key)] for i, b in enumerate(data)])
+
+def encode_bytes(file_bytes: bytes, filename: str):
+    """Encrypt + wrap into Python file."""
+    # Step 1: Zip the file
+    zipped = make_zip_bytes(filename, file_bytes)
+    
+    # Step 2: Encrypt with XOR
+    key = rand_bytes(16)
+    enc = xor_bytes(zipped, key)
+    enc_b64 = base64.b64encode(enc).decode()
+    key_b64 = base64.b64encode(key).decode()
+    
+    # Step 3: Build output Python file
+    wrapper = f"""# Auto Encrypted by JerryCoder Encryptor
+import base64,zlib
+D=base64.b64decode('{enc_b64}')
+K=base64.b64decode('{key_b64}')
+O=bytes([b^K[i%len(K)] for i,b in enumerate(D)])
+import io,zipfile,tempfile,os,runpy
+t=tempfile.mkdtemp()
+zipfile.ZipFile(io.BytesIO(O)).extractall(t)
+p=[f for f in os.listdir(t) if f.endswith('.py')][0]
+runpy.run_path(os.path.join(t,p))
+"""
+    output_name = os.path.splitext(filename)[0] + "_enc.py"
+    return wrapper.encode(), output_name
+
+# -------------------------------
+# 🌐 FastAPI Endpoint
+# -------------------------------
+
 @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 index():
-    return {"ok": True, "status": "HF backend alive"}
+async def encode_endpoint(request: Request):
+    """Receive file URL from Cloudflare bot, encode, return result."""
+    data = await request.json()
+    file_url = data.get("url")
+    filename = data.get("filename", "file.py")
+
+    if not file_url:
+        return {"ok": False, "error": "No file URL provided"}
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(file_url) as resp:
+            if resp.status != 200:
+                return {"ok": False, "error": f"Failed to fetch file ({resp.status})"}
+            file_bytes = await resp.read()
+
+    encoded_bytes, out_name = encode_bytes(file_bytes, filename)
+
+    out_path = os.path.join(tempfile.gettempdir(), out_name)
+    with open(out_path, "wb") as f:
+        f.write(encoded_bytes)
+
+    # ⚠️ NOTE: Hugging Face doesn’t auto-host tmp files publicly.
+    # You’ll need to upload to a hosting service (like tmpfiles.org, catbox.moe, etc.)
+    # For now, just return ok=True and filename.
+    return {
+        "ok": True,
+        "result_url": f"https://jerrycoder-kuttuxd.hf.space/tmp/{out_name}",
+        "filename": out_name
+    }
+
+# -------------------------------
+# 🏁 Run App
+# -------------------------------
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))