Update main.py

main.py

@@ -210,14 +210,17 @@ async def chat_completions(request: ChatRequest):
     async_generator = Blackbox.create_async_generator(
         model=request.model,
         messages=messages,
-        image=None,
-        image_name=None
+        image=None,  # Pass the image if required
+        image_name=None  # Pass image name if required
     )
 
     if request.stream:
         async def generate():
             async for chunk in async_generator:
-                yield f"data: {json.dumps(create_response(chunk.content if isinstance(chunk, ImageResponse) else chunk, request.model))}\n\n"  # Replace this line
+                if isinstance(chunk, ImageResponse):
+                    yield f"data: {json.dumps(create_response(chunk.url, request.model))}\n\n"
+                else:
+                    yield f"data: {json.dumps(create_response(chunk, request.model))}\n\n"
             yield "data: [DONE]\n\n"
 
         return StreamingResponse(generate(), media_type="text/event-stream")
@@ -227,7 +230,7 @@ async def chat_completions(request: ChatRequest):
             if isinstance(chunk, ImageResponse):
                 response_content += f"Image URL: {chunk.url}\n"
             else:
-                response_content += chunk
+                response_content += chunk  # Concatenate text responses
 
         return {
             "id": f"chatcmpl-{uuid.uuid4()}",