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": "flux", "name": "ImageGeneration"},
+    {"id": "Flux", "name": "Flux"},  # Changed from "ImageGeneration" to "Flux"
     {"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",
-    "flux": "ImageGeneration",
+    "ImageGeneration": "Flux",  # Changed mapping for "ImageGeneration" to "Flux"
     "llama-3.1-8b": "llama-3.1-8b",
     "llama-3.1-70b": "llama-3.1-70b",
     "llama-3.1-405b": "llama-3.1-405b",
@@ -88,7 +88,7 @@ MODEL_MAPPING = {
 
 # Agent modes
 AGENT_MODE = {
-    'ImageGeneration': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
+    'Flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Flux"},  # Renamed to "Flux"
 }
 
 TRENDING_AGENT_MODE = {