diff --git a/oai.py b/oai.py index fc0636d..6825803 100644 --- a/oai.py +++ b/oai.py @@ -147,8 +147,11 @@ def chat(): selected_model = selected_model_default # Set from DB load, or None client = OpenRouter(api_key=API_KEY) - - console.print("[bold blue]Welcome to oAI! Type your message, '/model [search]' to select/change model, '/config api/url/stream' to configure, '/help' for examples, or 'exit'/'quit' to end.[/]") + if selected_model: + console.print("[bold blue]Welcome to oAI! Type your message, use '/help' for examples, or 'exit'/'quit'/'bye' to end.[/] \n[bold red]Active model : %s[/]" %(selected_model["name"])) + else: + console.print("[bold blue]Welcome to oAI![/] [italic blue]No active model. Use: '/model [search]' to select model.[/] [bold blue]Use '/help' for examples, or 'exit'/'quit'/'bye' to end.[/]") + if not selected_model: console.print("[bold yellow]No model selected. Use '/model' to choose one.[/]") @@ -294,7 +297,8 @@ def chat(): # Handle /clear command to clear the screen if user_input.lower() == "/clear": clear_screen() - console.print("[bold cyan]Screen cleared. Ready for your next input![/]") + DEFAULT_MODEL_ID = get_config('default_model') + console.print("[bold blue]Active model[/] [bold red]%s[/]" %(DEFAULT_MODEL_ID)) continue # Handle /help command. @@ -481,4 +485,5 @@ def chat(): console.print("[bold yellow]Try again or use '/model' to select a different model.[/]") if __name__ == "__main__": + clear_screen() app() \ No newline at end of file