Update bot.py

bot.py

@@ -4,7 +4,7 @@ import zipfile
 import base64
 import uuid
 import tempfile
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Query
 import aiohttp
 import uvicorn
 
@@ -46,22 +46,44 @@ runpy.run_path(os.path.join(t,p))
     unique_name = f"{os.path.splitext(filename)[0]}_{uuid.uuid4().hex}_enc.py"
     return wrapper.encode(), unique_name
 
+MAX_FILE_SIZE = 3 * 1024 * 1024  # 3MB limit
+
 # -------------------------------
-# 🌐 FastAPI Endpoint
+# POST /api/encode
 # -------------------------------
 
 @app.post("/api/encode")
 async def encode_endpoint(request: Request):
-    data = await request.json()
-    file_url = data.get("url")
-    filename = data.get("filename", "file.py")
+    try:
+        data = await request.json()
+        file_url = data.get("url")
+        filename = data.get("filename", "file.py")
 
-    # Only allow Python files
-    if not filename.endswith(".py"):
-        return {"ok": False, "error": "Only Python (.py) files are allowed"}
+        return await process_file(file_url, filename)
+
+    except Exception as e:
+        return {"ok": False, "error": f"Internal server error: {str(e)}"}
+
+# -------------------------------
+# GET /encode?url=...&filename=...
+# -------------------------------
 
+@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
+# -------------------------------
+
+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"}
 
     async with aiohttp.ClientSession() as session:
         async with session.get(file_url) as resp:
@@ -69,21 +91,17 @@ 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 temporarily before upload
+    # Save temporarily
     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()
     form_data.add_field("file", open(tmp_path, "rb"), filename=out_name)
@@ -94,11 +112,7 @@ async def encode_endpoint(request: Request):
                 return {"ok": False, "error": f"Failed to upload to AR Hosting ({resp.status})"}
             result = await resp.json()
 
-    return {
-        "ok": True,
-        "result_url": result.get("url"),
-        "filename": out_name
-    }
+    return {"ok": True, "result_url": result.get("url"), "filename": out_name}
 
 # -------------------------------
 # 🏁 Run App