cli.py

from textual.app import App, ComposeResult
from textual.widgets import Header, Footer, Input, RichLog
from chatbot import get_answer  
import asyncio


class ChatCLI(App):

    CSS = """
    Screen {
        background: black;
    }

    #chat {
        border: round cyan;
        padding: 1;
    }

    Input {
        border: round green;
    }
    """

    def compose(self) -> ComposeResult:
        yield Header(show_clock=True)
        yield RichLog(id="chat", wrap=True)
        yield Input(placeholder="Type your message...")
        yield Footer()

    # =========================
    # INPUT HANDLER
    # =========================
    def on_input_submitted(self, event: Input.Submitted):

        user_text = event.value.strip()

        if not user_text:
            return

        chat = self.query_one(RichLog)

        # USER MESSAGE
        chat.write(f"You:  {user_text}")

        # CLEAR INPUT
        event.input.value = ""

        # RUN AI ASYNC (IMPORTANT FIX)
        asyncio.create_task(self.ask_ai(user_text))

    # =========================
    # AI HANDLER
    # =========================
    async def ask_ai(self, text: str):

        chat = self.query_one(RichLog)

        # LOADING MESSAGE
        chat.write(" Bot is thinking... ")

        try:
            # CALL YOUR AI FUNCTION
            answer = await get_answer(text)

            chat.write(f" Bot:  {answer}")

        except Exception as e:
            chat.write(f" Error:  {e}")


# =========================
# RUN APP
# =========================
if __name__ == "__main__":
    ChatCLI().run()