mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-26 01:24:55 +02:00
eb580209d4
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Quentin <torroba.q@gmail.com> Co-authored-by: Simon <80467011+sorgfresser@users.noreply.github.com> Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: angelapopopo <angele.lenglemetz@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import random
|
|
|
|
from rich.text import Text
|
|
from textual.app import ComposeResult
|
|
from textual.message import Message
|
|
from textual.widget import Widget
|
|
from textual.widgets import Static
|
|
|
|
from vibe.cli.textual_ui.widgets.chat_input.text_area import ChatTextArea
|
|
|
|
FEEDBACK_PROBABILITY = 0.02
|
|
THANK_YOU_DURATION = 2.0
|
|
|
|
|
|
class FeedbackBar(Widget):
|
|
class FeedbackGiven(Message):
|
|
def __init__(self, rating: int) -> None:
|
|
super().__init__()
|
|
self.rating = rating
|
|
|
|
@staticmethod
|
|
def _prompt_text() -> Text:
|
|
text = Text()
|
|
text.append("How is Vibe doing so far? ")
|
|
text.append("1", style="blue")
|
|
text.append(": good ")
|
|
text.append("2", style="blue")
|
|
text.append(": fine ")
|
|
text.append("3", style="blue")
|
|
text.append(": bad")
|
|
return text
|
|
|
|
def compose(self) -> ComposeResult:
|
|
yield Static(self._prompt_text(), id="feedback-text")
|
|
|
|
def on_mount(self) -> None:
|
|
self.display = False
|
|
|
|
def maybe_show(self) -> None:
|
|
if self.display:
|
|
return
|
|
if random.random() <= FEEDBACK_PROBABILITY:
|
|
self._set_active(True)
|
|
|
|
def hide(self) -> None:
|
|
if self.display:
|
|
self._set_active(False)
|
|
|
|
def handle_feedback_key(self, rating: int) -> None:
|
|
try:
|
|
self.app.query_one(ChatTextArea).feedback_active = False
|
|
except Exception:
|
|
pass
|
|
self.query_one("#feedback-text", Static).update(
|
|
Text("Thank you for your feedback!")
|
|
)
|
|
self.post_message(self.FeedbackGiven(rating))
|
|
self.set_timer(THANK_YOU_DURATION, lambda: self._set_active(False))
|
|
|
|
def _set_active(self, active: bool) -> None:
|
|
if active:
|
|
self.query_one("#feedback-text", Static).update(self._prompt_text())
|
|
self.display = active
|
|
try:
|
|
self.app.query_one(ChatTextArea).feedback_active = active
|
|
except Exception:
|
|
pass
|