Update app.py

app.py

@@ -266,7 +266,17 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
         except:
             pass
          
-    #fore = create_square_image(fore)
+    #
+    # Select the endpoint based on the model type
+    if model_type == "anime":
+        api_endpoint = api_anime #to change to api_anime
+        
+    elif model_type == "realistic":
+        fore = create_square_image(fore)
+        api_endpoint = api_old
+        
+    else:
+        raise ValueError("Invalid model type selected.")
     size = fore.size
     image_width = size[0]
     image_height = size[1]
@@ -297,25 +307,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
         "second_stage_steps" : 20
     }
 
-    # Select the endpoint based on the model type
-    if model_type == "anime":
-        api_endpoint = api_anime #to change to api_anime
-        image = process(data, api_endpoint, api_key)
-    elif model_type == "realistic":
-        api_endpoint = api
-        if data['mode']=="full":
-            data['mode']="first-stage"
-            image = process(data, api_endpoint, api_key)
-            forestr = image_to_base64(image.convert("RGBA"))
-            data['foreground_image64']=forestr
-            data['mode']="refiner"
-            api_endpoint = api_old
-            image = process(data, api_endpoint, api_key)
-        elif data['mode']=="refiner":
-            api_endpoint = api_old
-            image = process(data, api_endpoint, api_key)
-    else:
-        raise ValueError("Invalid model type selected.")
+
     #print(f"DATA: {data}")
     
     '''
@@ -329,7 +321,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
         "refprompt" : " transparent glass "    
     }
     '''    
-    
+    image = process(data, api_endpoint, api_key)
 
     return image