Update bot.py

bot.py

@@ -4,7 +4,7 @@ import zipfile
 import base64
 import uuid
 import tempfile
-from fastapi import FastAPI, Request, Query
+from fastapi import FastAPI, Request, UploadFile, File
 import aiohttp
 import uvicorn
 
@@ -49,72 +49,72 @@ runpy.run_path(os.path.join(t,p))
 MAX_FILE_SIZE = 3 * 1024 * 1024  # 3 MB limit
 
 # -------------------------------
-# POST /api/encode
+# POST /api/encode - receive raw file bytes
 # -------------------------------
 
 @app.post("/api/encode")
-async def encode_post(request: Request):
+async def encode_post(file: UploadFile = File(...)):
+    filename = file.filename
+    if not filename.endswith(".py"):
+        return {"ok": False, "error": "Only Python (.py) files are allowed"}
+
     try:
-        data = await request.json()
-        file_url = data.get("url")
-        filename = data.get("filename", "file.py")
-        return await process_file(file_url, filename)
-    except Exception as e:
-        return {"ok": False, "error": f"Internal server error: {str(e)}"}
+        file_bytes = await file.read()
+        if len(file_bytes) > MAX_FILE_SIZE:
+            return {"ok": False, "error": "File too large (max 3MB)"}
 
-# -------------------------------
-# GET /encode?url=...&filename=...
-# -------------------------------
+        encoded_bytes, out_name = encode_bytes(file_bytes, filename)
+
+        tmp_path = os.path.join(tempfile.gettempdir(), out_name)
+        with open(tmp_path, "wb") as f:
+            f.write(encoded_bytes)
+
+        # Upload to AR Hosting
+        upload_url = "https://ar-hosting.pages.dev/upload"
+        form_data = aiohttp.FormData()
+        f = open(tmp_path, "rb")
+        form_data.add_field("file", f, filename=out_name, content_type="application/octet-stream")
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(upload_url, data=form_data) as resp:
+                try:
+                    result = await resp.json()
+                except Exception:
+                    f.close()
+                    return {"ok": False, "error": f"AR Hosting returned invalid JSON (status {resp.status})"}
+
+                if resp.status != 200 or "url" not in result:
+                    f.close()
+                    return {"ok": False, "error": f"AR Hosting upload failed: {result}"}
+        f.close()
+        return {"ok": True, "result_url": result.get("url"), "filename": out_name}
 
-@app.get("/encode")
-async def encode_get(url: str = Query(...), filename: str = Query("file.py")):
-    try:
-        return await process_file(url, filename)
     except Exception as e:
         return {"ok": False, "error": f"Internal server error: {str(e)}"}
 
 # -------------------------------
-# File processing function
+# GET /encode?url=...&filename=... (optional)
 # -------------------------------
 
-async def process_file(file_url: str, filename: str):
-    if not file_url:
-        return {"ok": False, "error": "No file URL provided"}
-    if not filename.endswith(".py"):
-        return {"ok": False, "error": "Only Python (.py) files are allowed"}
-
-    # -------------------------------
-    # Fetch the file
-    # -------------------------------
+@app.get("/encode")
+async def encode_get(url: str, filename: str = "file.py"):
+    # Keep GET for backward compatibility if URL is public
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.get(file_url) as resp:
+            async with session.get(url) as resp:
                 if resp.status != 200:
                     return {"ok": False, "error": f"Failed to fetch file ({resp.status})"}
                 file_bytes = await resp.read()
-    except Exception as e:
-        return {"ok": False, "error": f"Failed to fetch file: {str(e)}"}
-
-    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)
+        encoded_bytes, out_name = encode_bytes(file_bytes, filename)
 
-    # Save temporarily
-    tmp_path = os.path.join(tempfile.gettempdir(), out_name)
-    with open(tmp_path, "wb") as f:
-        f.write(encoded_bytes)
+        tmp_path = os.path.join(tempfile.gettempdir(), out_name)
+        with open(tmp_path, "wb") as f:
+            f.write(encoded_bytes)
 
-    # -------------------------------
-    # Upload to AR Hosting
-    # -------------------------------
-    try:
+        # Upload to AR Hosting
         upload_url = "https://ar-hosting.pages.dev/upload"
         form_data = aiohttp.FormData()
-        f = open(tmp_path, "rb")  # Keep file open for async upload
+        f = open(tmp_path, "rb")
         form_data.add_field("file", f, filename=out_name, content_type="application/octet-stream")
 
         async with aiohttp.ClientSession() as session:
@@ -124,16 +124,14 @@ async def process_file(file_url: str, filename: str):
                 except Exception:
                     f.close()
                     return {"ok": False, "error": f"AR Hosting returned invalid JSON (status {resp.status})"}
-
                 if resp.status != 200 or "url" not in result:
                     f.close()
                     return {"ok": False, "error": f"AR Hosting upload failed: {result}"}
+        f.close()
+        return {"ok": True, "result_url": result.get("url"), "filename": out_name}
 
-        f.close()  # Close after upload
     except Exception as e:
-        return {"ok": False, "error": f"AR Hosting request failed: {str(e)}"}
-
-    return {"ok": True, "result_url": result.get("url"), "filename": out_name}
+        return {"ok": False, "error": f"Internal server error: {str(e)}"}
 
 # -------------------------------
 # 🏁 Run App