Update main.py

main.py

@@ -169,8 +169,13 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
             
             async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response:
                 response.raise_for_status()
+                response_text = await response.text()
+                
+                # Check if the response is empty
+                if not response_text.strip():
+                    raise ModelNotWorkingException(model)
+
                 if model == 'ImageGenerationLV45LJp':
-                    response_text = await response.text()
                     url_match = re.search(r'https://storage\.googleapis\.com/[^\s\)]+', response_text)
                     if url_match:
                         image_url = url_match.group(0)
@@ -271,4 +276,4 @@ async def chat_completions(request: ChatRequest):
 
 @app.get("/niansuhai/v1/models")
 async def get_models():
-    return {"models": Blackbox.models}
+    return {"models": Blackbox.models}