thonny-codemate 0.1.0__py3-none-any.whl

thonnycontrib/thonny_codemate/ui/chat_view.py

@@ -0,0 +1,687 @@
+"""
+LLMチャットビュー
+GitHub Copilot風の右側パネルUIを提供
+"""
+import tkinter as tk
+from tkinter import ttk, scrolledtext, messagebox
+import threading
+import queue
+import logging
+import json
+from pathlib import Path
+from typing import Optional
+
+from thonny import get_workbench
+
+# 安全なロガーを使用
+try:
+    from .. import get_safe_logger
+    logger = get_safe_logger(__name__)
+except ImportError:
+    # フォールバック
+    logger = logging.getLogger(__name__)
+    logger.addHandler(logging.NullHandler())
+
+
+class LLMChatView(ttk.Frame):
+    """
+    LLMとのチャットインターフェースを提供するビュー
+    Thonnyの右側に表示される
+    """
+    
+    def __init__(self, master):
+        super().__init__(master)
+        
+        self.llm_client = None
+        self._init_ui()
+        self._init_llm()
+        
+        # メッセージキュー（スレッド間通信用）
+        self.message_queue = queue.Queue()
+        self._processing = False
+        self._first_token = True  # ストリーミング用のフラグ
+        self._stop_generation = False  # 生成を停止するフラグ
+        
+        # 定期的にキューをチェック
+        self._queue_check_id = self.after(100, self._process_queue)
+        
+        # ウィンドウ閉じるイベントをバインド
+        self.bind("<Destroy>", self._on_destroy)
+        
+        # チャット履歴を格納
+        self.chat_history = []
+        
+        # 会話履歴を読み込む
+        self._load_chat_history()
+    
+    def _init_ui(self):
+        """UIコンポーネントを初期化"""
+        # メインコンテナ
+        self.columnconfigure(0, weight=1)
+        self.rowconfigure(1, weight=1)
+        
+        # ヘッダー
+        header_frame = ttk.Frame(self)
+        header_frame.grid(row=0, column=0, sticky="ew", padx=5, pady=5)
+        
+        ttk.Label(header_frame, text="LLM Assistant", font=("", 12, "bold")).pack(side=tk.LEFT)
+        
+        # Clearボタン
+        self.clear_button = ttk.Button(
+            header_frame,
+            text="Clear",
+            command=self._clear_chat,
+            width=8
+        )
+        self.clear_button.pack(side=tk.LEFT, padx=10)
+        
+        # 設定ボタン
+        self.settings_button = ttk.Button(
+            header_frame,
+            text="⚙",
+            width=3,
+            command=self._show_settings
+        )
+        self.settings_button.pack(side=tk.RIGHT, padx=2)
+        
+        # ステータスフレーム（モデル名とステータス）
+        status_frame = ttk.Frame(header_frame)
+        status_frame.pack(side=tk.RIGHT, padx=5)
+        
+        self.status_label = ttk.Label(status_frame, text="Not loaded", foreground="gray")
+        self.status_label.pack(side=tk.RIGHT)
+        
+        # チャット表示エリア
+        chat_frame = ttk.Frame(self)
+        chat_frame.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
+        chat_frame.columnconfigure(0, weight=1)
+        chat_frame.rowconfigure(0, weight=1)
+        
+        self.chat_display = scrolledtext.ScrolledText(
+            chat_frame,
+            wrap=tk.WORD,
+            width=40,
+            height=20,
+            font=("Consolas", 10),
+            state=tk.DISABLED,
+            background="#f8f8f8"
+        )
+        self.chat_display.grid(row=0, column=0, sticky="nsew")
+        
+        # タグの設定
+        self.chat_display.tag_config("user", foreground="#0066cc", font=("Consolas", 10, "bold"))
+        self.chat_display.tag_config("assistant", foreground="#006600")
+        self.chat_display.tag_config("error", foreground="#cc0000")
+        self.chat_display.tag_config("code", background="#e8e8e8", font=("Consolas", 9))
+        
+        # 入力エリア
+        input_frame = ttk.Frame(self)
+        input_frame.grid(row=2, column=0, sticky="ew", padx=5, pady=5)
+        input_frame.columnconfigure(0, weight=1)
+        
+        # 入力テキスト
+        self.input_text = tk.Text(
+            input_frame,
+            height=3,
+            font=("Consolas", 10),
+            wrap=tk.WORD
+        )
+        self.input_text.grid(row=0, column=0, sticky="ew", pady=(0, 5))
+        
+        # ボタンフレーム
+        button_frame = ttk.Frame(input_frame)
+        button_frame.grid(row=1, column=0, sticky="ew")
+        
+        # send button        
+        self.send_button = ttk.Button(
+            button_frame,
+            text="Send",
+            command=self._handle_send_button,
+            state=tk.DISABLED
+        )
+        self.send_button.pack(side=tk.RIGHT, padx=2)
+
+        # Ctrl+Enterのヒントラベル
+        hint_label = ttk.Label(
+            button_frame,
+            text="Ctrl+Enter to send",
+            foreground="gray",
+            font=("", 9)
+        )
+        hint_label.pack(side=tk.RIGHT, padx=5)
+        
+        
+        # プリセットボタン（幅を指定して文字が切れないようにする）
+        ttk.Button(
+            button_frame,
+            text="Explain Error",
+            command=self._explain_last_error,
+            width=15  # 幅を指定
+        ).pack(side=tk.LEFT, padx=2)
+        
+        
+        # コンテキストボタン
+        self.context_var = tk.BooleanVar(value=False)
+        self.context_check = ttk.Checkbutton(
+            button_frame,
+            text="Include Context",
+            variable=self.context_var,
+            command=self._toggle_context
+        )
+        self.context_check.pack(side=tk.LEFT, padx=10)
+        
+        # コンテキストマネージャー
+        self.context_manager = None
+        
+        # キーバインディング
+        self.input_text.bind("<Control-Return>", lambda e: self._handle_send_button())
+        self.input_text.bind("<Shift-Return>", lambda e: "break")  # 改行を許可
+        
+        # Escapeキーで生成を停止
+        self.bind_all("<Escape>", lambda e: self._stop_if_processing())
+    
+    def _init_llm(self):
+        """LLMクライアントを初期化"""
+        try:
+            from .. import get_llm_client
+            from ..model_manager import ModelManager
+            
+            self.llm_client = get_llm_client()
+            
+            workbench = get_workbench()
+            provider = workbench.get_option("llm.provider", "local")
+            
+            if provider == "local":
+                # ローカルモデルの場合
+                model_path = workbench.get_option("llm.model_path", "")
+                
+                if not model_path or not Path(model_path).exists():
+                    # モデルがない場合、利用可能なモデルをチェック
+                    manager = ModelManager()
+                    available_model = manager.get_model_path("llama3.2-1b") or manager.get_model_path("llama3.2-3b")
+                    
+                    if available_model:
+                        # 利用可能なモデルがある場合は自動設定
+                        workbench.set_option("llm.model_path", available_model)
+                        model_path = available_model
+                    else:
+                        # モデルがない場合はダウンロードを促す
+                        self.status_label.config(text="No model loaded", foreground="red")
+                        self._append_message(
+                            "System",
+                            "No model found. Please download a model from Settings → Download Models.",
+                            "error"
+                        )
+                        
+                        # 設定ダイアログを開くボタンを表示
+                        # 親ウィンドウを指定してダイアログを表示
+                        if messagebox.askyesno(
+                            "No Model Found",
+                            "No LLM model found. Would you like to download recommended models?",
+                            parent=self
+                        ):
+                            self._show_settings()
+                        return
+                
+                # 非同期でモデルをロード
+                # モデル名を表示しながらロード
+                model_name = Path(model_path).name if model_path else "model"
+                self.status_label.config(text=f"Loading {model_name}...", foreground="orange")
+                self.llm_client.load_model_async(callback=self._on_model_loaded)
+            else:
+                # 外部プロバイダーの場合、モデル名も含める
+                external_model = workbench.get_option("llm.external_model", "")
+                display_text = f"{external_model} ({provider})" if external_model else f"Using {provider}"
+                self.status_label.config(text=display_text, foreground="blue")
+                # config取得で外部プロバイダーが設定される
+                self.llm_client.get_config()
+                self.send_button.config(state=tk.NORMAL)
+                self._append_message(
+                    "System", 
+                    f"Connected to {provider.upper()} API. Ready to chat!",
+                    "assistant"
+                )
+            
+        except Exception as e:
+            logger.error(f"Failed to initialize LLM client: {e}")
+            self.status_label.config(text="Error loading model", foreground="red")
+            self._append_message("System", f"Failed to initialize LLM: {str(e)}", "error")
+    
+    def _on_model_loaded(self, success: bool, error: Optional[Exception]):
+        """モデル読み込み完了時のコールバック"""
+        def update_ui():
+            if success:
+                # モデル名を取得
+                model_path = get_workbench().get_option("llm.model_path", "")
+                model_name = Path(model_path).name if model_path else "Unknown"
+                self.status_label.config(text=f"{model_name} | Ready", foreground="green")
+                self.send_button.config(state=tk.NORMAL)
+                self._append_message("System", "LLM model loaded successfully!", "assistant")
+            else:
+                self.status_label.config(text="Load failed", foreground="red")
+                self._append_message("System", f"Failed to load model: {error}", "error")
+        
+        # UIスレッドで更新
+        self.after(0, update_ui)
+    
+    def _append_message(self, sender: str, message: str, tag: str = None):
+        """チャット表示にメッセージを追加"""
+        self.chat_display.config(state=tk.NORMAL)
+        
+        # 送信者を表示
+        self.chat_display.insert(tk.END, f"\n{sender}: ", tag or "user")
+        
+        # メッセージを表示
+        if tag == "code" or "```" in message:
+            # コードブロックを検出して整形
+            parts = message.split("```")
+            for i, part in enumerate(parts):
+                if i % 2 == 0:
+                    self.chat_display.insert(tk.END, part, tag or "assistant")
+                else:
+                    # コードブロック
+                    lines = part.split('\n')
+                    if lines[0]:  # 言語指定がある場合
+                        self.chat_display.insert(tk.END, f"\n[{lines[0]}]\n", "assistant")
+                        code = '\n'.join(lines[1:])
+                    else:
+                        code = part
+                    self.chat_display.insert(tk.END, code, "code")
+        else:
+            self.chat_display.insert(tk.END, message, tag or "assistant")
+        
+        self.chat_display.insert(tk.END, "\n")
+        self.chat_display.see(tk.END)
+        self.chat_display.config(state=tk.DISABLED)
+        
+        # 履歴に追加（システムメッセージ以外）
+        if sender not in ["System"] and not message.startswith("["):
+            self.chat_history.append({"sender": sender, "message": message})
+            # ユーザーとアシスタントのメッセージのみ保存
+            if sender in ["User", "Assistant"]:
+                self._save_chat_history()
+    
+    def _handle_send_button(self):
+        """送信/停止ボタンのハンドラー"""
+        if self._processing:
+            # 生成中の場合は停止
+            self._stop_generation = True
+            self.send_button.config(text="Stopping...")
+            self._append_message("System", "Stopping generation...", "info")
+        else:
+            # 通常の送信処理
+            self._send_message()
+    
+    def _stop_if_processing(self):
+        """処理中の場合は生成を停止"""
+        if self._processing:
+            self._stop_generation = True
+            self.send_button.config(text="Stopping...")
+            self._append_message("System", "Stopping generation... (ESC pressed)", "info")
+    
+    def _send_message(self):
+        """メッセージを送信"""
+        message = self.input_text.get("1.0", tk.END).strip()
+        if not message:
+            return
+        
+        # UIをクリア
+        self.input_text.delete("1.0", tk.END)
+        
+        # コンテキスト情報を取得
+        context_info = None
+        if self.context_var.get() and self.context_manager:
+            workbench = get_workbench()
+            editor = workbench.get_editor_notebook().get_current_editor()
+            if editor:
+                current_file = editor.get_filename()
+                text_widget = editor.get_text_widget()
+                
+                if text_widget.tag_ranges("sel"):
+                    # 選択範囲がある場合
+                    start_line = int(text_widget.index("sel.first").split(".")[0])
+                    end_line = int(text_widget.index("sel.last").split(".")[0])
+                    file_name = Path(current_file).name if current_file else "Unknown"
+                    context_info = f"[Context: {file_name} (lines {start_line}-{end_line})]"
+                elif current_file:
+                    # ファイル全体の場合
+                    file_name = Path(current_file).name
+                    context_info = f"[Context: {file_name} (entire file)]"
+        
+        # ユーザーメッセージを追加（コンテキスト情報付き）
+        if context_info:
+            display_message = f"{message}\n\n{context_info}"
+        else:
+            display_message = message
+        
+        self._append_message("You", display_message, "user")
+        
+        # 処理中フラグ
+        self._processing = True
+        self._first_token = True  # ストリーミング用フラグをリセット
+        self._stop_generation = False  # 停止フラグをリセット
+        self.send_button.config(text="Stop", state=tk.NORMAL)  # ボタンを停止モードに変更
+        
+        # バックグラウンドで処理
+        thread = threading.Thread(
+            target=self._generate_response,
+            args=(message,),  # 元のメッセージを渡す（コンテキスト情報なし）
+            daemon=True
+        )
+        thread.start()
+    
+    def _generate_response(self, message: str):
+        """バックグラウンドで応答を生成"""
+        try:
+            # コンテキストを含める場合
+            if self.context_var.get() and self.context_manager:
+                # 現在のエディタのファイルパスを取得
+                workbench = get_workbench()
+                editor = workbench.get_editor_notebook().get_current_editor()
+                current_file = None
+                selected_text = None
+                selection_info = None
+                
+                if editor:
+                    current_file = editor.get_filename()
+                    text_widget = editor.get_text_widget()
+                    
+                    # 選択範囲があるかチェック
+                    if text_widget.tag_ranges("sel"):
+                        selected_text = text_widget.get("sel.first", "sel.last")
+                        # 選択範囲の行番号を取得
+                        start_line = int(text_widget.index("sel.first").split(".")[0])
+                        end_line = int(text_widget.index("sel.last").split(".")[0])
+                        selection_info = f"Selected lines: {start_line}-{end_line}"
+                
+                if selected_text:
+                    # 選択範囲がある場合はそれをコンテキストとして使用
+                    # ファイル拡張子から言語を判定
+                    file_ext = Path(current_file).suffix.lower() if current_file else '.py'
+                    lang_map = {'.py': 'python', '.js': 'javascript', '.java': 'java', '.cpp': 'cpp', '.c': 'c'}
+                    lang = lang_map.get(file_ext, 'python')
+                    
+                    context_str = f"""File: {Path(current_file).name if current_file else 'Unknown'}
+{selection_info}
+
+```{lang}
+{selected_text}
+```"""
+                    # コンテキスト使用メッセージは不要
+                elif editor and current_file:
+                    # 選択範囲がない場合は、ファイル全体を取得
+                    full_text = text_widget.get("1.0", tk.END).strip()
+                    if full_text:
+                        # ファイル拡張子から言語を判定
+                        file_ext = Path(current_file).suffix.lower()
+                        lang_map = {'.py': 'python', '.js': 'javascript', '.java': 'java', '.cpp': 'cpp', '.c': 'c'}
+                        lang = lang_map.get(file_ext, 'python')
+                        
+                        context_str = f"""File: {Path(current_file).name}
+Full file content:
+
+```{lang}
+{full_text}
+```"""
+                        # コンテキスト使用メッセージは不要
+                    else:
+                        context_str = None
+                else:
+                    context_str = None
+                
+                if context_str:
+                    # コンテキスト付きで生成
+                    full_prompt = f"""Here is the context from the current project:
+
+{context_str}
+
+Based on this context, {message}"""
+                    
+                    for token in self.llm_client.generate_stream(full_prompt):
+                        if self._stop_generation:
+                            self.message_queue.put(("info", "\n[Generation stopped by user]"))
+                            break
+                        self.message_queue.put(("token", token))
+                else:
+                    # 通常の生成
+                    for token in self.llm_client.generate_stream(message):
+                        if self._stop_generation:
+                            self.message_queue.put(("info", "\n[Generation stopped by user]"))
+                            break
+                        self.message_queue.put(("token", token))
+            else:
+                # 通常の生成
+                for token in self.llm_client.generate_stream(message):
+                    if self._stop_generation:
+                        self.message_queue.put(("info", "\n[Generation stopped by user]"))
+                        break
+                    self.message_queue.put(("token", token))
+            
+            # 完了
+            self.message_queue.put(("complete", None))
+            
+        except Exception as e:
+            logger.error(f"Error generating response: {e}")
+            self.message_queue.put(("error", str(e)))
+    
+    def _process_queue(self):
+        """メッセージキューを処理"""
+        try:
+            # キューから全てのメッセージを処理
+            while True:
+                msg_type, content = self.message_queue.get_nowait()
+                
+                if msg_type == "token":
+                    if self._first_token:
+                        # 最初のトークンの時だけAssistantラベルを追加
+                        self.chat_display.config(state=tk.NORMAL)
+                        self.chat_display.insert(tk.END, "\nAssistant: ", "role")
+                        self.chat_display.config(state=tk.DISABLED)
+                        self._first_token = False
+                    
+                    # トークンを追加（ラベルなしで）
+                    self.chat_display.config(state=tk.NORMAL)
+                    self.chat_display.insert(tk.END, content, "assistant")
+                    self.chat_display.see(tk.END)
+                    self.chat_display.config(state=tk.DISABLED)
+                
+                elif msg_type == "complete":
+                    self._processing = False
+                    self._stop_generation = False  # 停止フラグをリセット
+                    self.send_button.config(text="Send", state=tk.NORMAL)  # ボタンを送信モードに戻す
+                    self._first_token = True  # 次のメッセージ用にリセット
+                
+                elif msg_type == "error":
+                    self._append_message("System", f"Error: {content}", "error")
+                    self._processing = False
+                    self._stop_generation = False  # 停止フラグをリセット
+                    self.send_button.config(text="Send", state=tk.NORMAL)  # ボタンを送信モードに戻す
+                    self._first_token = True  # 次のメッセージ用にリセット
+                
+                elif msg_type == "info":
+                    self._append_message("System", content, "assistant")
+                    self._first_token = True  # 次のメッセージ用にリセット
+        
+        except queue.Empty:
+            pass
+        
+        # 次のチェックをスケジュール
+        self.after(50, self._process_queue)
+    
+    def explain_code(self, code: str):
+        """コードを説明（外部から呼ばれる）"""
+        # 現在のスキルレベルを取得
+        workbench = get_workbench()
+        skill_level = workbench.get_option("llm.skill_level", "beginner")
+        
+        # 現在のファイルから言語を検出
+        editor = workbench.get_editor_notebook().get_current_editor()
+        lang = 'python'  # デフォルト
+        if editor:
+            filename = editor.get_filename()
+            if filename:
+                file_ext = Path(filename).suffix.lower()
+                lang_map = {'.py': 'python', '.js': 'javascript', '.java': 'java', '.cpp': 'cpp', '.c': 'c'}
+                lang = lang_map.get(file_ext, 'python')
+        
+        # メッセージを作成
+        message = f"Please explain this code:\n```{lang}\n{code}\n```"
+        
+        # 入力欄に設定して送信
+        self.input_text.delete("1.0", tk.END)
+        self.input_text.insert("1.0", message)
+        self._send_message()
+    
+    def _explain_last_error(self):
+        """最後のエラーを説明"""
+        try:
+            # シェルビューを取得
+            shell_view = get_workbench().get_view("ShellView")
+            if not shell_view:
+                messagebox.showinfo("No Shell", "Shell view not found.")
+                return
+            
+            # シェルのテキストウィジェットを取得
+            shell_text = shell_view.text
+            
+            # シェルの内容を取得（最後の部分）
+            shell_content = shell_text.get("1.0", tk.END)
+            lines = shell_content.strip().split('\n')
+            
+            # エラーを探す（後ろから検索）
+            error_lines = []
+            error_found = False
+            
+            for i in range(len(lines) - 1, -1, -1):
+                line = lines[i]
+                
+                # エラーの終わりを検出
+                if error_found and (line.startswith(">>>") or line.startswith("===") or not line.strip()):
+                    break
+                
+                # エラーパターンを検出
+                if any(error_type in line for error_type in ["Error", "Exception", "Traceback"]):
+                    error_found = True
+                
+                if error_found:
+                    error_lines.insert(0, line)
+            
+            if not error_lines:
+                messagebox.showinfo("No Error", "No recent error found in shell.")
+                return
+            
+            # エラーメッセージを結合
+            error_message = '\n'.join(error_lines)
+            
+            # 関連するコードも取得（エディタから）
+            code = ""
+            editor = get_workbench().get_editor_notebook().get_current_editor()
+            if editor:
+                try:
+                    code = editor.get_text_widget().get("1.0", tk.END).strip()
+                except:
+                    pass
+            
+            # プロンプトを作成
+            if code:
+                prompt = f"I encountered this error:\n```\n{error_message}\n```\n\nIn this code:\n```python\n{code}\n```\n\nPlease explain what causes this error and how to fix it."
+            else:
+                prompt = f"I encountered this error:\n```\n{error_message}\n```\n\nPlease explain what causes this error and provide a solution."
+            
+            # 入力欄に設定して送信
+            self.input_text.delete("1.0", tk.END)
+            self.input_text.insert("1.0", prompt)
+            self._send_message()
+            
+        except Exception as e:
+            logger.error(f"Error in _explain_last_error: {e}")
+            messagebox.showerror("Error", f"Failed to get error information: {str(e)}")
+    
+    def _show_settings(self):
+        """設定ダイアログを表示"""
+        from .settings_dialog import SettingsDialog
+        dialog = SettingsDialog(self)
+        dialog.grab_set()
+        self.wait_window(dialog)
+        
+        # 設定が変更された可能性があるので再初期化
+        if hasattr(dialog, 'settings_changed') and dialog.settings_changed:
+            self._init_llm()
+    
+    def _clear_chat(self):
+        """チャットをクリア"""
+        self.chat_display.config(state=tk.NORMAL)
+        self.chat_display.delete("1.0", tk.END)
+        self.chat_display.config(state=tk.DISABLED)
+        
+        # 履歴もクリア
+        self.chat_history.clear()
+        self._save_chat_history()
+    
+    def _toggle_context(self):
+        """コンテキストの有効/無効を切り替え"""
+        if self.context_var.get():
+            # コンテキストマネージャーを初期化
+            if not self.context_manager:
+                from ..context_manager import ContextManager
+                self.context_manager = ContextManager()
+    
+    def _get_chat_history_path(self) -> Path:
+        """チャット履歴ファイルのパスを取得"""
+        workbench = get_workbench()
+        data_dir = Path(workbench.get_configuration_directory())
+        llm_dir = data_dir / "llm_assistant"
+        llm_dir.mkdir(exist_ok=True)
+        return llm_dir / "chat_history_text.json"
+    
+    def _save_chat_history(self):
+        """チャット履歴を保存"""
+        try:
+            # 最新の100メッセージのみ保存
+            messages_to_save = self.chat_history[-100:]
+            
+            history_path = self._get_chat_history_path()
+            with open(history_path, 'w', encoding='utf-8') as f:
+                json.dump(messages_to_save, f, ensure_ascii=False, indent=2)
+                
+        except Exception as e:
+            logger.error(f"Failed to save chat history: {e}")
+    
+    def _load_chat_history(self):
+        """チャット履歴を読み込む"""
+        try:
+            history_path = self._get_chat_history_path()
+            if history_path.exists():
+                with open(history_path, 'r', encoding='utf-8') as f:
+                    saved_messages = json.load(f)
+                
+                # 保存されたメッセージを復元
+                for msg in saved_messages:
+                    self._append_message(msg["sender"], msg["message"])
+                    # _append_messageがさらに履歴に追加するのを防ぐため、最後のものを削除
+                    if self.chat_history and self.chat_history[-1] == msg:
+                        self.chat_history.pop()
+                
+                self.chat_history = saved_messages
+                
+                if self.chat_history:
+                    self._append_message("System", "Previous conversation restored", "info")
+                    
+        except Exception as e:
+            logger.error(f"Failed to load chat history: {e}")
+    
+    def _on_destroy(self, event):
+        """ウィンドウが破棄される時のクリーンアップ"""
+        # チャット履歴を保存
+        self._save_chat_history()
+        
+        # キューチェックを停止
+        if hasattr(self, '_queue_check_id'):
+            self.after_cancel(self._queue_check_id)
+        
+        # 生成を停止
+        self._stop_generation = True
+        
+        # LLMクライアントをシャットダウン
+        if self.llm_client:
+            self.llm_client.shutdown()