Update main.py

main.py

@@ -209,20 +209,25 @@ async def chat_completions(request: ChatRequest):
 
     async_generator = Blackbox.create_async_generator(
         model=request.model,
-        messages=messages
+        messages=messages,
+        image=None,
+        image_name=None
     )
 
     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"
+                yield f"data: {json.dumps(create_response(chunk.content if isinstance(chunk, ImageResponse) else chunk, request.model))}\n\n"  # Replace this line
             yield "data: [DONE]\n\n"
 
         return StreamingResponse(generate(), media_type="text/event-stream")
     else:
         response_content = ""
         async for chunk in async_generator:
-            response_content += chunk.content if isinstance(chunk, ImageResponse) else chunk  # Concatenate response
+            if isinstance(chunk, ImageResponse):
+                response_content += f"Image URL: {chunk.url}\n"
+            else:
+                response_content += chunk
 
         return {
             "id": f"chatcmpl-{uuid.uuid4()}",