Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import datetime
 
-import toml
 import gradio as gr
 from pnginfo import read_info_from_image, send_paras
 from images_history import img_history_ui
@@ -9,8 +8,6 @@ from images_history import img_history_ui
 from utils import set_token, generate_novelai_image, image_from_bytes
 from PIL import PngImagePlugin
 
-
-client_config = toml.load("config.toml")['client']
 today_count = 0
 today = datetime.date.today().strftime('%Y-%m-%d')
 
@@ -82,7 +79,7 @@ def control_ui():
 
 def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps, sampler, scheduler, smea, dyn, dyn_threshold, cfg_rescale, save):
     global today_count
-    set_token(client_config['token'])
+    set_token(os.environ.get('token'))
     img_data, payload = generate_novelai_image(
         f"{prompt}, {quality_tags}", neg_prompt, seed, scale, 
         width, height, steps, sampler, scheduler, 
@@ -93,17 +90,6 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
     today_count += 1
     img = image_from_bytes(img_data)
     
-    if save:
-        save_path = client_config['save_path']
-        today = datetime.date.today().strftime('%Y-%m-%d')
-        today_path = os.path.join(save_path, today)
-        if not os.path.exists(today_path):
-            os.makedirs(today_path, mode=777, exist_ok=True)
-        filename = str(today_count).rjust(5, '0') + '-' + str(payload['parameters']['seed']) + '.png'
-        pnginfo_data = PngImagePlugin.PngInfo()
-        for k, v in img.info.items():
-            pnginfo_data.add_text(k, str(v))
-        img.save(os.path.join(today_path, filename), pnginfo=pnginfo_data)
     
     return img, payload