ui/console.py - master - sc_p/EnergoAI

ui/console.py
69 строк3 KB
Павел Щёголев
Загрузить файлы в «»
12 мар 2026, 04:54
12 мар 2026, 04:54628f0d6
def run_console(on_send_chat, on_run_code, initial_role="default"):
    print("\n" + "=" * 70)
    print("ИИ-АССИСТЕНТ ДЛЯ ЭНЕРГЕТИЧЕСКИХ РАСЧЕТОВ - КОНСОЛЬНЫЙ РЕЖИМ")
    print("=" * 70)
    print("\nДОСТУПНЫЕ КОМАНДЫ:")
    print(" /exit - выход из программы")
    print(" /code - войти в режим ввода кода")
    print(" /clear - очистить историю")
    print(" /help - показать справку")
    print("=" * 70)

    while True:
        try:
            user = input(f"\n Вы: ").strip()

            if user.lower() == "/exit":
                print("Завершение работы...")
                break

            elif user.lower() == "/help":
                print("\nСПРАВКА ПО КОМАНДАМ:")
                print(" /exit - выход из программы")
                print(" /code - войти в режим ввода Python кода")
                print(" /clear - очистить историю чата")
                print(" /help - эта справка")
                continue

            elif user.lower() == "/code":
                print("\n" + "=" * 50)
                print("РЕЖИМ ВВОДА КОДА (Python для энергетических расчётов)")
                print("Введите код построчно. Пустая строка - выполнить.")
                print("=" * 50)
                lines = []
                while True:
                    try:
                        line = input(">>> " if not lines else "... ")
                        if line == "":
                            break
                        lines.append(line)
                    except KeyboardInterrupt:
                        print("\n✗ Ввод кода прерван")
                        lines = []
                        break

                if lines:
                    code = "\n".join(lines)
                    print(f"\nВыполняю код...\n")
                    on_run_code(code)
                else:
                    print("Код не введён")
                continue

            elif user.lower() == "/clear":
                print("\nИстория чата очищена (визуально)")
                continue

            # Обычное сообщение
            if user:
                on_send_chat(user)

        except KeyboardInterrupt:
            print("\n\nДля выхода введите /exit")
            continue
        except EOFError:
            print("\n\nЗавершение работы...")
            break
        except Exception as e:
            print(f"\nОшибка: {e}")
            continue