Update inference.py

inference.py

@@ -1,15 +1,14 @@
 # ============================================================
-# RQA UX Inference — FINAL
+# RQA UX Inference — IMPROVED INTERACTIVE VERSION
 # Google Colab + CLI friendly
 # ============================================================
 
 import os
 import sys
 import json
-import argparse
 import csv
 import torch
-from typing import List, Union
+from typing import List
 from transformers import AutoTokenizer, AutoModel
 
 # ============================================================
@@ -193,72 +192,183 @@ def load_texts_from_file(path: str) -> List[str]:
     raise ValueError("Неподдерживаемый формат файла")
 
 # ============================================================
-# CLI / Colab entrypoint
+# Interactive CLI Interface
 # ============================================================
 
-def main():
-    parser = argparse.ArgumentParser(
-        description="RQA — анализ логических ошибок"
-    )
-
-    parser.add_argument(
-        "--text",
-        type=str,
-        help="Один текст для анализа"
-    )
-
-    parser.add_argument(
-        "--file",
-        type=str,
-        help="Файл с текстами (.txt, .csv, .json)"
-    )
-
-    parser.add_argument(
-        "--multiline",
-        action="store_true",
-        help="Ввод нескольких строк (каждая строка — отдельный текст)"
-    )
-
-    args, unknown = parser.parse_known_args()
-
-    judge = RQAJudge()
-
-    texts = []
-
-    # ---------- FILE MODE ----------
-    if args.file:
-        if not os.path.exists(args.file):
-            raise FileNotFoundError(args.file)
-        texts = load_texts_from_file(args.file)
-
-    # ---------- SINGLE TEXT ----------
-    elif args.text:
-        texts = [args.text]
-
-    # ---------- MULTILINE ----------
-    elif args.multiline:
-        print("Введите тексты (пустая строка — конец ввода):")
+class InteractiveCLI:
+    def __init__(self):
+        self.judge = RQAJudge()
+        self.mode_stack = []
+        
+    def clear_screen(self):
+        """Очистка экрана для Google Colab"""
+        print("\n" * 2)
+        
+    def show_mode_menu(self):
+        """Показать меню выбора режима"""
+        self.clear_screen()
+        print("=" * 60)
+        print("🤖 RQA — АНАЛИЗ ЛОГИЧЕСКИХ ОШИБОК")
+        print("=" * 60)
+        print("\nВыберите режим работы:")
+        print("1. 📝 Одиночный ввод (одна фраза для анализа)")
+        print("2. 📄 Множественный ввод (несколько фраз, каждая с новой строки)")
+        print("3. 📂 Загрузка из файла (.txt, .csv, .json)")
+        print("\nНажмите Enter без ввода для выхода.")
+        print("-" * 60)
+        
+    def process_single_mode(self):
+        """Обработка одиночного режима"""
+        self.clear_screen()
+        print("[📝 РЕЖИМ: ОДИНОЧНЫЙ ВВОД]")
+        print("Введите текст для анализа:")
+        print("(Нажмите Enter без ввода для возврата в меню)")
+        print("-" * 40)
+        
+        text = input("> ").strip()
+        if not text:
+            return True  # Возврат в меню
+            
+        result = self.judge.infer(text)
+        self.judge.pretty_print(result)
+        
+        print("\n" + "-" * 40)
+        input("Нажмите Enter для продолжения...")
+        return False  # Остаемся в том же режиме
+        
+    def process_multiline_mode(self):
+        """Обработка режима множественного ввода"""
+        self.clear_screen()
+        print("[📄 РЕЖИМ: МНОЖЕСТВЕННЫЙ ВВОД]")
+        print("Введите тексты для анализа (каждый с новой строки).")
+        print("Оставьте строку пустой для завершения ввода.")
+        print("(Нажмите Enter без ввода для возврата в меню)")
+        print("-" * 40)
+        
+        texts = []
+        print("Ввод текстов:")
+        while True:
+            line = input("> ").strip()
+            if not line:
+                if not texts:  # Пустой ввод сразу - возврат в меню
+                    return True
+                break  # Завершение ввода
+            texts.append(line)
+            
+        if texts:
+            self.clear_screen()
+            print(f"[📄 РЕЖИМ: МНОЖЕСТВЕННЫЙ ВВОД] — найдено {len(texts)} текстов")
+            print("-" * 40)
+            
+            for i, text in enumerate(texts, 1):
+                print(f"\n🔍 Текст #{i}:")
+                result = self.judge.infer(text)
+                self.judge.pretty_print(result)
+                
+            print("\n" + "=" * 60)
+            input("Нажмите Enter для продолжения...")
+            
+        return False  # Остаемся в том же режиме
+        
+    def process_file_mode(self):
+        """Обработка режима загрузки из файла"""
+        self.clear_screen()
+        print("[📂 РЕЖИМ: ЗАГРУЗКА ИЗ ФАЙЛА]")
+        print("Поддерживаемые форматы: .txt, .csv, .json")
+        print("Укажите путь к файлу:")
+        print("(Нажмите Enter без ввода для возврата в меню)")
+        print("-" * 40)
+        
+        file_path = input("Путь к файлу> ").strip()
+        if not file_path:
+            return True  # Возврат в меню
+            
+        try:
+            # Проверка существования файла
+            if not os.path.exists(file_path):
+                print(f"\n❌ Ошибка: Файл '{file_path}' не найден!")
+                input("\nНажмите Enter для продолжения...")
+                return False  # Остаемся в том же режиме
+                
+            # Загрузка текстов
+            texts = load_texts_from_file(file_path)
+            
+            if not texts:
+                print(f"\n⚠️  Файл '{file_path}' пуст или не содержит текстов!")
+                input("\nНажмите Enter для продолжения...")
+                return False  # Остаемся в том же режиме
+                
+            # Обработка текстов
+            self.clear_screen()
+            print(f"[📂 РЕЖИМ: ЗАГРУЗКА ИЗ ФАЙЛА] — загружено {len(texts)} текстов")
+            print(f"Файл: {file_path}")
+            print("-" * 40)
+            
+            for i, text in enumerate(texts, 1):
+                print(f"\n🔍 Текст #{i}:")
+                result = self.judge.infer(text)
+                self.judge.pretty_print(result)
+                
+            print("\n" + "=" * 60)
+            input("Нажмите Enter для продолжения...")
+            
+        except Exception as e:
+            print(f"\n❌ Ошибка при обработке файла: {str(e)}")
+            input("\nНажмите Enter для продолжения...")
+            
+        return False  # Остаемся в том же режиме
+        
+    def run_interactive(self):
+        """Основной цикл интерактивного интерфейса"""
+        current_mode = None
+        
         while True:
-            line = input()
-            if not line.strip():
-                break
-            texts.append(line.strip())
+            # Если нет текущего режима, показываем главное меню
+            if not current_mode:
+                self.show_mode_menu()
+                choice = input("Ваш выбор (1-3)> ").strip()
+                
+                if not choice:  # Пустой ввод - выход
+                    print("\n👋 Выход из программы...")
+                    break
+                    
+                if choice == "1":
+                    current_mode = "single"
+                elif choice == "2":
+                    current_mode = "multiline"
+                elif choice == "3":
+                    current_mode = "file"
+                else:
+                    print("\n❌ Неверный выбор! Попробуйте снова.")
+                    input("Нажмите Enter для продолжения...")
+                    continue
+                    
+            # Обработка текущего режима
+            should_return_to_menu = False
+            
+            if current_mode == "single":
+                should_return_to_menu = self.process_single_mode()
+            elif current_mode == "multiline":
+                should_return_to_menu = self.process_multiline_mode()
+            elif current_mode == "file":
+                should_return_to_menu = self.process_file_mode()
+                
+            # Возврат в меню при необходимости
+            if should_return_to_menu:
+                current_mode = None
 
-    # ---------- INTERACTIVE FALLBACK ----------
-    else:
-        print("Введите текст для анализа:")
-        line = input().strip()
-        if line:
-            texts = [line]
-        else:
-            print("❌ Пустой ввод — выхожу")
-            return
+# ============================================================
+# Точка входа
+# ============================================================
 
-    # ---------- RUN ----------
-    for t in texts:
-        result = judge.infer(t)
-        judge.pretty_print(result)
+def main():
+    """Основная функция - запускает интерактивный интерфейс"""
+    cli = InteractiveCLI()
+    cli.run_interactive()
 
+# ============================================================
+# Запуск
+# ============================================================
 
 if __name__ == "__main__":
-    main()
+    main()