Update main.py

main.py

@@ -1,4 +1,5 @@
 import logging
+import base64
 from contextlib import asynccontextmanager
 from akenoai import AkenoXToJs as js
 from akenoai.runner import run_fast
@@ -47,12 +48,10 @@ async def hello():
     return {"success": "hello world!"}
 
 @fast_app.get("/user/get_story")
-async def get_user_story(link=None):
+async def get_user_story(link: str = None):
     username, random_id = get_random_from_channel(link)
     try:
         stories = await user_client.get_stories(username, story_ids=[random_id])
-        # LOGS.info(f"Updates: {stories}")
-        # return {"done": "nothing"}
     except Exception as e:
         return {"error": str(e)}
     if stories:
@@ -63,15 +62,20 @@ async def get_user_story(link=None):
             )
             caption = story.caption or f"By {user_client.me.mention}"
             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")
                 return {
+                    "download": base64_image,
                     "random_file_id": file_id,
                     "caption": caption
                 }
-            else:
-                return {
-                    "random_file_id": None,
-                    "caption": None
-                }
+    return {
+        "random_file_id": None,
+        "caption": None
+    }
+
     
 @fast_app.get("/user/get_user")
 async def get_user(user_id=None):