Update main.py

main.py

@@ -106,7 +106,6 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
     ) -> Any:
         model = cls.get_model(model)
 
-        # Check if the model is working
         if not cls.working or model not in cls.models:
             raise ModelNotWorkingException(model)
 
@@ -176,8 +175,8 @@ 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
+
+                    # Check if the response is empty or just whitespace
                     if not response_text.strip():
                         raise ModelNotWorkingException(model)
 
@@ -188,13 +187,15 @@ class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
                             yield ImageResponse(image_url, alt=messages[-1]['content'])
                         else:
                             raise Exception("Image URL not found in the response")
-                    else:
-                        async for chunk in response.content.iter_any():
-                            if chunk:
-                                decoded_chunk = chunk.decode(errors='ignore')  # Handle decoding errors
-                                decoded_chunk = re.sub(r'\$@\$v=[^$]+\$@\$', '', decoded_chunk)
-                                if decoded_chunk.strip():
-                                    yield decoded_chunk
+                    
+                    # Handle normal responses
+                    async for chunk in response.content.iter_any():
+                        if chunk:
+                            decoded_chunk = chunk.decode(errors='ignore')
+                            decoded_chunk = re.sub(r'\$@\$v=[^$]+\$@\$', '', decoded_chunk)
+                            if decoded_chunk.strip():
+                                yield decoded_chunk
+                        
             except ClientError as e:
                 logger.error(f"HTTP request failed: {e}")
                 raise HTTPException(status_code=503, detail="Service is unavailable. Please try again later.")