Update bot.py

bot.py

@@ -2,6 +2,7 @@ import os
 import io
 import zipfile
 import base64
+import uuid
 import tempfile
 from fastapi import FastAPI, Request
 import aiohttp
@@ -42,8 +43,8 @@ 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
+    unique_name = f"{os.path.splitext(filename)[0]}_{uuid.uuid4().hex}_enc.py"
+    return wrapper.encode(), unique_name
 
 # -------------------------------
 # 🌐 FastAPI Endpoint
@@ -55,7 +56,7 @@ async def encode_endpoint(request: Request):
     file_url = data.get("url")
     filename = data.get("filename", "file.py")
 
-    # ✅ Only allow Python files
+    # Only allow Python files
     if not filename.endswith(".py"):
         return {"ok": False, "error": "Only Python (.py) files are allowed"}
 
@@ -68,21 +69,34 @@ async def encode_endpoint(request: Request):
                 return {"ok": False, "error": f"Failed to fetch file ({resp.status})"}
             file_bytes = await resp.read()
 
+    MAX_FILE_SIZE = 3 * 1024 * 1024  # 3MB limit
+    if len(file_bytes) > MAX_FILE_SIZE:
+        return {"ok": False, "error": "File too large (max 3MB)"}
+
+    # Encode the file
     encoded_bytes, out_name = encode_bytes(file_bytes, filename)
 
-    # ✅ Save to static folder for public download
-    static_dir = os.path.join(os.getcwd(), "static")
-    os.makedirs(static_dir, exist_ok=True)
-    out_path = os.path.join(static_dir, out_name)
-    with open(out_path, "wb") as f:
+    # Save temporarily before upload
+    tmp_path = os.path.join(tempfile.gettempdir(), out_name)
+    with open(tmp_path, "wb") as f:
         f.write(encoded_bytes)
 
-    # Public URL
-    result_url = f"https://jerrycoder-kuttuxd.hf.space/static/{out_name}"
+    # -------------------------------
+    # Upload to AR Hosting
+    # -------------------------------
+    upload_url = "https://ar-hosting.pages.dev/upload"
+    form_data = aiohttp.FormData()
+    form_data.add_field("file", open(tmp_path, "rb"), filename=out_name)
+
+    async with aiohttp.ClientSession() as session:
+        async with session.post(upload_url, data=form_data) as resp:
+            if resp.status != 200:
+                return {"ok": False, "error": f"Failed to upload to AR Hosting ({resp.status})"}
+            result = await resp.json()
 
     return {
         "ok": True,
-        "result_url": result_url,
+        "result_url": result.get("url"),
         "filename": out_name
     }