XA

API/app.py

@@ -27,6 +27,8 @@ env_path = os.path.dirname(os.path.realpath(__file__)) + "/../.env"
 if os.path.exists(env_path):
     load_dotenv(env_path)
 PHPSESSID = os.getenv("PHPSESSID")
+
+print(PHPSESSID)
 cookies = {"PHPSESSID": PHPSESSID}
 
 headers = {
@@ -64,6 +66,7 @@ async def search(raw, pages, ai_only=True, cookies=None, headers=None):
         responses = await asyncio.gather(*tasks)
         for data in responses:
             if ai_only:
+                print(data['body']['illustManga']['data'])
                 posts = [post for post in data['body']['illustManga']['data'] if post['aiType'] == 2]
             else:
                 posts = data['body']['illustManga']['data']

Client/Extract Pixiv/ai_search.py

@@ -1,6 +1,11 @@
 import requests
 import os
 
+local = False
+if local:
+    endpoint = "http://127.0.0.1:7860"
+else:
+    endpoint = "https://q6-p.hf.space/"
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
 input_url = input("Enter the URL: ")
@@ -13,9 +18,8 @@ params = {
     'ai_only': True,
 }
 
-response = requests.get('https://q6-p.hf.space/search', params=params)
+response = requests.get(f'{endpoint}/search', params=params)
 
 data = response.json()
-
 with open(f"../{data['filename']}.txt", "w") as f:
     f.write("\n".join(data['post_ids']))