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")
+api_anime = os.getenv("ANIME_API")
 rmbgkey = os.getenv("RMBGKEY")
 
 #fore, prompt, intensity, mode, refprompt, isrmbg
@@ -215,7 +216,7 @@ def resize_to_fit(max_size, original_size):
     return new_width, new_height
 
     
-def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg): 
+def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type): 
 
     max_size = (1500,1500)
     if prompt in prompt_list:
@@ -255,6 +256,14 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
         "height" : 1500,
         "refprompt" : refprompt
     }
+
+    # Select the endpoint based on the model type
+    if model_type == "anime":
+        api_endpoint = api_anime
+    elif model_type == "realistic":
+        api_endpoint = api
+    else:
+        raise ValueError("Invalid model type selected.")
     #print(f"DATA: {data}")
     
     '''
@@ -268,7 +277,7 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg):
         "refprompt" : " transparent glass "    
     }
     '''    
-    image = process(data, api, api_key)
+    image = process(data, api_endpoint, api_key)
 
     return image