Update api/config.py

api/config.py

@@ -24,7 +24,7 @@ APP_SECRET = os.getenv("APP_SECRET")
 ALLOWED_MODELS = [
     {"id": "blackboxai", "name": "blackboxai"},
     {"id": "blackboxai-pro", "name": "blackboxai-pro"},
-    {"id": "ImageGeneration", "name": "ImageGeneration"},
+    {"id": "flux", "name": "ImageGeneration"},
     {"id": "llama-3.1-8b", "name": "llama-3.1-8b"},
     {"id": "llama-3.1-70b", "name": "llama-3.1-70b"},
     {"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
@@ -55,7 +55,7 @@ ALLOWED_MODELS = [
 MODEL_MAPPING = {
     "blackboxai": "blackboxai",
     "blackboxai-pro": "blackboxai-pro",
-    "ImageGeneration": "ImageGeneration",
+    "flux": "ImageGeneration",
     "llama-3.1-8b": "llama-3.1-8b",
     "llama-3.1-70b": "llama-3.1-70b",
     "llama-3.1-405b": "llama-3.1-405b",
@@ -80,7 +80,6 @@ MODEL_MAPPING = {
     # Additional mappings
     "gemini-flash": "gemini-1.5-flash",
     "claude-3.5-sonnet": "claude-sonnet-3.5",
-    "flux": "ImageGeneration",
     "gemini-1.5-pro-latest": "gemini-pro",
     "gemini-1.5-pro": "gemini-1.5-pro",
     "claude-3-5-sonnet-20240620": "claude-sonnet-3.5",