Update main.py

main.py

@@ -8,8 +8,7 @@ import logging
 import asyncio
 import time
 from collections import defaultdict
-from typing import List, Dict, Any, Optional, Union
-from datetime import datetime
+from typing import List, Dict, Any, Optional
 
 from aiohttp import ClientSession, ClientResponseError
 from fastapi import FastAPI, HTTPException, Request, Depends, Header
@@ -45,13 +44,16 @@ class Blackbox:
     api_endpoint = "https://www.blackbox.ai/api/chat"
     working = True
     supports_gpt_4 = True
+    supports_stream = True
     supports_system_message = True
     supports_message_history = True
 
     default_model = 'blackboxai'
+    image_models = ['ImageGeneration']
     models = [
         default_model,
         'blackboxai-pro',
+        *image_models,
         "llama-3.1-8b",
         'llama-3.1-70b',
         'llama-3.1-405b',
@@ -59,9 +61,25 @@ class Blackbox:
         'gemini-pro',
         'gemini-1.5-flash',
         'claude-sonnet-3.5',
+        'PythonAgent',
+        'JavaAgent',
+        'JavaScriptAgent',
+        'HTMLAgent',
+        'GoogleCloudAgent',
+        'AndroidDeveloper',
+        'SwiftDeveloper',
+        'Next.jsAgent',
+        'MongoDBAgent',
+        'PyTorchAgent',
+        'ReactAgent',
+        'XcodeAgent',
+        'AngularJSAgent',
     ]
 
-    agentMode = {}
+    agentMode = {
+        'ImageGeneration': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
+    }
+
     trendingAgentMode = {
         "blackboxai": {},
         "gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
@@ -69,6 +87,19 @@ class Blackbox:
         'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
         'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405b"},
         'blackboxai-pro': {'mode': True, 'id': "BLACKBOXAI-PRO"},
+        'PythonAgent': {'mode': True, 'id': "Python Agent"},
+        'JavaAgent': {'mode': True, 'id': "Java Agent"},
+        'JavaScriptAgent': {'mode': True, 'id': "JavaScript Agent"},
+        'HTMLAgent': {'mode': True, 'id': "HTML Agent"},
+        'GoogleCloudAgent': {'mode': True, 'id': "Google Cloud Agent"},
+        'AndroidDeveloper': {'mode': True, 'id': "Android Developer"},
+        'SwiftDeveloper': {'mode': True, 'id': "Swift Developer"},
+        'Next.jsAgent': {'mode': True, 'id': "Next.js Agent"},
+        'MongoDBAgent': {'mode': True, 'id': "MongoDB Agent"},
+        'PyTorchAgent': {'mode': True, 'id': "PyTorch Agent"},
+        'ReactAgent': {'mode': True, 'id': "React Agent"},
+        'XcodeAgent': {'mode': True, 'id': "Xcode Agent"},
+        'AngularJSAgent': {'mode': True, 'id': "AngularJS Agent"},
     }
 
     userSelectedModel = {
@@ -81,19 +112,48 @@ class Blackbox:
         'gpt-4o': '@GPT-4o',
         'gemini-pro': '@Gemini-PRO',
         'claude-sonnet-3.5': '@Claude-Sonnet-3.5',
+        'PythonAgent': '@Python Agent',
+        'JavaAgent': '@Java Agent',
+        'JavaScriptAgent': '@JavaScript Agent',
+        'HTMLAgent': '@HTML Agent',
+        'GoogleCloudAgent': '@Google Cloud Agent',
+        'AndroidDeveloper': '@Android Developer',
+        'SwiftDeveloper': '@Swift Developer',
+        'Next.jsAgent': '@Next.js Agent',
+        'MongoDBAgent': '@MongoDB Agent',
+        'PyTorchAgent': '@PyTorch Agent',
+        'ReactAgent': '@React Agent',
+        'XcodeAgent': '@Xcode Agent',
+        'AngularJSAgent': '@AngularJS Agent',
         'blackboxai-pro': '@BLACKBOXAI-PRO',
+        'ImageGeneration': '@Image Generation',
     }
 
     model_referers = {
         "blackboxai": "/?model=blackboxai",
         "gpt-4o": "/?model=gpt-4o",
         "gemini-pro": "/?model=gemini-pro",
-        "claude-sonnet-3.5": "/?model=claude-sonnet-3.5"
+        "claude-sonnet-3.5": "/?model=claude-sonnet-3.5",
+        "PythonAgent": "/?model=PythonAgent",
+        "JavaAgent": "/?model=JavaAgent",
+        "JavaScriptAgent": "/?model=JavaScriptAgent",
+        "HTMLAgent": "/?model=HTMLAgent",
+        "GoogleCloudAgent": "/?model=GoogleCloudAgent",
+        "AndroidDeveloper": "/?model=AndroidDeveloper",
+        "SwiftDeveloper": "/?model=SwiftDeveloper",
+        "Next.jsAgent": "/?model=Next.jsAgent",
+        "MongoDBAgent": "/?model=MongoDBAgent",
+        "PyTorchAgent": "/?model=PyTorchAgent",
+        "ReactAgent": "/?model=ReactAgent",
+        "XcodeAgent": "/?model=XcodeAgent",
+        "AngularJSAgent": "/?model=AngularJSAgent",
+        "ImageGeneration": "/?model=ImageGeneration",
     }
 
     model_aliases = {
         "gemini-flash": "gemini-1.5-flash",
         "claude-3.5-sonnet": "claude-sonnet-3.5",
+        "flux": "ImageGeneration",
     }
 
     @classmethod