Update utils.py

utils.py

@@ -1,83 +1,29 @@
 import random
 import io
 import zipfile
+import requests
 
 from PIL import Image
-from httpx import AsyncClient
 
 
 jwt_token = ''
 url = "https://api.novelai.net/ai/generate-image"
-global_client = AsyncClient(timeout=100)
+headers = {}
 
 
 def set_token(token):
-    global jwt_token, global_client
+    global jwt_token, headers
     if jwt_token == token:
         return
     jwt_token = token
-    global_client = AsyncClient(
-        timeout=100,
-        headers = {
+    headers = {
             "Authorization": f"Bearer {jwt_token}",
             "Content-Type": "application/json",
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/"
         }
-    )
 
-
-async def remote_login(end_point, password):
-    payload = {"password": password}
-    response = await global_client.post(f'{end_point}/login', params=payload)
-    if response.status_code == 200:
-        return response.json()["status"]
-    else:
-        return None
-
-
-async def remote_gen(
-    end_point="http://127.0.0.1:7000",
-    input_text="",
-    quality_tags="",
-    negative_prompt="", 
-    seed=-1, 
-    scale=5.0, 
-    width=1024, 
-    height=1024, 
-    steps=28, 
-    sampler="k_euler",
-    schedule='native',
-    smea=False,
-    dyn=False,
-    dyn_threshold=False,
-    chg_rescale=0,
-):
-    payload = {
-        "prompt": f'{input_text}, {quality_tags}',
-        "neg_prompt": negative_prompt,
-        "seed": seed,
-        "scale": scale,
-        "width": width,
-        "height": height,
-        "steps": steps,
-        "sampler": sampler,
-        "schedule": schedule,
-        "smea": smea,
-        "dyn": dyn,
-        "dyn_threshold": dyn_threshold,
-        "chg_rescale": chg_rescale,
-    }
-    response = await global_client.post(f'{end_point}/gen', json=payload)
-    if response.status_code == 200:
-        mem_file = io.BytesIO(response.content)
-        mem_file.seek(0)
-        return Image.open(mem_file)
-    else:
-        return None
-
-
-async def generate_novelai_image(
+def generate_novelai_image(
     input_text="", 
     negative_prompt="", 
     seed=-1, 
@@ -125,7 +71,7 @@ async def generate_novelai_image(
     }
 
     # Send the POST request
-    response = await global_client.post(url, json=payload)
+    response = requests.post(url, json=payload, headers=headers)
 
     # Process the response
     if response.headers.get('Content-Type') == 'application/x-zip-compressed':
@@ -141,8 +87,6 @@ async def generate_novelai_image(
         return "Generation failed", response
 
 
-def free_check(width, height, steps):
-    return width * height <= 1024 * 1024 and steps<=28
 
 
 def image_from_bytes(data):