Update app.py

app.py

@@ -15,6 +15,7 @@ from enum import Enum
 
 api_key = os.getenv("FAI_API_KEY")
 api = os.getenv("FAI_API_5")
+api_old = os.getenv("FAI_API_old")
 api_anime = os.getenv("FAI_anime") #api_anime = os.getenv("ANIME_API")
 rmbgkey = os.getenv("RMBGKEY")
 
@@ -299,8 +300,20 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
     # 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}")
@@ -316,7 +329,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
         "refprompt" : " transparent glass "    
     }
     '''    
-    image = process(data, api_endpoint, api_key)
+    
 
     return image