"""
title: Markdown Image Remover
author: Violet
funding_url: https://github.com/open-webui
version: 1.0
"""

import re
from typing import Callable, Any, Optional, Awaitable


def remove_markdown_images(content: str) -> str:
    return re.sub(r'!\[.*?\]\([^)]*\)', '', content)


class EventEmitter:
    def __init__(self, event_emitter: Callable[[dict], Any] = None):
        self.event_emitter = event_emitter

    async def emit(self, description="Unknown State", status="in_progress", done=False):
        if self.event_emitter:
            await self.event_emitter(
                {
                    "type": "status",
                    "data": {
                        "status": status,
                        "description": description,
                        "done": done,
                    },
                }
            )


class Filter:
    def __init__(self):
        self.output_text = ""

    async def inlet(
            self,
            body: dict,
            __event_emitter__: Callable[[Any], Awaitable[None]],
            __user__: Optional[dict] = None,
            __model__: Optional[dict] = None,
    ) -> dict:
        emitter = EventEmitter(__event_emitter__)

        await emitter.emit("Removing markdown Images")

        for i, msg in enumerate(body["messages"]):
            body["messages"][i]["content"] = remove_markdown_images(msg["content"])

        await emitter.emit(status="complete", description="Markdown images removed", done=True)

        return body