Update app.py

app.py

@@ -6,14 +6,11 @@ import io
 import random
 
 url = "http://73.255.78.150:7909/sdapi/v1/img2img"
-session_seed, last_session_seed = None, None
-
-def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
-    global session_seed, last_session_seed
 
+def set_seed_values(reset_seed, increment_seed, session_seed, last_session_seed):
     if reset_seed:
         last_session_seed, session_seed = session_seed, -1 if session_seed != -1 else None
-    elif not reset_seed and last_session_seed is not None:
+    elif not reset_seed and last_session_seed:
         session_seed = last_session_seed
     elif session_seed is None:
         session_seed = random.randint(0, 10000)
@@ -21,6 +18,13 @@ def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
     if increment_seed and session_seed != -1:
         session_seed += 1
 
+    return session_seed, last_session_seed
+
+def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
+    global session_seed, last_session_seed
+    
+    session_seed, last_session_seed = set_seed_values(reset_seed, increment_seed, session_seed, last_session_seed)
+
     init_image_value = ""
     if image_file:
         image_data = image_file.read()
@@ -38,13 +42,10 @@ def generate_image(prompt, steps, reset_seed, increment_seed, image_file=None):
     response_data = response.json()
 
     try:
-        image = Image.open(io.BytesIO(base64.b64decode(response_data['images'][0])))
-        return image
+        return Image.open(io.BytesIO(base64.b64decode(response_data['images'][0])))
     except KeyError:
-        error_message = response_data.get('error', 'Unknown error')
         error_image = Image.new('RGB', (512, 512), color=(73, 109, 137))
-        d = ImageDraw.Draw(error_image)
-        d.text((10, 10), f"Error: {error_message}", fill=(255, 255, 0))
+        ImageDraw.Draw(error_image).text((10, 10), f"Error: {response_data.get('error', 'Unknown error')}", fill=(255, 255, 0))
         return error_image
 
 def main_interface():