Update main.py

main.py

@@ -55,25 +55,23 @@ async def get_user_story(link=None):
         # return {"done": "nothing"}
     except Exception as e:
         return {"error": str(e)}
-    file_id = (
-        stories.video.file_id or stories.photo.file_id
-    )
-    caption = stories.caption or f"By {user_client.me.mention}"
-    return {
-        "id": stories.id,
-        "date": stories.date,
-        "expire_date": stories.expire_date,
-        "media": stories.media,
-        "has_protected_content": stories.has_protected_content,
-        "random_file_id": file_id,
-        "caption": caption,
-        "edited": stories.edited,
-        "pinned": stories.pinned,
-        "close_friends": stories.close_friends,
-        "contacts": stories.contacts,
-        "selected_contacts": stories.selected_contacts,
-        "outgoing": stories.outgoing
-    }
+    if stories:
+        for story in stories:
+            file_id = (
+                story.photo.file_id if story and story.photo else None
+                or story.video.file_id if story and story.video else None
+            )
+            caption = story.caption or f"By {user_client.me.mention}"
+            if file_id:
+                return {
+                    "random_file_id": file_id,
+                    "caption": caption
+                }
+            else:
+                return {
+                    "random_file_id": None,
+                    "caption": None
+                }
     
 @fast_app.get("/user/get_user")
 async def get_user(user_id=None):