Update main.py

main.py

@@ -218,7 +218,9 @@ async def chat_completions(request: ChatRequest):
         async def generate():
             async for chunk in async_generator:
                 if isinstance(chunk, ImageResponse):
-                    yield f"data: {json.dumps(create_response(chunk.url, request.model))}\n\n"
+                    # Format the response as a Markdown image
+                    image_markdown = f"![image]({chunk.url})"
+                    yield f"data: {json.dumps(create_response(image_markdown, request.model))}\n\n"
                 else:
                     yield f"data: {json.dumps(create_response(chunk, request.model))}\n\n"
             yield "data: [DONE]\n\n"
@@ -228,7 +230,8 @@ async def chat_completions(request: ChatRequest):
         response_content = ""
         async for chunk in async_generator:
             if isinstance(chunk, ImageResponse):
-                response_content += f"Image URL: {chunk.url}\n"
+                # Add Markdown image to the response
+                response_content += f"![image]({chunk.url})\n"
             else:
                 response_content += chunk  # Concatenate text responses