Update main.py

main.py

@@ -2,6 +2,8 @@ import logging
 import base64
 import requests
 import uvloop
+import aiofiles
+import aiohttp
 from bs4 import BeautifulSoup
 from contextlib import asynccontextmanager
 from akenoai import AkenoXToJs as js
@@ -51,6 +53,26 @@ def get_random_from_channel(link):
     )
     return username, random_id
 
+async def upload_to_catbox(dl_path: str) -> str:
+    base_url = "https://catbox.moe/user/api.php"
+
+    async with aiohttp.ClientSession() as session:
+        form_data = aiohttp.FormData()
+        form_data.add_field("reqtype", "fileupload")
+
+        async with aiofiles.open(dl_path, mode="rb") as file:
+            file_data = await file.read()
+            form_data.add_field(
+                "fileToUpload",
+                file_data,
+                filename=dl_path.split("/")[-1],
+                content_type="application/octet-stream"
+            )
+
+        async with session.post(base_url, data=form_data) as response:
+            response.raise_for_status()
+            return (await response.text()).strip()
+
 @fast_app.on_event("startup")
 async def startup_event():
     user = await client.start()
@@ -100,10 +122,11 @@ async def get_user_story(link: str = None):
             if file_id:
                 image_path = await user_client.download_media(file_id)
                 
-                with open(image_path, "rb") as f:
-                    base64_image = base64.b64encode(f.read()).decode("utf-8")
+                # with open(image_path, "rb") as f:
+                #    base64_image = base64.b64encode(f.read()).decode("utf-8")
+                url_dl = await upload_to_catbox(image_path)
                 return {
-                    "download": base64_image,
+                    "download": url_dl,
                     "random_file_id": file_id,
                     "caption": caption
                 }