feat: wip, v2

views/GenerationState.py

@@ -0,0 +1,21 @@
+from enum import Enum
+
+import discord
+
+class GenerationState(Enum):
+    GENERATING = "generating"
+    FINISHED = "finished"
+
+class GenerationStateButton(discord.ui.Button):
+    def __init__(self, label: str, style: discord.ButtonStyle, emoji: str | discord.Emoji | discord.PartialEmoji | None = None, disabled: bool=False):
+        super().__init__(label=label, style=style, emoji=emoji, disabled=disabled)
+
+class GenerationStateView(discord.ui.View):
+    def __init__(self, state: GenerationState):
+        super().__init__()
+        self.state = state
+
+        if state == GenerationState.GENERATING:
+            self.add_item(GenerationStateButton(label="Generating...", style=discord.ButtonStyle.grey, emoji="✨", disabled=True))
+        elif state == GenerationState.FINISHED:
+            pass