Update main.py

main.py

@@ -7,7 +7,7 @@ import httpx
 
 from pikpakapi import PikPakApi
 
-from typing import Union
+from typing import Union, Any, Dict, List, Optional
 from fastapi import (
     FastAPI,
     Depends,
@@ -121,8 +121,22 @@ async def home(request: Request):
 
 
 @app.get("/files")
-async def get_files(kw: str = ""):
-    return await THUNDERX_CLIENT.file_list()
+async def get_files(
+    size: int = 100,
+    parent_id: Optional[str] = None,
+    next_page_token: Optional[str] = None,
+    additional_filters: Optional[Dict[str, Any]] = None,
+):
+    return await THUNDERX_CLIENT.file_list(
+        size, parent_id, next_page_token, additional_filters
+    )
+
+
+@app.get("/offline")
+async def offline(
+    file_url: str, parent_id: Optional[str] = None, name: Optional[str] = None
+):
+    return await THUNDERX_CLIENT.offline_download(file_url, parent_id, name)
 
 
 @app.get("/userinfo")