diff --git a/requirements.txt b/requirements.txt index 5a638c20e..e4fb1be5f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,26 +1,32 @@ -sphinx>=5.0.0 -sphinx_rtd_theme>=1.0.0 -contourpy==1.1.1 -cycler==0.12.1 -entrypoints==0.3 -flake8==3.7.7 -fonttools==4.57.0 -https://github.com/aishw31/hdlparse/tarball/python3compat -importlib_resources==6.4.5 -kiwisolver==1.4.7 -matplotlib==3.7.5 -mccabe==0.6.1 -numpy==1.24.4 -packaging==25.0 -pillow==10.4.0 -pycodestyle==2.5.0 -pyflakes==2.1.1 -pyparsing==3.1.4 -PyQt5==5.15.7 -PyQt5-Qt5==5.15.17 -PyQt5_sip==12.15.0 -python-dateutil==2.9.0.post0 -scipy==1.10.1 -six==1.17.0 -watchdog==4.0.2 -zipp==3.20.2 \ No newline at end of file +sphinx>=5.0.0 +sphinx_rtd_theme>=1.0.0 +contourpy==1.1.1 +cycler==0.12.1 +entrypoints==0.3 +flake8==3.7.7 +fonttools==4.57.0 +importlib_resources==6.4.5 +kiwisolver==1.4.7 +matplotlib==3.7.5 +mccabe==0.6.1 +numpy==1.24.4 +packaging==25.0 +pillow==10.4.0 +pycodestyle==2.5.0 +pyflakes==2.1.1 +pyparsing==3.1.4 +PyQt5>=5.15.7 +PyQt5-Qt5==5.15.17 +PyQt5-sip>=12.15.0 +PyQtWebEngine +python-dateutil==2.9.0.post0 +scipy==1.10.1 +six==1.17.0 +watchdog==4.0.2 +zipp==3.20.2 +setuptools>=57.5.0,<58 +wheel +ollama>=0.3.0 +faster-whisper>=1.0.0 +SpeechRecognition>=3.10.0 +PyAudio>=0.2.13 \ No newline at end of file diff --git a/src/chatbot/chatbot_thread.py b/src/chatbot/chatbot_thread.py new file mode 100644 index 000000000..702c6fcd0 --- /dev/null +++ b/src/chatbot/chatbot_thread.py @@ -0,0 +1,670 @@ +import os +import re +import socket +import subprocess +import time +import threading +import ollama +from PyQt5.QtCore import QThread, pyqtSignal + +# ── Optional imports ────────────────────────────────────────────────────────── + +try: + import speech_recognition as sr + _SR_AVAILABLE = True +except ImportError: + _SR_AVAILABLE = False + +try: + from PIL import Image as _PilImage + import io as _io + _PIL_AVAILABLE = True +except ImportError: + _PIL_AVAILABLE = False + + +# ── Image preprocessing ─────────────────────────────────────────────────────── + +# llava internally resizes images to 336×336 anyway. +# Downscaling large images before sending saves encoding time and reduces +# the number of tokens the model spends on the image. +_MAX_IMAGE_DIM = 512 # pixels on longest side — fast, good quality + + +def _downscale_image_bytes(raw_bytes: bytes) -> bytes: + """ + Downscale image to _MAX_IMAGE_DIM on the longest side using PIL. + Returns original bytes if PIL is unavailable or image is already small. + """ + if not _PIL_AVAILABLE: + return raw_bytes + try: + img = _PilImage.open(_io.BytesIO(raw_bytes)) + w, h = img.size + if max(w, h) <= _MAX_IMAGE_DIM: + return raw_bytes # already small enough + scale = _MAX_IMAGE_DIM / max(w, h) + new_w = max(1, int(w * scale)) + new_h = max(1, int(h * scale)) + img = img.resize((new_w, new_h), _PilImage.LANCZOS) + # Convert to RGB (handles RGBA/P mode PNGs) + if img.mode not in ("RGB", "L"): + img = img.convert("RGB") + buf = _io.BytesIO() + img.save(buf, format="JPEG", quality=85) + return buf.getvalue() + except Exception: + return raw_bytes # fall back to original on any error + + +# ── Connectivity / runtime helpers ─────────────────────────────────────────── + +def _check_internet(host="8.8.8.8", port=53, timeout=2): + try: + socket.setdefaulttimeout(timeout) + sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + sock.connect((host, port)) + sock.close() + return True + except Exception: + return False + + +def get_stt_backend() -> str: + """Returns 'google' if SpeechRecognition is installed, else 'none'.""" + if _SR_AVAILABLE: + return "google" + return "none" + + +def is_ollama_running(): + try: + sock = socket.create_connection(("localhost", 11434), timeout=0.5) + sock.close() + return True + except Exception: + return False + + +def start_ollama(stop_flag=None): + """ + Start Ollama server if needed. + + Accepts an optional stop_flag callable so the caller can cancel startup. + The polling loop checks stop_flag() each second and exits early if cancelled. + """ + if os.name == 'nt': + subprocess.Popen('start cmd /k "ollama serve"', shell=True) + else: + subprocess.Popen( + ['bash', '-c', + 'x-terminal-emulator -e "ollama serve" || ' + 'gnome-terminal -- ollama serve || ' + 'xterm -e "ollama serve"'] + ) + + for _ in range(30): + if stop_flag is not None and stop_flag(): + return False + time.sleep(1) + if is_ollama_running(): + return True + return False + + +# ── Topic switch detection ─────────────────────────────────────────────────── + +_STOP_WORDS = { + 'a', 'an', 'the', 'is', 'in', 'it', 'of', 'to', 'do', 'i', + 'me', 'my', 'we', 'on', 'at', 'by', 'be', 'as', 'or', 'if', + 'how', 'what', 'why', 'can', 'for', 'with', 'this', 'that', + 'are', 'was', 'and', 'you', 'he', 'she', 'they', 'have', 'has', +} +_TOPIC_SWITCH_THRESHOLD = 0.15 + + +def _word_set(text: str) -> set: + words = re.findall(r'[a-z]+', text.lower()) + return {w for w in words if w not in _STOP_WORDS and len(w) > 2} + + +def detect_topic_switch(prev_text: str, curr_text: str) -> bool: + if not prev_text: + return False + a, b = _word_set(prev_text), _word_set(curr_text) + if not a or not b: + return False + return len(a & b) / len(a | b) < _TOPIC_SWITCH_THRESHOLD + + +# ── Model / service background workers ─────────────────────────────────────── + +class OllamaStatusWorker(QThread): + result_signal = pyqtSignal(bool) + + def run(self): + self.result_signal.emit(is_ollama_running()) + + +class ModelFetchWorker(QThread): + result_signal = pyqtSignal(list) + + def run(self): + try: + models_data = ollama.list() + raw = (models_data.get('models', []) + if isinstance(models_data, dict) + else getattr(models_data, 'models', [])) + + names = [] + for m in raw: + name = (m.get('name') or m.get('model', '') + if isinstance(m, dict) + else getattr(m, 'model', str(m))) + if name: + names.append(name) + + # Keep the vision model cache warm so image sends don't block + _refresh_model_cache() + + self.result_signal.emit(names if names else ['qwen2.5-coder:3b']) + except Exception: + self.result_signal.emit(['qwen2.5-coder:3b']) + + +# ── Smart token budget ─────────────────────────────────────────────────────── + +_COMPLEX_KEYWORDS = { + # netlist / SPICE + 'netlist', 'spice', 'ngspice', '.tran', '.ac', '.dc', '.model', + 'subcircuit', 'convergence', 'singular', 'timestep', + # debugging + 'error', 'debug', 'fix', 'wrong', 'issue', 'problem', 'fail', + 'simulate', 'simulation', 'analyse', 'analyze', + # circuit design + 'schematic', 'kicad', 'footprint', 'component', 'resistor', + 'capacitor', 'mosfet', 'transistor', 'opamp', 'voltage', 'current', +} + +_SIMPLE_KEYWORDS = { + 'what is', 'define', 'explain', 'meaning', 'how does', + 'difference between', 'example of', +} + + +def _smart_num_predict(user_messages: list, user_override: int = 1024) -> int: + """ + Choose a token budget based on message complexity rather than a flat cap. + This is the single biggest speed improvement: most answers need far fewer + tokens than the hard 1024 limit. + + Tiers (all well within correct/complete answer range for each type): + simple question → 256 tokens (~1-2 min on slow CPU, ~20s on fast) + technical → 512 tokens (~2-4 min on slow CPU, ~40s on fast) + netlist/debug → 768 tokens (~3-6 min on slow CPU, ~60s on fast) + + If the user has manually set a lower budget via the settings slider, + we always respect that as an upper bound. + + Combines up to 4 recent history lines (2 user turns + their bot replies) + for a better complexity signal than looking at user messages alone. + """ + # Combine the last 4 history lines (up to 2 user turns + their bot replies) + combined = " ".join( + line[5:].lower() for line in user_messages[-4:] + if line.startswith("User:") + ) + + # Check for complex technical content + is_complex = any(kw in combined for kw in _COMPLEX_KEYWORDS) + # Check for simple definitional questions + is_simple = any(kw in combined for kw in _SIMPLE_KEYWORDS) and not is_complex + # Long message = detailed question = needs detailed answer + is_long = len(combined) > 300 + + if is_simple and not is_long: + budget = 256 + elif is_complex or is_long: + budget = 768 + else: + budget = 384 + + # Respect the user's slider setting as a ceiling + return min(budget, user_override) + + +# ── System prompts ──────────────────────────────────────────────────────────── + +_SYSTEM_PROMPT = """You are an expert electronics engineer and the AI assistant embedded inside eSim, an open-source EDA tool developed by FOSSEE at IIT Bombay. + +Your expertise includes: +- KiCad schematic capture, symbols, labels, ERC issues, footprints +- NgSpice simulations and SPICE netlists +- Circuit debugging and simulation troubleshooting +- FPGA, MCU, power, reset, SPI, pull-up/pull-down, decoupling and connector review +- Reading partial schematic screenshots from EDA tools +- eSim workflow: KiCad → netlist → NgSpice → analysis + +Rules: +- Be practical, direct, and technically useful. +- Match response length to question complexity. +- For debugging, explain both WHY and HOW to fix the issue. +- When code or SPICE is needed, use a fenced code block. +- If the user provides a schematic image, analyze the visible block instead of giving a generic refusal. +- If uncertain, say likely / appears to be, but still provide analysis. +""" + + +_VISION_SYSTEM_PROMPT = """You are an expert electronics engineer and the AI assistant inside eSim, an open-source EDA tool by FOSSEE at IIT Bombay. + +You are given one or more schematic images from eSim or KiCad. Your job is to answer the user's question about those images as accurately and helpfully as possible. + +Rules: +- Read every visible label, net name, component reference, value, and pin number from the image. +- If the user asks a specific question (e.g. "how to build this in eSim", "what does this component do", "why is this connection wrong"), answer THAT question directly and completely. +- If no specific question is given, describe the circuit: identify its function, list components, and flag any design issues. +- Never refuse to analyse. If parts of the image are unclear, do your best and note any uncertainty. +- Be concise and practical. Match the length of your answer to the complexity of the question. +- When referring to components, use their visible reference designators (R1, C3, U2, etc.). +""" + + +# ── Text chat worker ────────────────────────────────────────────────────────── + +class OllamaWorker(QThread): + response_signal = pyqtSignal(str) + status_signal = pyqtSignal(str) + + def __init__(self, chat_history, model="qwen2.5-coder:3b", + temperature=0.25, num_predict=1024): + super().__init__() + self.chat_history = chat_history + self.model = model + self.temperature = temperature + self.num_predict = num_predict + self._stop_requested = False + + def stop(self): + self._stop_requested = True + + def run(self): + try: + if not is_ollama_running(): + self.status_signal.emit("Starting Ollama server — please wait…") + started = start_ollama(stop_flag=lambda: self._stop_requested) + if not started: + if self._stop_requested: + return # user cancelled cleanly + self.response_signal.emit( + "❌ Could not start Ollama automatically.\n" + "Please open a terminal and run: ollama serve" + ) + return + self.status_signal.emit("Ollama started! Getting response…") + time.sleep(1) + + # Keep last 10 history lines (5 turns). + # Sending 20 lines fills most of the context window before the + # question is even added, forcing the model to load more tokens. + messages = [{"role": "system", "content": _SYSTEM_PROMPT}] + for line in self.chat_history[-10:]: + if line.startswith("User:"): + messages.append({"role": "user", "content": line[5:].strip()}) + elif line.startswith("Bot:"): + messages.append({"role": "assistant", "content": line[4:].strip()}) + + # Smart token budget: short questions get fewer tokens so they + # finish faster; complex questions still get enough for a full answer. + budget = _smart_num_predict(self.chat_history, self.num_predict) + + stream = ollama.chat( + model=self.model, + messages=messages, + stream=True, + options={ + "temperature": self.temperature, + "num_predict": budget, + # 2048 ctx handles 5 turns of history + question comfortably. + # Allocating 4096 forces Ollama to malloc a larger KV-cache, + # adding 2-4s overhead before token 1 is generated. + "num_ctx": 2048, + "repeat_penalty": 1.08, + # Keep model loaded in RAM between requests. + # Without this, Ollama unloads after 5 min and the next + # question pays a 30-60s reload cost. + "keep_alive": "10m", + } + ) + + bot_response = "" + for chunk in stream: + if self._stop_requested: + bot_response += "\n\n⏹ Generation stopped." + break + bot_response += chunk["message"]["content"] + + bot_response = bot_response.strip() + if not bot_response: + bot_response = ( + "⚠️ Received an empty response. " + "The model may still be loading — please try again." + ) + + except Exception as e: + bot_response = ( + f"❌ Error: {str(e)}\n" + "Make sure Ollama is installed and 'ollama serve' is running." + ) + + self.response_signal.emit(bot_response) + + +# ── Vision model helpers ────────────────────────────────────────────────────── + +def _is_vision_model(model_name: str) -> bool: + if not model_name: + return False + m = model_name.lower() + return any(k in m for k in [ + "llava", "bakllava", "vision", "moondream", "minicpm-v", "qwen2-vl" + ]) +# QThread reads/writes don't produce a data race. +_cache_lock = threading.Lock() +_installed_models_cache: list = [] +_installed_models_cache_valid: bool = False + + +def _refresh_model_cache(): + global _installed_models_cache, _installed_models_cache_valid + try: + models_data = ollama.list() + raw = (models_data.get('models', []) + if isinstance(models_data, dict) + else getattr(models_data, 'models', [])) + names = [] + for m in raw: + name = (m.get('name') or m.get('model', '') + if isinstance(m, dict) + else getattr(m, 'model', str(m))) + if name: + names.append(name) + with _cache_lock: + _installed_models_cache = names + _installed_models_cache_valid = True + except Exception: + with _cache_lock: + _installed_models_cache_valid = False + + +def _pick_best_vision_model(preferred: str = "") -> str: + """ + Pick the fastest available vision model. + Priority: user-selected (if vision-capable) → moondream → llava:7b → llava → llava:13b + Smaller/faster models come FIRST so CPU inference is quick. + Uses a cached model list — no blocking ollama.list() call at send time. + """ + with _cache_lock: + cache_valid = _installed_models_cache_valid + cache_copy = list(_installed_models_cache) + + if not cache_valid: + _refresh_model_cache() + with _cache_lock: + cache_copy = list(_installed_models_cache) + + installed_map = {name.lower(): name for name in cache_copy} + + # If the user explicitly selected a vision model, respect that choice first + if preferred and _is_vision_model(preferred): + return preferred + + # Prefer smaller/faster models for speed on CPU + speed_order = [ + "moondream", # ~1.6 GB — fastest + "llava:7b", # ~4 GB — good balance + "llava", # ~4 GB — default tag (usually 7b) + "bakllava", # ~4 GB + "llava:13b", # ~8 GB — slowest, last resort + ] + for cand in speed_order: + if cand.lower() in installed_map: + return installed_map[cand.lower()] + + # Fallback: any installed vision model + for name in cache_copy: + if _is_vision_model(name): + return name + + # No vision-capable model is installed. + # Return None so the caller can show a clear error instead of + # sending the images to a text-only model that will hallucinate. + return None + + +def _build_schematic_vision_prompt(extra_prompt: str, image_count: int) -> str: + """ + Build the prompt sent to the vision model alongside the image(s). + If the user typed a question, that question drives the response. + If no question was given, request a general circuit analysis. + """ + n = "this schematic" if image_count == 1 else f"these {image_count} schematics" + if extra_prompt and extra_prompt.strip(): + # User asked something specific - make that the primary request. + return ( + f"Looking at {n}: {extra_prompt.strip()}\n\n" + "Base your answer on what is actually visible in the image." + ) + else: + # No question given - do a general analysis. + return ( + f"Please analyse {n}. " + "Identify the circuit's function, list all visible components with their " + "reference designators and values, name the nets and signal rails, " + "and flag any potential design issues you can see." + ) + + +# ── Vision worker ───────────────────────────────────────────────────────────── + +class OllamaVisionWorker(QThread): + response_signal = pyqtSignal(str) + status_signal = pyqtSignal(str) + + def __init__(self, image_paths=None, extra_prompt: str = "", + model: str = "llava", image_path: str = ""): + super().__init__() + + if image_paths: + self.image_paths = image_paths if isinstance(image_paths, list) else [image_paths] + elif image_path: + self.image_paths = [image_path] + else: + self.image_paths = [] + + self.extra_prompt = extra_prompt + self.model = model + self._stop_requested = False + + def stop(self): + self._stop_requested = True + + def _chat_once(self, model_name: str, prompt: str, image_bytes_list): + stream = ollama.chat( + model=model_name, + messages=[ + {"role": "system", "content": _VISION_SYSTEM_PROMPT}, + { + "role": "user", + "content": prompt, + "images": image_bytes_list, + } + ], + stream=True, + options={ + "temperature": 0.15, + # llava: ~576 tokens/image patch + ~200 prompt + 512 predict. + # 3072 gives comfortable headroom without the overhead of 4096. + "num_ctx": 3072, + "num_predict": 512, + "repeat_penalty": 1.05, + "keep_alive": "10m", + } + ) + + response = "" + token_count = 0 + for chunk in stream: + if self._stop_requested: + response += "\n\n\u23f9 Generation stopped." + break + piece = chunk["message"]["content"] + response += piece + token_count += 1 + + # Emit progress every 20 tokens so the status label + # shows the model is actively working + if token_count % 20 == 0: + self.status_signal.emit( + f"Generating… ({token_count} tokens so far)" + ) + return response.strip() + + def run(self): + try: + if not is_ollama_running(): + self.status_signal.emit("Starting Ollama server — please wait…") + started = start_ollama(stop_flag=lambda: self._stop_requested) + if not started: + if self._stop_requested: + return + self.response_signal.emit( + "❌ Could not start Ollama automatically.\n" + "Please open a terminal and run: ollama serve" + ) + return + self.status_signal.emit("Ollama started!") + time.sleep(1) + + if not self.image_paths: + self.response_signal.emit("❌ No image paths provided.") + return + + # Load and downscale images before sending. + # llava resizes internally to 336px anyway; sending 4K images + # just wastes encoding time. Downscaling to 512px is invisible + # to the model but saves significant transfer overhead. + image_bytes_list = [] + for path in self.image_paths: + if not os.path.exists(path): + self.response_signal.emit(f"❌ Image file not found: {path}") + return + with open(path, "rb") as f: + raw = f.read() + image_bytes_list.append(_downscale_image_bytes(raw)) + + image_count = len(image_bytes_list) + n_label = "image" if image_count == 1 else f"{image_count} images" + self.status_signal.emit(f"Analysing {n_label}…") + + vision_model = _pick_best_vision_model(self.model) + + if vision_model is None: + # No vision-capable model is installed. A text model cannot + # see images and will hallucinate plausible-sounding but + # completely fabricated answers. + self.response_signal.emit( + "❌ No vision model is installed.\n\n" + "Image analysis requires a vision-capable model. " + "Install one by running this in a terminal:\n" + "```bash\n" + "ollama pull llava\n" + "```\n" + "Then restart eSim and select `llava` from the model dropdown." + ) + return + + self.status_signal.emit(f"Using model: {vision_model}") + prompt = _build_schematic_vision_prompt(self.extra_prompt, image_count) + + response = self._chat_once(vision_model, prompt, image_bytes_list) + + if not response: + response = "⚠️ Vision model returned an empty response. Try again." + + except Exception as e: + err = str(e) + if "model" in err.lower() and ("not found" in err.lower() or "pull" in err.lower()): + response = ( + "❌ Vision model not found.\n" + "Install one with:\n" + "```bash\n" + "ollama pull llava\n" + "# or\n" + "ollama pull llava:7b\n" + "```" + ) + else: + response = ( + f"❌ Vision error: {err}\n" + "Make sure a vision model such as `llava` is installed." + ) + + self.response_signal.emit(response) + + +# ── Microphone worker ───────────────────────────────────────────────────────── + +class MicWorker(QThread): + text_signal = pyqtSignal(str) + error_signal = pyqtSignal(str) + status_signal = pyqtSignal(str) + + def run(self): + """Record from microphone and transcribe using Google Speech Recognition.""" + if not _SR_AVAILABLE: + self.error_signal.emit( + "SpeechRecognition not installed.\nRun: pip install SpeechRecognition pyaudio" + ) + return + + try: + recognizer = sr.Recognizer() + recognizer.energy_threshold = 300 + recognizer.dynamic_energy_threshold = True + + with sr.Microphone() as source: + self.status_signal.emit("🎤 Adjusting for ambient noise…") + recognizer.adjust_for_ambient_noise(source, duration=0.6) + self.status_signal.emit("🎤 Listening… speak now") + audio = recognizer.listen(source, timeout=8, phrase_time_limit=25) + + except sr.WaitTimeoutError: + self.error_signal.emit("🎤 No speech detected — please try again.") + return + except OSError: + self.error_signal.emit("🎤 Microphone not found. Run: pip install pyaudio") + return + except Exception as e: + self.error_signal.emit(f"🎤 Microphone error: {str(e)}") + return + + self._transcribe_google(audio) + + def _transcribe_google(self, audio): + try: + recognizer = sr.Recognizer() + self.status_signal.emit("🎤 Processing speech (online)…") + text = recognizer.recognize_google(audio) + if text: + self.text_signal.emit(text) + else: + self.error_signal.emit("🎤 Could not understand — please try again.") + except sr.UnknownValueError: + self.error_signal.emit("🎤 Could not understand speech — please try again.") + except sr.RequestError as e: + self.error_signal.emit( + f"🎤 Online STT failed: {e}\n" + "Install offline STT: pip install faster-whisper" + ) + except Exception as e: + self.error_signal.emit(f"🎤 STT error: {str(e)}") diff --git a/src/frontEnd/Application.py b/src/frontEnd/Application.py index 35aae135f..e551947c8 100644 --- a/src/frontEnd/Application.py +++ b/src/frontEnd/Application.py @@ -1,930 +1,835 @@ -# ========================================================================= -# FILE: Application.py -# -# USAGE: --- -# -# DESCRIPTION: This main file use to start the Application -# -# OPTIONS: --- -# REQUIREMENTS: --- -# BUGS: --- -# NOTES: --- -# AUTHOR: Fahim Khan, fahim.elex@gmail.com -# MAINTAINED: Rahul Paknikar, rahulp@iitb.ac.in -# Sumanto Kar, sumantokar@iitb.ac.in -# Pranav P, pranavsdreams@gmail.com -# ORGANIZATION: eSim Team at FOSSEE, IIT Bombay -# CREATED: Tuesday 24 February 2015 -# REVISION: Wednesday 07 June 2023 -# ========================================================================= - -import os -import sys -import traceback -import webbrowser - -if os.name == 'nt': - from frontEnd import pathmagic # noqa:F401 - init_path = '' -else: - import pathmagic # noqa:F401 - init_path = '../../' - -from PyQt5 import QtGui, QtCore, QtWidgets -from PyQt5.Qt import QSize -from configuration.Appconfig import Appconfig -from frontEnd import ProjectExplorer -from frontEnd import Workspace -from frontEnd import DockArea -from projManagement.openProject import OpenProjectInfo -from projManagement.newProject import NewProjectInfo -from projManagement.Kicad import Kicad -from projManagement.Validation import Validation -from projManagement import Worker - -# Its our main window of application. - - -class Application(QtWidgets.QMainWindow): - """This class initializes all objects used in this file.""" - global project_name - simulationEndSignal = QtCore.pyqtSignal(QtCore.QProcess.ExitStatus, int) - - def __init__(self, *args): - """Initialize main Application window.""" - - # Calling __init__ of super class - QtWidgets.QMainWindow.__init__(self, *args) - - # Set slot for simulation end signal to plot simulation data - self.simulationEndSignal.connect(self.plotSimulationData) - - #the plotFlag - self.plotFlag = False - - # Creating require Object - self.obj_workspace = Workspace.Workspace() - self.obj_Mainview = MainView() - self.obj_kicad = Kicad(self.obj_Mainview.obj_dockarea) - self.obj_appconfig = Appconfig() - self.obj_validation = Validation() - # Initialize all widget - self.setCentralWidget(self.obj_Mainview) - self.initToolBar() - - self.setGeometry(self.obj_appconfig._app_xpos, - self.obj_appconfig._app_ypos, - self.obj_appconfig._app_width, - self.obj_appconfig._app_heigth) - self.setWindowTitle( - self.obj_appconfig._APPLICATION + "-" + self.obj_appconfig._VERSION - ) - self.showMaximized() - self.setWindowIcon(QtGui.QIcon(init_path + 'images/logo.png')) - - self.systemTrayIcon = QtWidgets.QSystemTrayIcon(self) - self.systemTrayIcon.setIcon(QtGui.QIcon(init_path + 'images/logo.png')) - self.systemTrayIcon.setVisible(True) - - def initToolBar(self): - """ - This function initializes Tool Bars. - It setups the icons, short-cuts and defining functonality for: - - - Top-tool-bar (New project, Open project, Close project, \ - Mode switch, Help option) - - Left-tool-bar (Open Schematic, Convert KiCad to Ngspice, \ - Simuation, Model Editor, Subcircuit, NGHDL, Modelica \ - Converter, OM Optimisation) - """ - # Top Tool bar - self.newproj = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/newProject.png'), - 'New Project', self - ) - self.newproj.setShortcut('Ctrl+N') - self.newproj.triggered.connect(self.new_project) - - self.openproj = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/openProject.png'), - 'Open Project', self - ) - self.openproj.setShortcut('Ctrl+O') - self.openproj.triggered.connect(self.open_project) - - self.closeproj = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/closeProject.png'), - 'Close Project', self - ) - self.closeproj.setShortcut('Ctrl+X') - self.closeproj.triggered.connect(self.close_project) - - self.wrkspce = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/workspace.ico'), - 'Change Workspace', self - ) - self.wrkspce.setShortcut('Ctrl+W') - self.wrkspce.triggered.connect(self.change_workspace) - - self.helpfile = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/helpProject.png'), - 'Help', self - ) - self.helpfile.setShortcut('Ctrl+H') - self.helpfile.triggered.connect(self.help_project) - - # added devDocs logo and called functions - self.devdocs = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/dev_docs.png'), - 'Dev Docs', self - ) - self.devdocs.setShortcut('Ctrl+D') - self.devdocs.triggered.connect(self.dev_docs) - - self.topToolbar = self.addToolBar('Top Tool Bar') - self.topToolbar.addAction(self.newproj) - self.topToolbar.addAction(self.openproj) - self.topToolbar.addAction(self.closeproj) - self.topToolbar.addAction(self.wrkspce) - self.topToolbar.addAction(self.helpfile) - self.topToolbar.addAction(self.devdocs) - - # ## This part is meant for SoC Generation which is currently ## - # ## under development and will be will be required in future. ## - # self.soc = QtWidgets.QToolButton(self) - # self.soc.setText('Generate SoC') - # self.soc.setToolTip( - # 'SPICE to Verilog Conversion
' + \ - # '
The feature is under development.' + \ - # '
It will be released soon.' + \ - # '

Thank you for your patience!!!' - # ) - # self.soc.setStyleSheet(" \ - # QWidget { border-radius: 15px; border: 1px \ - # solid gray; padding: 10px; margin-left: 20px; } \ - # ") - # self.soc.clicked.connect(self.showSoCRelease) - # self.topToolbar.addWidget(self.soc) - - # This part is setting fossee logo to the right - # corner in the application window. - self.spacer = QtWidgets.QWidget() - self.spacer.setSizePolicy( - QtWidgets.QSizePolicy.Expanding, - QtWidgets.QSizePolicy.Expanding) - self.topToolbar.addWidget(self.spacer) - self.logo = QtWidgets.QLabel() - self.logopic = QtGui.QPixmap( - os.path.join( - os.path.abspath(''), init_path + 'images', 'fosseeLogo.png' - )) - self.logopic = self.logopic.scaled( - QSize(150, 150), QtCore.Qt.KeepAspectRatio) - self.logo.setPixmap(self.logopic) - self.logo.setStyleSheet("padding:0 15px 0 0;") - self.topToolbar.addWidget(self.logo) - - # Left Tool bar Action Widget - self.kicad = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/kicad.png'), - 'Open Schematic', self - ) - self.kicad.triggered.connect(self.obj_kicad.openSchematic) - - self.conversion = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/ki-ng.png'), - 'Convert KiCad to Ngspice', self - ) - self.conversion.triggered.connect(self.obj_kicad.openKicadToNgspice) - - self.ngspice = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/ngspice.png'), - 'Simulate', self - ) - self.ngspice.triggered.connect(self.plotFlagPopBox) - - self.model = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/model.png'), - 'Model Editor', self - ) - self.model.triggered.connect(self.open_modelEditor) - - self.subcircuit = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/subckt.png'), - 'Subcircuit', self - ) - self.subcircuit.triggered.connect(self.open_subcircuit) - - self.nghdl = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/nghdl.png'), 'NGHDL', self - ) - self.nghdl.triggered.connect(self.open_nghdl) - - self.makerchip = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/makerchip.png'), - 'Makerchip-NgVeri', self - ) - self.makerchip.triggered.connect(self.open_makerchip) - - self.omedit = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/omedit.png'), - 'Modelica Converter', self - ) - self.omedit.triggered.connect(self.open_OMedit) - - self.omoptim = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/omoptim.png'), - 'OM Optimisation', self - ) - self.omoptim.triggered.connect(self.open_OMoptim) - - self.conToeSim = QtWidgets.QAction( - QtGui.QIcon(init_path + 'images/icon.png'), - 'Schematic converter', self - ) - self.conToeSim.triggered.connect(self.open_conToeSim) - - # Adding Action Widget to tool bar - self.lefttoolbar = QtWidgets.QToolBar('Left ToolBar') - self.addToolBar(QtCore.Qt.LeftToolBarArea, self.lefttoolbar) - self.lefttoolbar.addAction(self.kicad) - self.lefttoolbar.addAction(self.conversion) - self.lefttoolbar.addAction(self.ngspice) - self.lefttoolbar.addAction(self.model) - self.lefttoolbar.addAction(self.subcircuit) - self.lefttoolbar.addAction(self.makerchip) - self.lefttoolbar.addAction(self.nghdl) - self.lefttoolbar.addAction(self.omedit) - self.lefttoolbar.addAction(self.omoptim) - self.lefttoolbar.addAction(self.conToeSim) - self.lefttoolbar.setOrientation(QtCore.Qt.Vertical) - self.lefttoolbar.setIconSize(QSize(40, 40)) - - def plotFlagPopBox(self): - """This function displays a pop-up box with message- Do you want Ngspice plots? and oprions Yes and NO. - - If the user clicks on Yes, both the NgSpice and python plots are displayed and if No is clicked then only the python plots.""" - - msg_box = QtWidgets.QMessageBox(self) - msg_box.setWindowTitle("Ngspice Plots") - msg_box.setText("Do you want Ngspice plots?") - - yes_button = msg_box.addButton("Yes", QtWidgets.QMessageBox.YesRole) - no_button = msg_box.addButton("No", QtWidgets.QMessageBox.NoRole) - - msg_box.exec_() - - if msg_box.clickedButton() == yes_button: - self.plotFlag = True - else: - self.plotFlag = False - - self.open_ngspice() - - def closeEvent(self, event): - ''' - This function closes the ongoing program (process). - When exit button is pressed a Message box pops out with \ - exit message and buttons 'Yes', 'No'. - - 1. If 'Yes' is pressed: - - check that program (process) in procThread_list \ - (a list made in Appconfig.py): - - - if available it terminates that program. - - if the program (process) is not available, \ - then check it in process_obj (a list made in \ - Appconfig.py) and if found, it closes the program. - - 2. If 'No' is pressed: - - the program just continues as it was doing earlier. - ''' - exit_msg = "Are you sure you want to exit the program?" - exit_msg += " All unsaved data will be lost." - reply = QtWidgets.QMessageBox.question( - self, 'Message', exit_msg, QtWidgets.QMessageBox.Yes, - QtWidgets.QMessageBox.No - ) - - if reply == QtWidgets.QMessageBox.Yes: - for proc in self.obj_appconfig.procThread_list: - try: - proc.terminate() - except BaseException: - pass - try: - for process_object in self.obj_appconfig.process_obj: - try: - process_object.close() - except BaseException: - pass - except BaseException: - pass - - # Check if "Open project" and "New project" window is open. - # If yes, just close it when application is closed. - try: - self.project.close() - except BaseException: - pass - event.accept() - self.systemTrayIcon.showMessage('Exit', 'eSim is Closed.') - - elif reply == QtWidgets.QMessageBox.No: - event.ignore() - - def new_project(self): - """This function call New Project Info class.""" - text, ok = QtWidgets.QInputDialog.getText( - self, 'New Project Info', 'Enter Project Name:' - ) - updated = False - - if ok: - self.projname = (str(text)) - self.project = NewProjectInfo() - directory, filelist = self.project.createProject(self.projname) - - if directory and filelist: - self.obj_Mainview.obj_projectExplorer.addTreeNode( - directory, filelist - ) - updated = True - - if not updated: - print("No new project created") - self.obj_appconfig.print_info('No new project created') - try: - self.obj_appconfig.print_info( - 'Current project is : ' + - self.obj_appconfig.current_project["ProjectName"] - ) - except BaseException: - pass - - def open_project(self): - """This project call Open Project Info class.""" - print("Function : Open Project") - self.project = OpenProjectInfo() - try: - directory, filelist = self.project.body() - self.obj_Mainview.obj_projectExplorer.addTreeNode( - directory, filelist) - except BaseException: - pass - - def close_project(self): - """ - This function closes the saved project. - It first checks whether project (file) is present in list. - - - If present: - - it first kills that process-id. - - closes that file. - - Shows message "Current project is closed" - - - If not present: pass - """ - print("Function : Close Project") - current_project = self.obj_appconfig.current_project['ProjectName'] - if current_project is None: - pass - else: - temp = self.obj_appconfig.current_project['ProjectName'] - for pid in self.obj_appconfig.proc_dict[temp]: - try: - os.kill(pid, 9) - except BaseException: - pass - self.obj_Mainview.obj_dockarea.closeDock() - self.obj_appconfig.current_project['ProjectName'] = None - self.systemTrayIcon.showMessage( - 'Close', 'Current project ' + - os.path.basename(current_project) + ' is Closed.' - ) - - def change_workspace(self): - """ - This function call changes Workspace - """ - print("Function : Change Workspace") - self.obj_workspace.returnWhetherClickedOrNot(self) - self.hide() - self.obj_workspace.show() - - def help_project(self): - """ - This function opens usermanual in dockarea. - - It prints the message ""Function : Help"" - - Uses print_info() method of class Appconfig - from Configuration/Appconfig.py file. - - Call method usermanual() from ./DockArea.py. - """ - print("Function : Help") - self.obj_appconfig.print_info('Help is called') - print("Current Project is : ", self.obj_appconfig.current_project) - self.obj_Mainview.obj_dockarea.usermanual() - - def dev_docs(self): - """ - This function guides the user to readthedocs website for the developer docs - """ - print("Function : DevDocs") - self.obj_appconfig.print_info('DevDocs is called') - print("Current Project is : ", self.obj_appconfig.current_project) - webbrowser.open("https://esim.readthedocs.io/en/latest/index.html") - - @QtCore.pyqtSlot(QtCore.QProcess.ExitStatus, int) - def plotSimulationData(self, exitCode, exitStatus): - """Enables interaction for new simulation and - displays the plotter dock where graphs can be plotted. - """ - self.ngspice.setEnabled(True) - self.conversion.setEnabled(True) - self.closeproj.setEnabled(True) - self.wrkspce.setEnabled(True) - - if exitStatus == QtCore.QProcess.NormalExit and exitCode == 0: - try: - self.obj_Mainview.obj_dockarea.plottingEditor() - except Exception as e: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle("Error Message") - self.msg.showMessage( - 'Data could not be plotted. Please try again.' - ) - self.msg.exec_() - print("Exception Message:", str(e), traceback.format_exc()) - self.obj_appconfig.print_error('Exception Message : ' - + str(e)) - - def open_ngspice(self): - """This Function execute ngspice on current project.""" - projDir = self.obj_appconfig.current_project["ProjectName"] - - if projDir is not None: - projName = os.path.basename(projDir) - ngspiceNetlist = os.path.join(projDir, projName + ".cir.out") - - if not os.path.isfile(ngspiceNetlist): - print( - "Netlist file (*.cir.out) not found." - ) - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle("Error Message") - self.msg.showMessage( - 'Netlist (*.cir.out) not found.' - ) - self.msg.exec_() - return - - self.obj_Mainview.obj_dockarea.ngspiceEditor( - projName, ngspiceNetlist, self.simulationEndSignal, self.plotFlag) - - self.ngspice.setEnabled(False) - self.conversion.setEnabled(False) - self.closeproj.setEnabled(False) - self.wrkspce.setEnabled(False) - - else: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle("Error Message") - self.msg.showMessage( - 'Please select the project first.' - ' You can either create new project or open existing project' - ) - self.msg.exec_() - - def open_subcircuit(self): - """ - This function opens 'subcircuit' option in left-tool-bar. - When 'subcircuit' icon is clicked wich is present in - left-tool-bar of main page: - - - Meassge shown on screen "Subcircuit editor is called". - - 'subcircuiteditor()' function is called using object - 'obj_dockarea' of class 'Mainview'. - """ - print("Function : Subcircuit editor") - self.obj_appconfig.print_info('Subcircuit editor is called') - self.obj_Mainview.obj_dockarea.subcircuiteditor() - - def open_nghdl(self): - """ - This function calls NGHDL option in left-tool-bar. - It uses validateTool() method from Validation.py: - - - If 'nghdl' is present in executables list then - it passes command 'nghdl -e' to WorkerThread class of - Worker.py. - - If 'nghdl' is not present, then it shows error message. - """ - print("Function : NGHDL") - self.obj_appconfig.print_info('NGHDL is called') - - if self.obj_validation.validateTool('nghdl'): - self.cmd = 'nghdl -e' - self.obj_workThread = Worker.WorkerThread(self.cmd) - self.obj_workThread.start() - else: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle('NGHDL Error') - self.msg.showMessage('Error while opening NGHDL. ' + - 'Please make sure it is installed') - self.obj_appconfig.print_error('Error while opening NGHDL. ' + - 'Please make sure it is installed') - self.msg.exec_() - - def open_makerchip(self): - """ - This function opens 'subcircuit' option in left-tool-bar. - When 'subcircuit' icon is clicked wich is present in - left-tool-bar of main page: - - - Meassge shown on screen "Subcircuit editor is called". - - 'subcircuiteditor()' function is called using object - 'obj_dockarea' of class 'Mainview'. - """ - print("Function : Makerchip and Verilator to Ngspice Converter") - self.obj_appconfig.print_info('Makerchip is called') - self.obj_Mainview.obj_dockarea.makerchip() - - def open_modelEditor(self): - """ - This function opens model editor option in left-tool-bar. - When model editor icon is clicked which is present in - left-tool-bar of main page: - - - Meassge shown on screen "Model editor is called". - - 'modeleditor()' function is called using object - 'obj_dockarea' of class 'Mainview'. - """ - print("Function : Model editor") - self.obj_appconfig.print_info('Model editor is called') - self.obj_Mainview.obj_dockarea.modelEditor() - - def open_OMedit(self): - """ - This function calls ngspice to OMEdit converter and then launch OMEdit. - """ - self.obj_appconfig.print_info('OMEdit is called') - self.projDir = self.obj_appconfig.current_project["ProjectName"] - - if self.projDir is not None: - if self.obj_validation.validateCirOut(self.projDir): - self.projName = os.path.basename(self.projDir) - self.ngspiceNetlist = os.path.join( - self.projDir, self.projName + ".cir.out" - ) - self.modelicaNetlist = os.path.join( - self.projDir, self.projName + ".mo" - ) - - """ - try: - # Creating a command for Ngspice to Modelica converter - self.cmd1 = " - python3 ../ngspicetoModelica/NgspicetoModelica.py "\ - + self.ngspiceNetlist - self.obj_workThread1 = Worker.WorkerThread(self.cmd1) - self.obj_workThread1.start() - if self.obj_validation.validateTool("OMEdit"): - # Creating command to run OMEdit - self.cmd2 = "OMEdit "+self.modelicaNetlist - self.obj_workThread2 = Worker.WorkerThread(self.cmd2) - self.obj_workThread2.start() - else: - self.msg = QtWidgets.QMessageBox() - self.msgContent = "There was an error while - opening OMEdit.
\ - Please make sure OpenModelica is installed in your\ - system.
\ - To install it on Linux : Go to\ - OpenModelica Linux and \ - install nigthly build release.
\ - To install it on Windows : Go to\ - OpenModelica Windows\ - and install latest version.
" - self.msg.setTextFormat(QtCore.Qt.RichText) - self.msg.setText(self.msgContent) - self.msg.setWindowTitle("Missing OpenModelica") - self.obj_appconfig.print_info(self.msgContent) - self.msg.exec_() - - except Exception as e: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle( - "Ngspice to Modelica conversion error") - self.msg.showMessage( - 'Unable to convert NgSpice netlist to\ - Modelica netlist :'+str(e)) - self.msg.exec_() - self.obj_appconfig.print_error(str(e)) - """ - - self.obj_Mainview.obj_dockarea.modelicaEditor(self.projDir) - - else: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle("Missing Ngspice Netlist") - self.msg.showMessage( - 'Current project does not contain any Ngspice file. ' + - 'Please create Ngspice file with extension .cir.out' - ) - self.msg.exec_() - else: - self.msg = QtWidgets.QErrorMessage() - self.msg.setModal(True) - self.msg.setWindowTitle("Error Message") - self.msg.showMessage( - 'Please select the project first. You can either ' + - 'create a new project or open an existing project' - ) - self.msg.exec_() - - def open_OMoptim(self): - """ - This function uses validateTool() method from Validation.py: - - - If 'OMOptim' is present in executables list then - it passes command 'OMOptim' to WorkerThread class of Worker.py - - If 'OMOptim' is not present, then it shows error message with - link to download it on Linux and Windows. - """ - print("Function : OMOptim") - self.obj_appconfig.print_info('OMOptim is called') - # Check if OMOptim is installed - if self.obj_validation.validateTool("OMOptim"): - # Creating a command to run - self.cmd = "OMOptim" - self.obj_workThread = Worker.WorkerThread(self.cmd) - self.obj_workThread.start() - else: - self.msg = QtWidgets.QMessageBox() - self.msgContent = ( - "There was an error while opening OMOptim.
" - "Please make sure OpenModelica is installed in your" - " system.
" - "To install it on Linux : Go to OpenModelica Linux and install nightly build" - " release.
" - "To install it on Windows : Go to OpenModelica Windows and install latest version.
" - ) - self.msg.setTextFormat(QtCore.Qt.RichText) - self.msg.setText(self.msgContent) - self.msg.setWindowTitle("Error Message") - self.obj_appconfig.print_info(self.msgContent) - self.msg.exec_() - - def open_conToeSim(self): - print("Function : Schematic converter") - self.obj_appconfig.print_info('Schematic converter is called') - self.obj_Mainview.obj_dockarea.eSimConverter() - -# This class initialize the Main View of Application -class MainView(QtWidgets.QWidget): - """ - This class defines whole view and style of main page: - - - Position of tool bars: - - Top tool bar. - - Left tool bar. - - Project explorer Area. - - Dock area. - - Console area. - """ - - def __init__(self, *args): - # call init method of superclass - QtWidgets.QWidget.__init__(self, *args) - - self.obj_appconfig = Appconfig() - - self.leftSplit = QtWidgets.QSplitter() - self.middleSplit = QtWidgets.QSplitter() - - self.mainLayout = QtWidgets.QVBoxLayout() - # Intermediate Widget - self.middleContainer = QtWidgets.QWidget() - self.middleContainerLayout = QtWidgets.QVBoxLayout() - - # Area to be included in MainView - self.noteArea = QtWidgets.QTextEdit() - self.noteArea.setReadOnly(True) - - # Set explicit scrollbar policy - self.noteArea.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded) - self.noteArea.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded) - - self.obj_appconfig.noteArea['Note'] = self.noteArea - self.obj_appconfig.noteArea['Note'].append( - ' eSim Started......') - self.obj_appconfig.noteArea['Note'].append('Project Selected : None') - self.obj_appconfig.noteArea['Note'].append('\n') - - # Enhanced CSS with proper scrollbar styling - self.noteArea.setStyleSheet(""" - QTextEdit { - border-radius: 15px; - border: 1px solid gray; - padding: 5px; - background-color: white; - } - - QScrollBar:vertical { - border: 1px solid #999999; - background: #f0f0f0; - width: 16px; - margin: 16px 0 16px 0; - border-radius: 3px; - } - - QScrollBar::handle:vertical { - background: #606060; - min-height: 20px; - border-radius: 3px; - margin: 1px; - } - - QScrollBar::handle:vertical:hover { - background: #505050; - } - - QScrollBar::add-line:vertical { - border: 1px solid #999999; - background: #d0d0d0; - height: 15px; - width: 16px; - subcontrol-position: bottom; - subcontrol-origin: margin; - border-radius: 2px; - } - - QScrollBar::sub-line:vertical { - border: 1px solid #999999; - background: #d0d0d0; - height: 15px; - width: 16px; - subcontrol-position: top; - subcontrol-origin: margin; - border-radius: 2px; - } - - QScrollBar::add-line:vertical:hover, - QScrollBar::sub-line:vertical:hover { - background: #c0c0c0; - } - - QScrollBar::add-page:vertical, - QScrollBar::sub-page:vertical { - background: none; - } - - QScrollBar::up-arrow:vertical { - width: 8px; - height: 8px; - background-color: #606060; - } - - QScrollBar::down-arrow:vertical { - width: 8px; - height: 8px; - background-color: #606060; - } - - QScrollBar:horizontal { - border: 1px solid #999999; - background: #f0f0f0; - height: 16px; - margin: 0 16px 0 16px; - border-radius: 3px; - } - - QScrollBar::handle:horizontal { - background: #606060; - min-width: 20px; - border-radius: 3px; - margin: 1px; - } - - QScrollBar::handle:horizontal:hover { - background: #505050; - } - - QScrollBar::add-line:horizontal, - QScrollBar::sub-line:horizontal { - border: 1px solid #999999; - background: #d0d0d0; - width: 15px; - height: 16px; - border-radius: 2px; - } - - QScrollBar::add-line:horizontal:hover, - QScrollBar::sub-line:horizontal:hover { - background: #c0c0c0; - } - """) - - self.obj_dockarea = DockArea.DockArea() - self.obj_projectExplorer = ProjectExplorer.ProjectExplorer() - - # Adding content to vertical middle Split. - self.middleSplit.setOrientation(QtCore.Qt.Vertical) - self.middleSplit.addWidget(self.obj_dockarea) - self.middleSplit.addWidget(self.noteArea) - - # Adding middle split to Middle Container Widget - self.middleContainerLayout.addWidget(self.middleSplit) - self.middleContainer.setLayout(self.middleContainerLayout) - - # Adding content of left split - self.leftSplit.addWidget(self.obj_projectExplorer) - self.leftSplit.addWidget(self.middleContainer) - - # Adding to main Layout - self.mainLayout.addWidget(self.leftSplit) - self.leftSplit.setSizes([int(self.width() / 4.5), self.height()]) - self.middleSplit.setSizes([self.width(), int(self.height() / 2)]) - self.setLayout(self.mainLayout) - - def collapse_console_area(self): - """Collapse the console area to minimal height.""" - current_sizes = self.middleSplit.sizes() - total_height = sum(current_sizes) - minimal_console_height = 0 - dock_area_height = total_height - minimal_console_height - self.middleSplit.setSizes([dock_area_height, minimal_console_height]) - - def restore_console_area(self): - """Restore the console area to normal height.""" - total_height = sum(self.middleSplit.sizes()) - dock_area_height = int(total_height * 0.7) # 70% for dock area - console_height = total_height - dock_area_height # 30% for console - self.middleSplit.setSizes([dock_area_height, console_height]) - - -# It is main function of the module and starts the application -def main(args): - """ - The splash screen opened at the starting of screen is performed - by this function. - """ - print("Starting eSim......") - # Set non-native dialogs globally - QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_DontUseNativeDialogs, True) - app = QtWidgets.QApplication(args) - app.setApplicationName("eSim") - - appView = Application() - appView.hide() - - splash_pix = QtGui.QPixmap(init_path + 'images/splash_screen_esim.png') - splash = QtWidgets.QSplashScreen( - appView, splash_pix, QtCore.Qt.WindowStaysOnTopHint - ) - splash.setMask(splash_pix.mask()) - splash.setDisabled(True) - splash.show() - - appView.splash = splash - appView.obj_workspace.returnWhetherClickedOrNot(appView) - - try: - if os.name == 'nt': - user_home = os.path.join('library', 'config') - else: - user_home = os.path.expanduser('~') - - file = open(os.path.join(user_home, ".esim/workspace.txt"), 'r') - work = int(file.read(1)) - file.close() - except IOError: - work = 0 - - if work != 0: - appView.obj_workspace.defaultWorkspace() - else: - appView.obj_workspace.show() - - sys.exit(app.exec_()) - - -# Call main function -if __name__ == '__main__': - # Create and display the splash screen - try: - main(sys.argv) - except Exception as err: - print("Error: ", err) +# ========================================================================= +# FILE: Application.py +# +# USAGE: --- +# +# DESCRIPTION: This main file use to start the Application +# +# OPTIONS: --- +# REQUIREMENTS: --- +# BUGS: --- +# NOTES: --- +# AUTHOR: Fahim Khan, fahim.elex@gmail.com +# MAINTAINED: Rahul Paknikar, rahulp@iitb.ac.in +# Sumanto Kar, sumantokar@iitb.ac.in +# Pranav P, pranavsdreams@gmail.com +# ORGANIZATION: eSim Team at FOSSEE, IIT Bombay +# CREATED: Tuesday 24 February 2015 +# REVISION: Wednesday 07 June 2023 +# ========================================================================= + +import os +import sys +import traceback + +if os.name == 'nt': + from frontEnd import pathmagic # noqa:F401 + init_path = '' +else: + import pathmagic # noqa:F401 + init_path = '../../' + +from PyQt5 import QtGui, QtCore, QtWidgets +from PyQt5.Qt import QSize +from configuration.Appconfig import Appconfig +from frontEnd import ProjectExplorer +from frontEnd import Workspace +from frontEnd import DockArea +from projManagement.openProject import OpenProjectInfo +from projManagement.newProject import NewProjectInfo +from projManagement.Kicad import Kicad +from projManagement.Validation import Validation +from projManagement import Worker +from frontEnd.Chatbot import ChatbotGUI +from PyQt5.QtCore import QTimer +# Its our main window of application. + + +class Application(QtWidgets.QMainWindow): + """This class initializes all objects used in this file.""" + + # FIX #12: Removed the `global project_name` declaration that was here. + # A `global` statement at class scope has no effect in Python — class bodies + # do not create a new scope the way functions do — and the variable was + # never assigned or read anywhere in the class, making it pure dead code. + + simulationEndSignal = QtCore.pyqtSignal(QtCore.QProcess.ExitStatus, int) + errorDetectedSignal = QtCore.pyqtSignal(str) + + def __init__(self, *args): + """Initialize main Application window.""" + + # Calling __init__ of super class + QtWidgets.QMainWindow.__init__(self, *args) + + # Set slot for simulation end signal to plot simulation data + self.simulationEndSignal.connect(self.plotSimulationData) + self.errorDetectedSignal.connect(self.handleError) + # Creating require Object + self.obj_workspace = Workspace.Workspace() + self.obj_Mainview = MainView() + self.obj_kicad = Kicad(self.obj_Mainview.obj_dockarea) + self.obj_appconfig = Appconfig() + self.obj_validation = Validation() + self.chatbot_window = ChatbotGUI() + # Initialize all widget + self.setCentralWidget(self.obj_Mainview) + self.initToolBar() + self.initchatbot() + + self.setGeometry(self.obj_appconfig._app_xpos, + self.obj_appconfig._app_ypos, + self.obj_appconfig._app_width, + self.obj_appconfig._app_heigth) + self.setWindowTitle( + self.obj_appconfig._APPLICATION + "-" + self.obj_appconfig._VERSION + ) + self.showMaximized() + self.setWindowIcon(QtGui.QIcon(init_path + 'images/logo.png')) + + self.systemTrayIcon = QtWidgets.QSystemTrayIcon(self) + self.systemTrayIcon.setIcon(QtGui.QIcon(init_path + 'images/logo.png')) + self.systemTrayIcon.setVisible(True) + + def initchatbot(self): + """ + This function initializes the ChatbotIcon and embeds the ChatbotGUI + as a dockable panel on the right side of the main window. + Clicking the icon toggles the panel open/closed. + """ + # ── Dock widget (embedded in main window) ────────────────────── + self.chatbot_dock = QtWidgets.QDockWidget("🤖 eSim AI Assistant", self) + self.chatbot_dock.setWidget(self.chatbot_window) + self.chatbot_dock.setMinimumWidth(360) + self.chatbot_dock.setFeatures( + QtWidgets.QDockWidget.DockWidgetClosable | + QtWidgets.QDockWidget.DockWidgetMovable | + QtWidgets.QDockWidget.DockWidgetFloatable + ) + # Style the dock title bar + self.chatbot_dock.setStyleSheet(""" + QDockWidget { + font-weight: bold; + font-size: 13px; + color: #0055a5; + } + QDockWidget::title { + background: qlineargradient( + x1:0, y1:0, x2:1, y2:0, + stop:0 #e8f4fd, stop:1 #d0e8f8 + ); + padding: 6px 10px; + border-bottom: 2px solid #0078d4; + border-top-left-radius: 6px; + border-top-right-radius: 6px; + } + """) + self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.chatbot_dock) + self.chatbot_dock.hide() # Hidden by default; toggled by the icon button + # When user closes dock via the X button, reposition the floating icon + self.chatbot_dock.visibilityChanged.connect( + lambda _: self._reposition_chatbot_icon() + ) + + # ── Floating icon button (bottom-right corner) ────────────────── + self.chatboticon = QtWidgets.QPushButton( + self, icon=QtGui.QIcon(init_path + 'images/chatbot.png') + ) + self.chatboticon.setIconSize(QtCore.QSize(30, 30)) + self.chatboticon.setToolTip("Toggle AI Assistant") + self.chatboticon.setStyleSheet(""" + QPushButton { + border-radius: 26px; + background-color: transparent; + border: none; + } + QPushButton:hover { + background-color: rgba(0, 120, 212, 0.12); + border: 2px solid #0078d4; + } + QPushButton:pressed { + background-color: rgba(0, 63, 110, 0.18); + border: 2px solid #003f6e; + } + """) + self.chatboticon.setFixedSize(52, 52) + self.chatboticon.clicked.connect(self.openChatbot) + + def openChatbot(self): + """Toggle the chatbot dock panel open or closed.""" + if self.chatbot_dock.isVisible(): + self.chatbot_dock.hide() + else: + self.chatbot_dock.show() + self.chatbot_dock.raise_() + self.obj_appconfig.print_info('Chat Bot function is called') + # Reposition icon after dock visibility changes + self._reposition_chatbot_icon() + + def _reposition_chatbot_icon(self): + """ + Keep the icon in the bottom-right corner of the visible area. + When the dock is open, shift the icon left so it sits just + outside the dock panel instead of on top of it. + + FIX #13: Added a max(0, ...) guard so that if the dock is very wide + (e.g. undocked and then re-docked at a large size), the computed x + coordinate cannot go negative and push the button off-screen. + """ + margin = 12 + btn_w = self.chatboticon.width() + btn_h = self.chatboticon.height() + bottom_y = self.height() - btn_h - margin + + if self.chatbot_dock.isVisible(): + dock_w = self.chatbot_dock.width() + # FIX #13: Clamp x to 0 so the icon never goes off-screen + x = max(0, self.width() - dock_w - btn_w - margin) + else: + x = self.width() - btn_w - margin + + self.chatboticon.move(x, bottom_y) + self.chatboticon.raise_() # Always keep on top + + def resizeEvent(self, event): + """ + Adjust chatbot icon button position during window resize. + """ + super().resizeEvent(event) + self._reposition_chatbot_icon() + + def initToolBar(self): + """ + This function initializes Tool Bars. + It setups the icons, short-cuts and defining functonality for: + + - Top-tool-bar (New project, Open project, Close project, \ + Mode switch, Help option) + - Left-tool-bar (Open Schematic, Convert KiCad to Ngspice, \ + Simuation, Model Editor, Subcircuit, NGHDL, Modelica \ + Converter, OM Optimisation) + """ + # Top Tool bar + self.newproj = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/newProject.png'), + 'New Project', self + ) + self.newproj.setShortcut('Ctrl+N') + self.newproj.triggered.connect(self.new_project) + + self.openproj = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/openProject.png'), + 'Open Project', self + ) + self.openproj.setShortcut('Ctrl+O') + self.openproj.triggered.connect(self.open_project) + + self.closeproj = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/closeProject.png'), + 'Close Project', self + ) + self.closeproj.setShortcut('Ctrl+X') + self.closeproj.triggered.connect(self.close_project) + + self.wrkspce = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/workspace.ico'), + 'Change Workspace', self + ) + self.wrkspce.setShortcut('Ctrl+W') + self.wrkspce.triggered.connect(self.change_workspace) + + self.helpfile = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/helpProject.png'), + 'Help', self + ) + self.helpfile.setShortcut('Ctrl+H') + self.helpfile.triggered.connect(self.help_project) + + self.topToolbar = self.addToolBar('Top Tool Bar') + self.topToolbar.addAction(self.newproj) + self.topToolbar.addAction(self.openproj) + self.topToolbar.addAction(self.closeproj) + self.topToolbar.addAction(self.wrkspce) + self.topToolbar.addAction(self.helpfile) + + # ## This part is meant for SoC Generation which is currently ## + # ## under development and will be will be required in future. ## + # self.soc = QtWidgets.QToolButton(self) + # self.soc.setText('Generate SoC') + # ... + + # This part is setting fossee logo to the right + # corner in the application window. + self.spacer = QtWidgets.QWidget() + self.spacer.setSizePolicy( + QtWidgets.QSizePolicy.Expanding, + QtWidgets.QSizePolicy.Expanding) + self.topToolbar.addWidget(self.spacer) + self.logo = QtWidgets.QLabel() + self.logopic = QtGui.QPixmap( + os.path.join( + os.path.abspath(''), init_path + 'images', 'fosseeLogo.png' + )) + self.logopic = self.logopic.scaled( + QSize(150, 150), QtCore.Qt.KeepAspectRatio) + self.logo.setPixmap(self.logopic) + self.logo.setStyleSheet("padding:0 15px 0 0;") + self.topToolbar.addWidget(self.logo) + + # Left Tool bar Action Widget + self.kicad = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/kicad.png'), + 'Open Schematic', self + ) + self.kicad.triggered.connect(self.obj_kicad.openSchematic) + + self.conversion = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/ki-ng.png'), + 'Convert KiCad to Ngspice', self + ) + self.conversion.triggered.connect(self.obj_kicad.openKicadToNgspice) + + self.ngspice = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/ngspice.png'), + 'Simulate', self + ) + self.ngspice.triggered.connect(self.open_ngspice) + + self.model = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/model.png'), + 'Model Editor', self + ) + self.model.triggered.connect(self.open_modelEditor) + + self.subcircuit = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/subckt.png'), + 'Subcircuit', self + ) + self.subcircuit.triggered.connect(self.open_subcircuit) + + self.nghdl = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/nghdl.png'), 'NGHDL', self + ) + self.nghdl.triggered.connect(self.open_nghdl) + + self.makerchip = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/makerchip.png'), + 'Makerchip-NgVeri', self + ) + self.makerchip.triggered.connect(self.open_makerchip) + + self.omedit = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/omedit.png'), + 'Modelica Converter', self + ) + self.omedit.triggered.connect(self.open_OMedit) + + self.omoptim = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/omoptim.png'), + 'OM Optimisation', self + ) + self.omoptim.triggered.connect(self.open_OMoptim) + + self.conToeSim = QtWidgets.QAction( + QtGui.QIcon(init_path + 'images/icon.png'), + 'Schematics converter', self + ) + self.conToeSim.triggered.connect(self.open_conToeSim) + + # Adding Action Widget to tool bar + self.lefttoolbar = QtWidgets.QToolBar('Left ToolBar') + self.addToolBar(QtCore.Qt.LeftToolBarArea, self.lefttoolbar) + self.lefttoolbar.addAction(self.kicad) + self.lefttoolbar.addAction(self.conversion) + self.lefttoolbar.addAction(self.ngspice) + self.lefttoolbar.addAction(self.model) + self.lefttoolbar.addAction(self.subcircuit) + self.lefttoolbar.addAction(self.makerchip) + self.lefttoolbar.addAction(self.nghdl) + self.lefttoolbar.addAction(self.omedit) + self.lefttoolbar.addAction(self.omoptim) + self.lefttoolbar.addAction(self.conToeSim) + self.lefttoolbar.setOrientation(QtCore.Qt.Vertical) + self.lefttoolbar.setIconSize(QSize(40, 40)) + + def closeEvent(self, event): + ''' + This function closes the ongoing program (process). + When exit button is pressed a Message box pops out with \ + exit message and buttons 'Yes', 'No'. + + 1. If 'Yes' is pressed: + - check that program (process) in procThread_list \ + (a list made in Appconfig.py): + + - if available it terminates that program. + - if the program (process) is not available, \ + then check it in process_obj (a list made in \ + Appconfig.py) and if found, it closes the program. + + 2. If 'No' is pressed: + - the program just continues as it was doing earlier. + ''' + exit_msg = "Are you sure you want to exit the program?" + exit_msg += " All unsaved data will be lost." + reply = QtWidgets.QMessageBox.question( + self, 'Message', exit_msg, QtWidgets.QMessageBox.Yes, + QtWidgets.QMessageBox.No + ) + + if reply == QtWidgets.QMessageBox.Yes: + for proc in self.obj_appconfig.procThread_list: + try: + proc.terminate() + except BaseException: + pass + try: + for process_object in self.obj_appconfig.process_obj: + try: + process_object.close() + except BaseException: + pass + except BaseException: + pass + + # Check if "Open project" and "New project" window is open. + # If yes, just close it when application is closed. + try: + self.project.close() + except BaseException: + pass + if self.chatbot_dock.isVisible(): + self.chatbot_dock.close() + event.accept() + self.systemTrayIcon.showMessage('Exit', 'eSim is Closed.') + + elif reply == QtWidgets.QMessageBox.No: + event.ignore() + + def new_project(self): + """This function call New Project Info class.""" + text, ok = QtWidgets.QInputDialog.getText( + self, 'New Project Info', 'Enter Project Name:' + ) + updated = False + + if ok: + self.projname = (str(text)) + self.project = NewProjectInfo() + directory, filelist = self.project.createProject(self.projname) + + if directory and filelist: + self.obj_Mainview.obj_projectExplorer.addTreeNode( + directory, filelist + ) + updated = True + + if not updated: + print("No new project created") + self.obj_appconfig.print_info('No new project created') + try: + self.obj_appconfig.print_info( + 'Current project is : ' + + self.obj_appconfig.current_project["ProjectName"] + ) + except BaseException: + pass + + def open_project(self): + """This project call Open Project Info class.""" + print("Function : Open Project") + self.project = OpenProjectInfo() + try: + directory, filelist = self.project.body() + self.obj_Mainview.obj_projectExplorer.addTreeNode( + directory, filelist) + except BaseException: + pass + + def close_project(self): + """ + This function closes the saved project. + It first checks whether project (file) is present in list. + + - If present: + - it first kills that process-id. + - closes that file. + - Shows message "Current project is closed" + + - If not present: pass + """ + print("Function : Close Project") + current_project = self.obj_appconfig.current_project['ProjectName'] + if current_project is None: + pass + else: + temp = self.obj_appconfig.current_project['ProjectName'] + for pid in self.obj_appconfig.proc_dict[temp]: + try: + os.kill(pid, 9) + except BaseException: + pass + self.obj_Mainview.obj_dockarea.closeDock() + self.obj_appconfig.current_project['ProjectName'] = None + self.systemTrayIcon.showMessage( + 'Close', 'Current project ' + + os.path.basename(current_project) + ' is Closed.' + ) + + def change_workspace(self): + """ + This function call changes Workspace + """ + print("Function : Change Workspace") + self.obj_workspace.returnWhetherClickedOrNot(self) + self.hide() + self.obj_workspace.show() + + def help_project(self): + """ + This function opens usermanual in dockarea. + - It prints the message ""Function : Help"" + - Uses print_info() method of class Appconfig + from Configuration/Appconfig.py file. + - Call method usermanual() from ./DockArea.py. + """ + print("Function : Help") + self.obj_appconfig.print_info('Help is called') + print("Current Project is : ", self.obj_appconfig.current_project) + self.obj_Mainview.obj_dockarea.usermanual() + + @QtCore.pyqtSlot(QtCore.QProcess.ExitStatus, int) + def plotSimulationData(self, exitCode, exitStatus): + """Enables interaction for new simulation and + displays the plotter dock where graphs can be plotted. + """ + self.ngspice.setEnabled(True) + self.conversion.setEnabled(True) + self.closeproj.setEnabled(True) + self.wrkspce.setEnabled(True) + + if exitStatus == QtCore.QProcess.NormalExit and exitCode == 0: + try: + self.obj_Mainview.obj_dockarea.plottingEditor() + except Exception as e: + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle("Error Message") + self.msg.showMessage( + 'Data could not be plotted. Please try again.' + ) + self.msg.exec_() + print("Exception Message:", str(e), traceback.format_exc()) + self.obj_appconfig.print_error('Exception Message : ' + + str(e)) + + self.errorDetectedSignal.emit("Simulation failed.") + + def handleError(self): + self.projDir = self.obj_appconfig.current_project["ProjectName"] + self.output_file = os.path.join(self.projDir, "ngspice_error.log") + # Auto-open the chatbot dock so the user sees the error analysis + if not self.chatbot_dock.isVisible(): + self.chatbot_dock.show() + self.delayed_function_call() + + def delayed_function_call(self): + QTimer.singleShot(2000, lambda: self.chatbot_window.debug_error(self.output_file)) + + def open_ngspice(self): + """This Function execute ngspice on current project.""" + projDir = self.obj_appconfig.current_project["ProjectName"] + + if projDir is not None: + projName = os.path.basename(projDir) + ngspiceNetlist = os.path.join(projDir, projName + ".cir.out") + + if not os.path.isfile(ngspiceNetlist): + print( + "Netlist file (*.cir.out) not found." + ) + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle("Error Message") + self.msg.showMessage( + 'Netlist (*.cir.out) not found.' + ) + self.msg.exec_() + return + + self.obj_Mainview.obj_dockarea.ngspiceEditor( + projName, ngspiceNetlist, self.simulationEndSignal, self.chatbot_window) + + self.ngspice.setEnabled(False) + self.conversion.setEnabled(False) + self.closeproj.setEnabled(False) + self.wrkspce.setEnabled(False) + + else: + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle("Error Message") + self.msg.showMessage( + 'Please select the project first.' + ' You can either create new project or open existing project' + ) + self.msg.exec_() + + def open_subcircuit(self): + """ + This function opens 'subcircuit' option in left-tool-bar. + When 'subcircuit' icon is clicked wich is present in + left-tool-bar of main page: + + - Meassge shown on screen "Subcircuit editor is called". + - 'subcircuiteditor()' function is called using object + 'obj_dockarea' of class 'Mainview'. + """ + print("Function : Subcircuit editor") + self.obj_appconfig.print_info('Subcircuit editor is called') + self.obj_Mainview.obj_dockarea.subcircuiteditor() + + def open_nghdl(self): + """ + This function calls NGHDL option in left-tool-bar. + It uses validateTool() method from Validation.py: + + - If 'nghdl' is present in executables list then + it passes command 'nghdl -e' to WorkerThread class of + Worker.py. + - If 'nghdl' is not present, then it shows error message. + """ + print("Function : NGHDL") + self.obj_appconfig.print_info('NGHDL is called') + + if self.obj_validation.validateTool('nghdl'): + self.cmd = 'nghdl -e' + self.obj_workThread = Worker.WorkerThread(self.cmd) + self.obj_workThread.start() + else: + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle('NGHDL Error') + self.msg.showMessage('Error while opening NGHDL. ' + + 'Please make sure it is installed') + self.obj_appconfig.print_error('Error while opening NGHDL. ' + + 'Please make sure it is installed') + self.msg.exec_() + + def open_makerchip(self): + """ + This function opens 'subcircuit' option in left-tool-bar. + When 'subcircuit' icon is clicked wich is present in + left-tool-bar of main page: + + - Meassge shown on screen "Subcircuit editor is called". + - 'subcircuiteditor()' function is called using object + 'obj_dockarea' of class 'Mainview'. + """ + print("Function : Makerchip and Verilator to Ngspice Converter") + self.obj_appconfig.print_info('Makerchip is called') + self.obj_Mainview.obj_dockarea.makerchip() + + def open_modelEditor(self): + """ + This function opens model editor option in left-tool-bar. + When model editor icon is clicked which is present in + left-tool-bar of main page: + + - Meassge shown on screen "Model editor is called". + - 'modeleditor()' function is called using object + 'obj_dockarea' of class 'Mainview'. + """ + print("Function : Model editor") + self.obj_appconfig.print_info('Model editor is called') + self.obj_Mainview.obj_dockarea.modelEditor() + + def open_OMedit(self): + """ + This function calls ngspice to OMEdit converter and then launch OMEdit. + """ + self.obj_appconfig.print_info('OMEdit is called') + self.projDir = self.obj_appconfig.current_project["ProjectName"] + + if self.projDir is not None: + if self.obj_validation.validateCirOut(self.projDir): + self.projName = os.path.basename(self.projDir) + self.ngspiceNetlist = os.path.join( + self.projDir, self.projName + ".cir.out" + ) + self.modelicaNetlist = os.path.join( + self.projDir, self.projName + ".mo" + ) + self.obj_Mainview.obj_dockarea.modelicaEditor(self.projDir) + + else: + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle("Missing Ngspice Netlist") + self.msg.showMessage( + 'Current project does not contain any Ngspice file. ' + + 'Please create Ngspice file with extension .cir.out' + ) + self.msg.exec_() + else: + self.msg = QtWidgets.QErrorMessage() + self.msg.setModal(True) + self.msg.setWindowTitle("Error Message") + self.msg.showMessage( + 'Please select the project first. You can either ' + + 'create a new project or open an existing project' + ) + self.msg.exec_() + + def open_OMoptim(self): + """ + This function uses validateTool() method from Validation.py: + + - If 'OMOptim' is present in executables list then + it passes command 'OMOptim' to WorkerThread class of Worker.py + - If 'OMOptim' is not present, then it shows error message with + link to download it on Linux and Windows. + """ + print("Function : OMOptim") + self.obj_appconfig.print_info('OMOptim is called') + # Check if OMOptim is installed + if self.obj_validation.validateTool("OMOptim"): + # Creating a command to run + self.cmd = "OMOptim" + self.obj_workThread = Worker.WorkerThread(self.cmd) + self.obj_workThread.start() + else: + self.msg = QtWidgets.QMessageBox() + self.msgContent = ( + "There was an error while opening OMOptim.
" + "Please make sure OpenModelica is installed in your" + " system.
" + "To install it on Linux : Go to OpenModelica Linux and install nightly build" + " release.
" + "To install it on Windows : Go to OpenModelica Windows and install latest version.
" + ) + self.msg.setTextFormat(QtCore.Qt.RichText) + self.msg.setText(self.msgContent) + self.msg.setWindowTitle("Error Message") + self.obj_appconfig.print_info(self.msgContent) + self.msg.exec_() + + def open_conToeSim(self): + print("Function : Schematics converter") + self.obj_appconfig.print_info('Schematics converter is called') + self.obj_Mainview.obj_dockarea.eSimConverter() + + +# This class initialize the Main View of Application +class MainView(QtWidgets.QWidget): + """ + This class defines whole view and style of main page: + + - Position of tool bars: + - Top tool bar. + - Left tool bar. + - Project explorer Area. + - Dock area. + - Console area. + """ + + def __init__(self, *args): + # call init method of superclass + QtWidgets.QWidget.__init__(self, *args) + + self.obj_appconfig = Appconfig() + + self.leftSplit = QtWidgets.QSplitter() + self.middleSplit = QtWidgets.QSplitter() + + self.mainLayout = QtWidgets.QVBoxLayout() + + self.middleContainer = QtWidgets.QWidget() + self.middleContainerLayout = QtWidgets.QVBoxLayout() + + + self.noteArea = QtWidgets.QTextEdit() + self.noteArea.setReadOnly(True) + self.obj_appconfig.noteArea['Note'] = self.noteArea + self.obj_appconfig.noteArea['Note'].append( + ' eSim Started......') + self.obj_appconfig.noteArea['Note'].append('Project Selected : None') + self.obj_appconfig.noteArea['Note'].append('\n') + # CSS + self.noteArea.setStyleSheet(" \ + QWidget { border-radius: 15px; border: 1px \ + solid gray; padding: 5px; } \ + ") + + self.obj_dockarea = DockArea.DockArea() + self.obj_projectExplorer = ProjectExplorer.ProjectExplorer() + + # Adding content to vertical middle Split. + self.middleSplit.setOrientation(QtCore.Qt.Vertical) + self.middleSplit.addWidget(self.obj_dockarea) + self.middleSplit.addWidget(self.noteArea) + + # Adding middle split to Middle Container Widget + self.middleContainerLayout.addWidget(self.middleSplit) + self.middleContainer.setLayout(self.middleContainerLayout) + + # Adding content of left split + self.leftSplit.addWidget(self.obj_projectExplorer) + self.leftSplit.addWidget(self.middleContainer) + + # Adding to main Layout + self.mainLayout.addWidget(self.leftSplit) + self.leftSplit.setSizes([int(self.width() / 4.5), self.height()]) + self.middleSplit.setSizes([self.width(), int(self.height() / 2)]) + self.setLayout(self.mainLayout) + + +def main(args): + """ + The splash screen opened at the starting of screen is performed + by this function. + """ + print("Starting eSim......") + app = QtWidgets.QApplication(args) + app.setApplicationName("eSim") + + appView = Application() + appView.hide() + + splash_pix = QtGui.QPixmap(init_path + 'images/splash_screen_esim.png') + splash = QtWidgets.QSplashScreen( + appView, splash_pix, QtCore.Qt.WindowStaysOnTopHint + ) + splash.setMask(splash_pix.mask()) + splash.setDisabled(True) + splash.show() + + appView.splash = splash + appView.obj_workspace.returnWhetherClickedOrNot(appView) + + try: + user_home = os.path.expanduser('~') + file = open(os.path.join(user_home, ".esim/workspace.txt"), 'r') + work = int(file.read(1)) + file.close() + except IOError: + work = 0 + + if work != 0: + appView.obj_workspace.defaultWorkspace() + else: + appView.obj_workspace.show() + + sys.exit(app.exec_()) + + +if __name__ == '__main__': + # Create and display the splash screen + try: + main(sys.argv) + except Exception as err: + print("Error: ", err) \ No newline at end of file diff --git a/src/frontEnd/Chatbot.py b/src/frontEnd/Chatbot.py new file mode 100644 index 000000000..1a3e75701 --- /dev/null +++ b/src/frontEnd/Chatbot.py @@ -0,0 +1,2914 @@ +from chatbot.chatbot_thread import ( + OllamaWorker, OllamaVisionWorker, MicWorker, + OllamaStatusWorker, ModelFetchWorker, + detect_topic_switch, get_stt_backend +) +from PyQt5.QtWidgets import ( + QWidget, QHBoxLayout, QTextBrowser, QVBoxLayout, + QLineEdit, QPushButton, QLabel, QComboBox, QApplication, + QFileDialog, QDialog, QListWidget, QListWidgetItem, QFrame, + QScrollArea, QSlider, QInputDialog +) +from PyQt5.QtCore import QTimer, Qt, pyqtSignal, QSize +from PyQt5.QtGui import QTextCursor, QKeyEvent, QDragEnterEvent, QDropEvent +from configuration.Appconfig import Appconfig +from datetime import datetime +import re +import os +import json +import uuid +import base64 + +if os.name == 'nt': + from frontEnd import pathmagic # noqa:F401 + init_path = '' +else: + import pathmagic # noqa:F401 + init_path = '../../' + +# ── Storage paths ───────────────────────────────────────────────────────────── +_ESIM_DIR = os.path.join(os.path.expanduser('~'), '.esim') +_HISTORY_FILE = os.path.join(_ESIM_DIR, 'chatbot_history.json') +_SESSIONS_DIR = os.path.join(_ESIM_DIR, 'chat_sessions') + +_IMG_FILTER = "Images (*.png *.jpg *.jpeg *.bmp *.gif *.tiff)" +_IMAGE_EXTS = {'.png', '.jpg', '.jpeg', '.bmp', '.gif', '.tiff', '.tif', '.webp'} +# NgSpice logs can be 10-50 KB; sending all of it blows past num_ctx: 2048. +# 60 lines is enough for any meaningful error message while staying well inside +# the context window even with history prepended. +_MAX_ERROR_LOG_LINES = 60 +# _save_history() is called after every bot response; without debouncing this +# causes synchronous I/O on the main thread on every message. +_SAVE_DEBOUNCE_MS = 5000 + +WELCOME_MESSAGE = """ +
+
🤖
+
+ eSim AI Assistant +
+
+ Ask me anything about KiCad, NgSpice,
+ netlists, simulation errors, or circuit design.
+ Attach an image 📎 or speak 🎤 your question. +
+
+ Use the sidebar to access past chats +
+
+
+""" + +_TYPING_FRAMES = [ + '● ● ●', + '● ● ●', + '● ● ●', +] + + +def _typing_bubble(frame=0): + dots = _TYPING_FRAMES[frame % 3] + return ( + '' + '' + '
' + '' + '
' + '
' + f'{dots}
' + ) + + +# ── Markdown renderer ───────────────────────────────────────────────────────── + +def _render_inline(text): + """ + Renders inline markdown: **bold**, *italic*, `code`, # headings, and [links](url). + """ + # Escape HTML special chars first so subsequent substitutions are safe + text = text.replace('&', '&').replace('<', '<').replace('>', '>') + + # Headings (must be processed line-by-line because they are block-level) + def _render_headings(t): + lines = t.split('\n') + out = [] + for line in lines: + m = re.match(r'^(#{1,4})\s+(.*)', line) + if m: + level = len(m.group(1)) + sizes = {1: '18px', 2: '16px', 3: '14px', 4: '13px'} + size = sizes.get(level, '13px') + content = m.group(2) + out.append( + f'{content}' + ) + else: + out.append(line) + return '\n'.join(out) + + text = _render_headings(text) + + # Bold (**text** or __text__) + text = re.sub(r'\*\*(.*?)\*\*', r'\1', text) + text = re.sub(r'__(.*?)__', r'\1', text) + + # Italic (*text* or _text_) — processed after bold so ** is already gone + text = re.sub(r'\*(.*?)\*', r'\1', text) + text = re.sub(r'_(.*?)_', r'\1', text) + + # Inline code (`code`) + text = re.sub( + r'`([^`]+)`', + r'\1', + text + ) + + # Markdown links [text](url) + text = re.sub( + r'\[([^\]]+)\]\((https?://[^\)]+)\)', + r'\1', + text + ) + + text = text.replace('\n', '
') + return text + + +def _render_markdown(text): + result = [] + pattern = re.compile(r'```(\w*)\n?(.*?)```', re.DOTALL) + last_end = 0 + + for match in pattern.finditer(text): + before = text[last_end:match.start()] + if before: + result.append(_render_inline(before)) + + lang = match.group(1) or 'code' + code = ( + match.group(2) + .replace('&', '&') + .replace('<', '<') + .replace('>', '>') + .replace('\n', '
') + .replace(' ', ' ') + ) + label = f'{lang}
' if lang else '' + result.append( + '' + '
' + '
' + f'{label}{code}
' + ) + last_end = match.end() + + tail = text[last_end:] + if tail: + result.append(_render_inline(tail)) + return ''.join(result) + + +# ── Bubble helpers ──────────────────────────────────────────────────────────── + +def _get_time(): + return datetime.now().strftime("%H:%M") + + +def _escape_text_preserve_breaks(text: str) -> str: + return ( + text.replace('&', '&') + .replace('<', '<') + .replace('>', '>') + .replace('\n', '
') + ) + + +def _image_thumbnail_html(b64_str: str, filename: str) -> str: + """Render a saved image as an inline base64 thumbnail in the chat.""" + safe_name = filename.replace('&', '&').replace('<', '<') + return ( + '' + '' + '
' + '' + '
' + f'' + f'
{safe_name}
' + '
' + '
' + ) + + +def _user_bubble(text, timestamp): + safe = _escape_text_preserve_breaks(text) + return ( + '' + '' + '
' + '' + '' + f'' + '
' + f'{safe}' + '
You  ·  {timestamp}
' + '
' + ) + + +def _approx_token_count(text: str) -> int: + return max(1, len(text) // 4) + + +def _bot_bubble(text, timestamp, response_idx): + rendered = _render_markdown(text) + copy_href = f'copy:///{response_idx}' + retry_href = f'retry:///{response_idx}' + token_est = _approx_token_count(text) + + return ( + '' + '' + '
' + '' + '' + '
' + f'{rendered}' + '
' + '' + f'' + f'' + '
' + f'eSim AI  ·  {timestamp}  ·  ~{token_est} tokens' + f'↻ Retry' + f'  ' + f'Copy
' + '
' + '
' + ) + + +def _bot_bubble_simple(text, timestamp): + rendered = _render_markdown(text) + return ( + '' + '' + '
' + '' + '' + f'' + '
' + f'{rendered}' + '
' + f'eSim AI  ·  {timestamp}
' + '
' + ) + + +def _system_bubble(text): + return ( + '' + '
' + '
' + f'{_escape_text_preserve_breaks(text)}
' + ) + + +def _staged_images_bubble(filenames, timestamp): + names_html = "".join( + f'' + f'📎 {_escape_text_preserve_breaks(n)} ' + for n in filenames + ) + return ( + '' + '' + '
' + '' + '' + f'' + '
' + f'{names_html}' + '
You  ·  {timestamp}
' + ) + + +def _topic_reset_banner(): + return ( + '' + '
' + '— New topic —' + '
' + ) + + +def _netlist_header_bubble(filename, timestamp): + safe = _escape_text_preserve_breaks(filename) + return ( + '' + '' + '
' + '' + '' + f'' + '
' + '
' + f'📄 Netlist: {safe}
' + '
You  ·  {timestamp}
' + ) + + +def _parse_custom_url(url): + scheme = url.scheme() + host = url.host() + path = url.path().strip('/') + + parts = [] + if host: + parts.append(host) + if path: + parts.extend([p for p in path.split('/') if p]) + + return scheme, parts + + +def _session_kind_badge(kind: str) -> str: + colors = { + "text": ("#eef4ff", "#2d6cdf"), + "image": ("#eefaf0", "#1f8b4c"), + "netlist": ("#fff4e8", "#b86a00"), + "simulation_error": ("#fff0f0", "#c62828"), + } + bg, fg = colors.get(kind, ("#f0f0f0", "#666")) + label = { + "text": "Text", + "image": "Image", + "netlist": "Netlist", + "simulation_error": "Sim Error", + }.get(kind, kind.title()) + return ( + f'{label}' + ) + + +def _is_image_file(path: str) -> bool: + return os.path.splitext(path)[1].lower() in _IMAGE_EXTS + + +# ── Smart input field ───────────────────────────────────────────────────────── + +class _HistoryLineEdit(QLineEdit): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + self._sent_history = [] + self._hist_idx = -1 + + def add_to_history(self, text): + if text and (not self._sent_history or self._sent_history[-1] != text): + self._sent_history.append(text) + self._hist_idx = -1 + + def keyPressEvent(self, event: QKeyEvent): + if event.key() == Qt.Key_Up and self._sent_history: + if self._hist_idx == -1: + self._draft = self.text() + self._hist_idx = len(self._sent_history) - 1 + elif self._hist_idx > 0: + self._hist_idx -= 1 + self.setText(self._sent_history[self._hist_idx]) + self.end(False) + elif event.key() == Qt.Key_Down and self._hist_idx >= 0: + self._hist_idx += 1 + if self._hist_idx >= len(self._sent_history): + self._hist_idx = -1 + self.setText(getattr(self, '_draft', '')) + else: + self.setText(self._sent_history[self._hist_idx]) + self.end(False) + else: + super().keyPressEvent(event) + + +# ── History viewer ──────────────────────────────────────────────────────────── + +class ChatHistoryViewer(QDialog): + def __init__(self, session: dict, parent=None): + super().__init__(parent) + raw_title = session.get('title', 'Chat') + self.setWindowTitle(raw_title[:60]) + self.setMinimumSize(460, 560) + self.resize(500, 640) + self.setStyleSheet("QDialog { background:#f9f9f9; }") + + msgs = session.get('messages', []) + n_usr = sum(1 for m in msgs if m.startswith("User:")) + n_bot = sum(1 for m in msgs if m.startswith("Bot:")) + created = session.get('created_at', '') + updated = session.get('updated_at', '') + kind = session.get('kind', 'text') + + root = QVBoxLayout(self) + root.setContentsMargins(0, 0, 0, 0) + root.setSpacing(0) + + top_bar = QWidget() + top_bar.setFixedHeight(54) + top_bar.setStyleSheet(""" + QWidget { + background:#ffffff; + border-bottom:1px solid #ececec; + } + """) + top_layout = QHBoxLayout(top_bar) + top_layout.setContentsMargins(16, 0, 12, 0) + top_layout.setSpacing(10) + + title_lbl = QLabel(raw_title[:50] + ("…" if len(raw_title) > 50 else "")) + title_lbl.setStyleSheet( + "font-size:14px; font-weight:700; color:#1a1a2e; background:transparent;" + ) + top_layout.addWidget(title_lbl, 1) + + meta_lbl = QLabel(f"{n_usr} msg{'s' if n_usr!=1 else ''} · {updated[:10]} · {kind}") + meta_lbl.setStyleSheet("font-size:10px; color:#aaa; background:transparent;") + top_layout.addWidget(meta_lbl) + + x_btn = QPushButton("✕") + x_btn.setFixedSize(26, 26) + x_btn.setStyleSheet(""" + QPushButton { + font-size:11px; color:#888; + background:transparent; border:none; border-radius:13px; + } + QPushButton:hover { background:#f0f0f0; color:#333; } + QPushButton:pressed{ background:#e0e0e0; } + """) + x_btn.clicked.connect(self.accept) + top_layout.addWidget(x_btn) + root.addWidget(top_bar) + + browser = QTextBrowser() + browser.setOpenLinks(False) + browser.setStyleSheet(""" + QTextBrowser { + background:#f9f9f9; + border:none; + padding:10px 4px; + font-family:'Segoe UI',Arial,sans-serif; + font-size:13px; + } + """) + + html = "" + for line in msgs: + if line.startswith("User:"): + html += _user_bubble(line[5:].strip(), "") + elif line.startswith("Bot:"): + html += _bot_bubble_simple(line[4:].strip(), "") + browser.setHtml(html if html else "

No messages

") + QTimer.singleShot(120, lambda: browser.verticalScrollBar().setValue(browser.verticalScrollBar().maximum())) + root.addWidget(browser) + + bot_bar = QWidget() + bot_bar.setFixedHeight(52) + bot_bar.setStyleSheet(""" + QWidget { + background:#ffffff; + border-top:1px solid #ececec; + } + """) + bot_layout = QHBoxLayout(bot_bar) + bot_layout.setContentsMargins(16, 0, 16, 0) + + info_lbl = QLabel(f"Created {created[:10]} · {n_usr + n_bot} total messages") + info_lbl.setStyleSheet("font-size:10px; color:#bbb; background:transparent;") + bot_layout.addWidget(info_lbl) + bot_layout.addStretch() + + done_btn = QPushButton("Done") + done_btn.setFixedHeight(32) + done_btn.setStyleSheet(""" + QPushButton { + font-size:12px; font-weight:600; + padding:4px 22px; + background:#0095f6; color:white; + border:none; border-radius:16px; + } + QPushButton:hover { background:#0082d8; } + QPushButton:pressed{ background:#006ab8; } + """) + done_btn.clicked.connect(self.accept) + bot_layout.addWidget(done_btn) + root.addWidget(bot_bar) + + +# ── Sidebar ─────────────────────────────────────────────────────────────────── + +class _DeleteConfirmDialog(QDialog): + def __init__(self, title: str, parent=None): + super().__init__(parent, Qt.FramelessWindowHint | Qt.Dialog) + self.setAttribute(Qt.WA_TranslucentBackground) + self.setMinimumWidth(320) + + outer = QWidget(self) + outer.setObjectName("card") + outer.setStyleSheet(""" + QWidget#card { + background: #ffffff; + border-radius: 20px; + border: 1px solid #e0e0e0; + } + """) + + card_layout = QVBoxLayout(outer) + card_layout.setContentsMargins(28, 24, 28, 20) + card_layout.setSpacing(14) + + title_lbl = QLabel("Delete chat?") + title_lbl.setStyleSheet("font-size:16px; font-weight:bold; color:#1a1a2e;") + title_lbl.setAlignment(Qt.AlignCenter) + card_layout.addWidget(title_lbl) + + body_lbl = QLabel( + f'' + f'Delete “{_escape_text_preserve_breaks(title[:40])}”?
' + f'This cannot be undone.' + ) + body_lbl.setWordWrap(True) + body_lbl.setAlignment(Qt.AlignCenter) + card_layout.addWidget(body_lbl) + + div = QFrame() + div.setFrameShape(QFrame.HLine) + div.setStyleSheet("color:#f0f0f0;") + card_layout.addWidget(div) + + btn_row = QHBoxLayout() + btn_row.setSpacing(10) + + cancel_btn = QPushButton("Cancel") + cancel_btn.setFixedHeight(40) + cancel_btn.setStyleSheet(""" + QPushButton { + font-size:13px; font-weight:600; + background:#f0f0f0; color:#333; + border:none; border-radius:20px; padding:0 24px; + } + QPushButton:hover { background:#e0e0e0; } + QPushButton:pressed{ background:#d0d0d0; } + """) + cancel_btn.clicked.connect(self.reject) + + delete_btn = QPushButton("Delete") + delete_btn.setFixedHeight(40) + delete_btn.setStyleSheet(""" + QPushButton { + font-size:13px; font-weight:600; + background:#ff3b30; color:white; + border:none; border-radius:20px; padding:0 24px; + } + QPushButton:hover { background:#e0302a; } + QPushButton:pressed{ background:#c02520; } + """) + delete_btn.clicked.connect(self.accept) + + btn_row.addWidget(cancel_btn) + btn_row.addWidget(delete_btn) + card_layout.addLayout(btn_row) + + main = QVBoxLayout(self) + main.setContentsMargins(0, 0, 0, 0) + main.addWidget(outer) + + +class _SessionItemWidget(QWidget): + delete_requested = pyqtSignal(str) + rename_requested = pyqtSignal(str) + + def __init__(self, session_id: str, title: str, date: str, + msg_count: int = 0, preview: str = "", kind: str = "text", parent=None): + super().__init__(parent) + self.session_id = session_id + self.title = title + self.kind = kind + + self.setMinimumHeight(78) + self.setStyleSheet("QWidget { background: transparent; }") + + outer = QHBoxLayout(self) + outer.setContentsMargins(10, 8, 8, 8) + outer.setSpacing(10) + + avatar = QLabel(title[0].upper() if title else "C") + avatar.setFixedSize(38, 38) + avatar.setAlignment(Qt.AlignCenter) + avatar.setStyleSheet(""" + QLabel { + background: qlineargradient( + x1:0, y1:0, x2:1, y2:1, + stop:0 #0095f6, stop:1 #0055a5 + ); + color: white; + font-size: 14px; + font-weight: 700; + border-radius: 19px; + } + """) + outer.addWidget(avatar) + + text_col = QVBoxLayout() + text_col.setSpacing(3) + text_col.setContentsMargins(0, 0, 0, 0) + + title_row = QHBoxLayout() + title_row.setSpacing(4) + title_row.setContentsMargins(0, 0, 0, 0) + + title_lbl = QLabel(title[:22] + ("…" if len(title) > 22 else "")) + title_lbl.setStyleSheet( + "font-size:12px; font-weight:700; color:#1a1a2e; background:transparent;" + ) + title_row.addWidget(title_lbl, 1) + + date_lbl = QLabel(date) + date_lbl.setStyleSheet("font-size:10px; color:#bbb; background:transparent;") + title_row.addWidget(date_lbl) + text_col.addLayout(title_row) + + meta_row = QHBoxLayout() + meta_row.setSpacing(4) + meta_row.setContentsMargins(0, 0, 0, 0) + + kind_lbl = QLabel() + kind_lbl.setText(_session_kind_badge(kind)) + kind_lbl.setTextFormat(Qt.RichText) + kind_lbl.setStyleSheet("background:transparent;") + meta_row.addWidget(kind_lbl) + + if msg_count > 0: + count_lbl = QLabel(str(msg_count)) + count_lbl.setFixedSize(20, 16) + count_lbl.setAlignment(Qt.AlignCenter) + count_lbl.setStyleSheet(""" + QLabel { + background:#0095f6; color:white; + font-size:9px; font-weight:700; + border-radius:8px; + } + """) + meta_row.addWidget(count_lbl) + meta_row.addStretch() + text_col.addLayout(meta_row) + + preview_text = (preview[:32] + "…") if len(preview) > 32 else preview + preview_lbl = QLabel(preview_text if preview_text else "No messages yet") + preview_lbl.setStyleSheet("font-size:11px; color:#888; background:transparent;") + text_col.addWidget(preview_lbl) + + outer.addLayout(text_col, 1) + + btn_col = QVBoxLayout() + btn_col.setSpacing(4) + btn_col.setContentsMargins(0, 0, 0, 0) + + self._rename_btn = QPushButton("✎") + self._rename_btn.setFixedSize(28, 28) + self._rename_btn.setToolTip("Rename this chat") + self._rename_btn.setStyleSheet(""" + QPushButton { + font-size:12px; + background:#f2f7ff; + color:#0055a5; + border:1px solid #d0e0ff; + border-radius:14px; + } + QPushButton:hover { background:#e6f0ff; } + """) + self._rename_btn.clicked.connect(lambda: self.rename_requested.emit(self.session_id)) + btn_col.addWidget(self._rename_btn) + + self._del_btn = QPushButton("🗑") + self._del_btn.setFixedSize(28, 28) + self._del_btn.setToolTip("Delete this chat") + self._del_btn.setStyleSheet(""" + QPushButton { + font-size:12px; + background:#fff0f0; + color:#cc0000; + border:1px solid #ffd0d0; + border-radius:14px; + } + QPushButton:hover { background:#ffe0e0; border:1px solid #ffb0b0; } + QPushButton:pressed{ background:#ffc8c8; } + """) + self._del_btn.clicked.connect(self._on_delete_clicked) + btn_col.addWidget(self._del_btn) + + btn_col.addStretch() + outer.addLayout(btn_col) + + def sizeHint(self): + return QSize(252, 78) + + def _on_delete_clicked(self): + dlg = _DeleteConfirmDialog(self.title, self) + if dlg.exec_() == QDialog.Accepted: + self.delete_requested.emit(self.session_id) + + +class ChatSidebar(QWidget): + new_chat_requested = pyqtSignal() + session_deleted = pyqtSignal(str) + delete_all_requested = pyqtSignal() + rename_requested = pyqtSignal(str) + + def __init__(self, parent=None): + super().__init__(parent) + self.setFixedWidth(290) + self._all_sessions_cache = [] + + self.setStyleSheet(""" + QWidget { + background:#ffffff; + border-right:1px solid #ececec; + } + """) + + root = QVBoxLayout(self) + root.setContentsMargins(0, 0, 0, 0) + root.setSpacing(0) + + top = QWidget() + top.setFixedHeight(52) + top.setStyleSheet(""" + QWidget { + background:#ffffff; + border-bottom:1px solid #f0f0f0; + } + """) + top_row = QHBoxLayout(top) + top_row.setContentsMargins(14, 0, 10, 0) + top_row.setSpacing(8) + + title_lbl = QLabel("Chats") + title_lbl.setStyleSheet( + "font-size:16px; font-weight:700; color:#1a1a2e; background:transparent;" + ) + top_row.addWidget(title_lbl, 1) + + close_btn = QPushButton("✕") + close_btn.setFixedSize(26, 26) + close_btn.setStyleSheet(""" + QPushButton { + font-size:11px; color:#888; + background:transparent; border:none; border-radius:13px; + } + QPushButton:hover { background:#f0f0f0; color:#333; } + QPushButton:pressed{ background:#e0e0e0; } + """) + close_btn.clicked.connect(self.hide) + top_row.addWidget(close_btn) + root.addWidget(top) + + controls = QWidget() + controls_layout = QVBoxLayout(controls) + controls_layout.setContentsMargins(12, 8, 12, 8) + controls_layout.setSpacing(8) + + self.new_btn = QPushButton("+ New Chat") + self.new_btn.setFixedHeight(36) + self.new_btn.setStyleSheet(""" + QPushButton { + font-size:12px; font-weight:600; + background:#0095f6; color:white; + border:none; border-radius:18px; + } + QPushButton:hover { background:#0082d8; } + QPushButton:pressed{ background:#006ab8; } + """) + self.new_btn.clicked.connect(self.new_chat_requested) + controls_layout.addWidget(self.new_btn) + + self.search_input = QLineEdit() + self.search_input.setPlaceholderText("Search chats…") + self.search_input.setFixedHeight(34) + self.search_input.setStyleSheet(""" + QLineEdit { + font-size:12px; + padding:7px 12px; + border:1px solid #e0e0e0; + border-radius:17px; + background:#f7f7f7; + color:#1a1a2e; + } + QLineEdit:focus { + border:1px solid #0095f6; + background:#ffffff; + } + """) + self.search_input.textChanged.connect(self._apply_filter) + controls_layout.addWidget(self.search_input) + + delete_all_btn = QPushButton("Delete All Chats") + delete_all_btn.setFixedHeight(30) + delete_all_btn.setStyleSheet(""" + QPushButton { + font-size:11px; + font-weight:600; + background:#fff0f0; + color:#cc0000; + border:1px solid #ffd0d0; + border-radius:15px; + } + QPushButton:hover { background:#ffe0e0; } + QPushButton:pressed{ background:#ffc8c8; } + """) + delete_all_btn.clicked.connect(self.delete_all_requested) + controls_layout.addWidget(delete_all_btn) + + root.addWidget(controls) + + sep = QFrame() + sep.setFrameShape(QFrame.HLine) + sep.setFixedHeight(1) + sep.setStyleSheet("QFrame { background:#f0f0f0; border:none; }") + root.addWidget(sep) + + self.session_list = QListWidget() + self.session_list.setSpacing(2) + self.session_list.setStyleSheet(""" + QListWidget { + background:#ffffff; + border:none; + outline:0; + padding:6px; + } + QListWidget::item { + border:none; + padding:0; + margin:0; + } + QListWidget::item:hover { background:#f5f8ff; } + QListWidget::item:selected { background:#eaf3ff; } + """) + root.addWidget(self.session_list) + + self._empty_lbl = QLabel("No saved chats yet.\nStart a conversation!") + self._empty_lbl.setAlignment(Qt.AlignCenter) + self._empty_lbl.setStyleSheet(""" + QLabel { + color:#ccc; font-size:12px; + padding:30px 10px; + background:transparent; + } + """) + self._empty_lbl.setWordWrap(True) + self._empty_lbl.hide() + root.addWidget(self._empty_lbl) + + def populate(self): + self._all_sessions_cache = [] + self.session_list.clear() + + if not os.path.exists(_SESSIONS_DIR): + self._empty_lbl.show() + return + + for fname in os.listdir(_SESSIONS_DIR): + if not fname.endswith('.json'): + continue + try: + with open(os.path.join(_SESSIONS_DIR, fname), encoding='utf-8') as f: + s = json.load(f) + self._all_sessions_cache.append(s) + except Exception: + pass + + self._all_sessions_cache.sort(key=lambda s: s.get('updated_at', ''), reverse=True) + self._apply_filter() + + def _apply_filter(self): + self.session_list.clear() + query = self.search_input.text().strip().lower() + + filtered = [] + for s in self._all_sessions_cache: + title = s.get('title', 'Chat') + msgs = s.get('messages', []) + preview = next((m[5:].strip() for m in msgs if m.startswith("User:")), "") + kind = s.get('kind', 'text') + haystack = f"{title} {preview} {kind}".lower() + if not query or query in haystack: + filtered.append(s) + + if not filtered: + self._empty_lbl.show() + return + + self._empty_lbl.hide() + + for s in filtered: + sid = s['id'] + title = s.get('title', 'Chat') + date = s.get('updated_at', '')[:10] + msgs = s.get('messages', []) + msg_count = sum(1 for m in msgs if m.startswith("User:")) + preview = next((m[5:].strip() for m in msgs if m.startswith("User:")), "") + kind = s.get('kind', 'text') + + item = QListWidgetItem() + item.setData(Qt.UserRole, sid) + widget = _SessionItemWidget(sid, title, date, msg_count, preview, kind, self.session_list) + widget.delete_requested.connect(self._delete_session) + widget.rename_requested.connect(self.rename_requested) + + item.setSizeHint(widget.sizeHint()) + self.session_list.addItem(item) + self.session_list.setItemWidget(item, widget) + + def upsert_session(self, session: dict): + """ + Insert or update a session entry in the sidebar immediately, + without reading from disk. Called as soon as the first bot reply + arrives so the chat appears in the sidebar right away instead of + waiting for the debounced disk save to complete. + """ + sid = session.get('id') + if not sid: + return + + # Update existing entry in the cache if present, otherwise prepend it. + for i, s in enumerate(self._all_sessions_cache): + if s.get('id') == sid: + self._all_sessions_cache[i] = session + break + else: + self._all_sessions_cache.insert(0, session) + + # Re-sort so the newest session stays at the top. + self._all_sessions_cache.sort( + key=lambda s: s.get('updated_at', ''), reverse=True + ) + self._apply_filter() + + def _delete_session(self, session_id: str): + path = os.path.join(_SESSIONS_DIR, f"{session_id}.json") + try: + if os.path.exists(path): + os.remove(path) + except Exception: + pass + self.session_deleted.emit(session_id) + self.populate() + + +# ── Main Chatbot GUI ────────────────────────────────────────────────────────── + +class ChatbotGUI(QWidget): + # Emitted from _suspend_worker's background callback to safely update + # the sidebar from the main thread after a background save completes. + _background_session_saved = pyqtSignal(dict) + + def __init__(self): + super().__init__() + self.setWindowTitle("eSim AI Assistant") + self.setMinimumSize(420, 340) + self.resize(430, 350) + self.setAcceptDrops(True) + + self.chat_history = [] + self._retry_history = [] + self._bot_responses = {} + self._response_counter = 0 + self._last_user_text = "" + self._typing_frame = 0 + self._typing_start_pos = -1 + self._was_ollama_offline = True + self._mic_active = False + self._viewing_past_session = False + self._staged_images = [] + self._temperature = 0.35 + self._num_predict = 1024 + self._current_session_id = str(uuid.uuid4()) + self._session_created_at = datetime.now().strftime("%Y-%m-%d %H:%M") + self._current_session_kind = "text" + self._session_title_override = None + self._is_generating = False + self._images_store = {} # key -> [base64_str, ...] for image replay + self._last_image_paths = [] # image paths from last vision send (for follow-ups) + # batched rather than firing synchronously after every bot response. + self._save_pending = False + self._save_debounce_timer = QTimer(self) + self._save_debounce_timer.setSingleShot(True) + self._save_debounce_timer.timeout.connect(self._flush_save) + + self._thinking_timer = QTimer(self) + self._thinking_timer.timeout.connect(self._animate_thinking) + + self._typing_anim_timer = QTimer(self) + self._typing_anim_timer.timeout.connect(self._animate_typing_bubble) + + self._status_poll_timer = QTimer(self) + self._status_poll_timer.timeout.connect(self._update_ollama_status) + self._status_poll_timer.start(5000) + + self._toast = QLabel(" ✅ Copied! ", self) + self._toast.setStyleSheet(""" + QLabel { + background-color:#1a1a2e; color:#ffffff; + font-size:12px; font-weight:bold; + border-radius:14px; padding:4px 14px; + } + """) + self._toast.setAlignment(Qt.AlignCenter) + self._toast.hide() + + root = QHBoxLayout(self) + root.setContentsMargins(0, 0, 0, 0) + root.setSpacing(0) + + self._sidebar = ChatSidebar(self) + self._sidebar.new_chat_requested.connect(self._new_chat) + self._sidebar.session_deleted.connect(self._on_session_deleted) + self._sidebar.delete_all_requested.connect(self._delete_all_chats) + self._sidebar.rename_requested.connect(self._rename_session_by_id) + self._sidebar.session_list.itemClicked.connect(self._on_session_clicked) + self._sidebar.session_list.itemDoubleClicked.connect(self._open_session_viewer) + self._sidebar.hide() + root.addWidget(self._sidebar) + # Route background-thread session saves through a signal so the + # sidebar upsert always runs on the main thread (Qt requirement). + self._background_session_saved.connect(self._sidebar_upsert_from_signal) + + chat_container = QWidget() + chat_layout = QVBoxLayout(chat_container) + chat_layout.setContentsMargins(8, 8, 8, 8) + chat_layout.setSpacing(5) + root.addWidget(chat_container, 1) + + header_layout = QHBoxLayout() + header_layout.setSpacing(5) + + self._history_btn = QPushButton("≡") + self._history_btn.setFixedSize(32, 32) + self._history_btn.setToolTip("Chat history") + self._history_btn.setStyleSheet(""" + QPushButton { + font-size:16px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + QPushButton:pressed { background:#e0e0e0; } + """) + self._history_btn.clicked.connect(self._toggle_sidebar) + header_layout.addWidget(self._history_btn) + + self.model_combo = QComboBox(self) + self.model_combo.setFixedHeight(30) + self.model_combo.setStyleSheet(""" + QComboBox { + font-size:12px; padding:2px 10px; + border:1px solid #e0e0e0; border-radius:8px; + background:#f7f7f7; color:#1a1a2e; + } + QComboBox:focus { border:1px solid #0095f6; background:#fff; } + QComboBox::drop-down { border:none; width:18px; } + """) + self._populate_models() + header_layout.addWidget(self.model_combo) + + self._refresh_models_btn = QPushButton("↻") + self._refresh_models_btn.setFixedSize(28, 28) + self._refresh_models_btn.setToolTip("Refresh available models") + self._refresh_models_btn.setStyleSheet(""" + QPushButton { + font-size:14px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + """) + self._refresh_models_btn.clicked.connect(self._populate_models) + header_layout.addWidget(self._refresh_models_btn) + + self._rename_btn = QPushButton("✎") + self._rename_btn.setFixedSize(28, 28) + self._rename_btn.setToolTip("Rename current chat") + self._rename_btn.setStyleSheet(""" + QPushButton { + font-size:13px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + """) + self._rename_btn.clicked.connect(self._rename_current_chat) + header_layout.addWidget(self._rename_btn) + + self._settings_btn = QPushButton("⚙") + self._settings_btn.setFixedSize(28, 28) + self._settings_btn.setToolTip("Model settings") + self._settings_btn.setCheckable(True) + self._settings_btn.setStyleSheet(""" + QPushButton { + font-size:14px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + QPushButton:checked { background:#e8f0ff; color:#0095f6; } + """) + header_layout.addWidget(self._settings_btn) + + self._export_btn = QPushButton("⤓") + self._export_btn.setFixedSize(28, 28) + self._export_btn.setToolTip("Export current chat") + self._export_btn.setStyleSheet(""" + QPushButton { + font-size:13px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + """) + self._export_btn.clicked.connect(self._export_current_chat) + header_layout.addWidget(self._export_btn) + + self._regen_btn = QPushButton("⟳") + self._regen_btn.setFixedSize(28, 28) + self._regen_btn.setToolTip("Regenerate last response") + self._regen_btn.setStyleSheet(""" + QPushButton { + font-size:13px; border:none; + border-radius:8px; background:transparent; color:#555; + } + QPushButton:hover { background:#f0f0f0; color:#1a1a2e; } + """) + self._regen_btn.clicked.connect(self._regenerate_last_response) + header_layout.addWidget(self._regen_btn) + + header_layout.addStretch() + + self.ollama_status_label = QLabel(self) + self.ollama_status_label.setFixedHeight(24) + header_layout.addWidget(self.ollama_status_label) + self._update_ollama_status() + + header_sep = QFrame() + header_sep.setFrameShape(QFrame.HLine) + header_sep.setStyleSheet("color:#ececec; margin:0;") + chat_layout.addLayout(header_layout) + chat_layout.addWidget(header_sep) + + self.chat_display = QTextBrowser(self) + self.chat_display.setOpenLinks(False) + self.chat_display.setOpenExternalLinks(False) + self.chat_display.setHtml(WELCOME_MESSAGE) + self.chat_display.anchorClicked.connect(self._handle_link_click) + self.chat_display.setStyleSheet(""" + QTextBrowser { + background-color:#fafafa; + border:none; + padding:8px 4px; + font-family:'Segoe UI',Arial,sans-serif; + font-size:13px; + selection-background-color:#cce4f7; + } + QScrollBar:vertical { + background:transparent; width:6px; + } + QScrollBar::handle:vertical { + background:#d0d0d0; border-radius:3px; min-height:24px; + } + QScrollBar::handle:vertical:hover { background:#a0a0a0; } + QScrollBar::add-line:vertical, QScrollBar::sub-line:vertical { + height:0px; + } + """) + chat_layout.addWidget(self.chat_display) + + status_layout = QHBoxLayout() + self.status_label = QLabel("", self) + self.status_label.setStyleSheet( + "color:#0095f6; font-size:11px; padding:1px 4px; background:transparent;" + ) + status_layout.addWidget(self.status_label) + status_layout.addStretch() + + chat_layout.addLayout(status_layout) + + self._settings_panel = QWidget() + self._settings_panel.setVisible(False) + self._settings_panel.setStyleSheet(""" + QWidget { + background:#f7f9fc; + border-top:1px solid #ececec; + border-bottom:1px solid #ececec; + } + """) + self._settings_btn.toggled.connect(lambda on: self._settings_panel.setVisible(on)) + + sp_layout = QHBoxLayout(self._settings_panel) + sp_layout.setContentsMargins(12, 8, 12, 8) + sp_layout.setSpacing(16) + + temp_col = QVBoxLayout() + self._temp_label = QLabel(f"Precision {self._temperature:.2f}") + self._temp_label.setStyleSheet("font-size:10px; color:#555;") + temp_col.addWidget(self._temp_label) + + self._temp_slider = QSlider(Qt.Horizontal) + self._temp_slider.setRange(1, 100) + self._temp_slider.setValue(int(self._temperature * 100)) + self._temp_slider.setFixedWidth(110) + self._temp_slider.valueChanged.connect(self._on_temp_changed) + temp_col.addWidget(self._temp_slider) + sp_layout.addLayout(temp_col) + + tok_col = QVBoxLayout() + self._tok_label = QLabel(f"Max tokens {self._num_predict}") + self._tok_label.setStyleSheet("font-size:10px; color:#555;") + tok_col.addWidget(self._tok_label) + + self._tok_slider = QSlider(Qt.Horizontal) + self._tok_slider.setRange(1, 40) + self._tok_slider.setValue(self._num_predict // 128) + self._tok_slider.setFixedWidth(110) + self._tok_slider.valueChanged.connect(self._on_tok_changed) + tok_col.addWidget(self._tok_slider) + sp_layout.addLayout(tok_col) + + sp_layout.addStretch() + + reset_btn = QPushButton("Reset") + reset_btn.setFixedHeight(26) + reset_btn.setStyleSheet(""" + QPushButton { + font-size:10px; padding:2px 12px; + background:#f0f0f0; color:#555; + border:none; border-radius:13px; + } + """) + reset_btn.clicked.connect(self._reset_settings) + sp_layout.addWidget(reset_btn) + chat_layout.addWidget(self._settings_panel) + + input_layout = QHBoxLayout() + + self.attach_button = QPushButton("📎") + self.attach_button.setFixedSize(38, 38) + self.attach_button.setToolTip( + "Attach image for analysis\n" + "Tip: install Pillow (pip install Pillow) to auto-downscale\n" + "large images for faster analysis" + ) + self.attach_button.setStyleSheet(""" + QPushButton { + font-size:16px; background:#f0f0f0; + border:none; border-radius:19px; + } + QPushButton:hover { background:#e0e8ff; } + """) + self.attach_button.clicked.connect(self._pick_image) + input_layout.addWidget(self.attach_button) + + self.mic_button = QPushButton("🎤") + self.mic_button.setFixedSize(38, 38) + QTimer.singleShot(200, self._update_mic_tooltip) + self.mic_button.setStyleSheet(""" + QPushButton { + font-size:15px; background:#f0f0f0; + border:none; border-radius:19px; + } + QPushButton:hover { background:#d0f8d0; } + """) + self.mic_button.clicked.connect(self._on_mic_clicked) + input_layout.addWidget(self.mic_button) + + self.user_input = _HistoryLineEdit( + self, placeholderText="Message eSim AI… (↑↓ for history)" + ) + self.user_input.setStyleSheet(""" + QLineEdit { + font-size:13px; padding:9px 14px; + border:1.5px solid #e0e0e0; border-radius:22px; + background:#f7f7f7; color:#1a1a2e; + } + QLineEdit:focus { + border:1.5px solid #0095f6; + background:#ffffff; + } + """) + self.user_input.returnPressed.connect(self.ask_ollama) + input_layout.addWidget(self.user_input) + + self.send_button = QPushButton("Send") + self.send_button.setFixedHeight(38) + self.send_button.setStyleSheet(""" + QPushButton { + font-size:13px; font-weight:600; padding:5px 20px; + background-color:#0095f6; color:white; + border:none; border-radius:19px; + } + QPushButton:hover { background-color:#0082d8; } + """) + self.send_button.clicked.connect(self.ask_ollama) + input_layout.addWidget(self.send_button) + + self.stop_button = QPushButton("Stop") + self.stop_button.setFixedHeight(38) + self.stop_button.setStyleSheet(""" + QPushButton { + font-size:13px; font-weight:600; padding:5px 16px; + background-color:#ff3b30; color:white; + border:none; border-radius:19px; + } + """) + self.stop_button.clicked.connect(self._stop_generating) + self.stop_button.hide() + input_layout.addWidget(self.stop_button) + + self.clear_button = QPushButton("Clear") + self.clear_button.setFixedHeight(38) + self.clear_button.setStyleSheet(""" + QPushButton { + font-size:13px; padding:5px 14px; + background-color:#f0f0f0; color:#666; + border:none; border-radius:19px; + } + QPushButton:hover { background-color:#ffe0e0; color:#cc0000; } + """) + self.clear_button.clicked.connect(self.clear_session) + input_layout.addWidget(self.clear_button) + + chat_layout.addLayout(input_layout) + + self._staging_area = QWidget() + self._staging_area.setStyleSheet("QWidget { background:#f5f8ff; border-radius:10px; }") + self._staging_area.setVisible(False) + + staging_outer = QVBoxLayout(self._staging_area) + staging_outer.setContentsMargins(6, 6, 6, 4) + staging_outer.setSpacing(4) + + staging_header = QHBoxLayout() + staged_lbl = QLabel("Images to send:") + staged_lbl.setStyleSheet("font-size:11px;color:#555;") + staging_header.addWidget(staged_lbl) + staging_header.addStretch() + + clear_all_btn = QPushButton("Remove all") + clear_all_btn.setFixedHeight(20) + clear_all_btn.setStyleSheet(""" + QPushButton { + font-size:10px; color:#cc0000; background:transparent; + border:none; padding:0 4px; + } + QPushButton:hover { text-decoration:underline; } + """) + clear_all_btn.clicked.connect(self._clear_staged_images) + staging_header.addWidget(clear_all_btn) + staging_outer.addLayout(staging_header) + + scroll = QScrollArea() + scroll.setFixedHeight(72) + scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded) + scroll.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff) + scroll.setWidgetResizable(True) + scroll.setStyleSheet("QScrollArea { border:none; background:transparent; }") + + self._thumb_container = QWidget() + self._thumb_row = QHBoxLayout(self._thumb_container) + self._thumb_row.setContentsMargins(0, 0, 0, 0) + self._thumb_row.setSpacing(6) + self._thumb_row.addStretch() + + scroll.setWidget(self._thumb_container) + staging_outer.addWidget(scroll) + chat_layout.addWidget(self._staging_area) + + self.move_to_bottom_right() + self._load_history() + + # ── Drag & drop ─────────────────────────────────────────────────── + + def dragEnterEvent(self, event: QDragEnterEvent): + mime = event.mimeData() + if mime.hasUrls(): + for url in mime.urls(): + if url.isLocalFile() and _is_image_file(url.toLocalFile()): + event.acceptProposedAction() + return + event.ignore() + + def dropEvent(self, event: QDropEvent): + mime = event.mimeData() + if not mime.hasUrls(): + event.ignore() + return + + added = 0 + for url in mime.urls(): + if not url.isLocalFile(): + continue + path = url.toLocalFile() + if _is_image_file(path) and path not in self._staged_images: + self._staged_images.append(path) + added += 1 + + if added: + self._refresh_staging_strip() + self.status_label.setText(f"📎 Added {added} image{'s' if added != 1 else ''} by drag-and-drop.") + QTimer.singleShot(2500, lambda: self.status_label.setText("")) + + event.acceptProposedAction() + + # ── Sidebar / sessions ──────────────────────────────────────────── + + def _toggle_sidebar(self): + if self._sidebar.isVisible(): + self._sidebar.hide() + else: + self._sidebar.populate() + self._sidebar.show() + + def _refresh_sidebar_if_open(self): + if self._sidebar.isVisible(): + self._sidebar.populate() + + def _delete_all_chats(self): + dlg = _DeleteConfirmDialog("all chats", self) + if dlg.exec_() != QDialog.Accepted: + return + + try: + if os.path.exists(_SESSIONS_DIR): + for fname in os.listdir(_SESSIONS_DIR): + if fname.endswith(".json"): + os.remove(os.path.join(_SESSIONS_DIR, fname)) + except Exception: + pass + + self._sidebar.populate() + + def _open_session_viewer(self, item): + session_id = item.data(Qt.UserRole) + path = os.path.join(_SESSIONS_DIR, f"{session_id}.json") + try: + with open(path, encoding='utf-8') as f: + session = json.load(f) + except Exception: + return + dlg = ChatHistoryViewer(session, self) + dlg.exec_() + + def _rename_current_chat(self): + title, ok = QInputDialog.getText( + self, "Rename Chat", "New chat title:", + text=self._session_title_override or self._derive_session_title() + ) + if ok: + title = title.strip() + if title: + self._session_title_override = title + self._save_history() + self.status_label.setText("✏️ Chat renamed.") + QTimer.singleShot(2000, lambda: self.status_label.setText("")) + + def _rename_session_by_id(self, session_id: str): + path = os.path.join(_SESSIONS_DIR, f"{session_id}.json") + try: + with open(path, encoding='utf-8') as f: + session = json.load(f) + except Exception: + return + + current_title = session.get("title", "Chat") + title, ok = QInputDialog.getText( + self, "Rename Chat", "New chat title:", text=current_title + ) + if not ok: + return + title = title.strip() + if not title: + return + + try: + session["title"] = title + with open(path, "w", encoding="utf-8") as f: + json.dump(session, f, ensure_ascii=False, indent=2) + except Exception: + return + + if session_id == self._current_session_id: + self._session_title_override = title + + self._sidebar.populate() + + def _derive_session_title(self): + if self._session_title_override: + return self._session_title_override + return next( + (m[5:].strip()[:50] for m in self.chat_history if m.startswith("User:")), + "Chat" + ) + + def _rebuild_chat_html_from_history(self): + self.chat_display.setHtml(WELCOME_MESSAGE) + self._bot_responses = {} + self._response_counter = 0 + + for line in self.chat_history: + if line.startswith("User:"): + self.chat_display.append(_user_bubble(line[5:].strip(), "")) + elif line.startswith("Bot:"): + idx = self._response_counter + text = line[4:].strip() + self._bot_responses[idx] = text + self.chat_display.append(_bot_bubble(text, "", idx)) + self._response_counter += 1 + self._scroll_to_bottom() + + def _on_session_clicked(self, item): + session_id = item.data(Qt.UserRole) + + # If this is the session already showing, do nothing. + if (session_id == self._current_session_id + and not self._viewing_past_session): + return + + # Suspend BEFORE changing self._current_session_id so the worker + # snapshot captures the correct (old) session ID and history. + # Then flush the current session to disk so the file exists for + # _on_background_response to update when the worker finishes. + if self._is_generating: + self._suspend_worker( + session_id=self._current_session_id, + history=self.chat_history, + session_kind=self._current_session_kind, + images_store=self._images_store, + ) + + self._save_debounce_timer.stop() + self._save_pending = False + self._save_current_session() + + # Load the target session — try disk first, fall back to the + # in-memory sidebar cache (handles sessions not yet written to disk). + path = os.path.join(_SESSIONS_DIR, f"{session_id}.json") + session = None + try: + with open(path, encoding='utf-8') as f: + session = json.load(f) + except Exception: + pass + + if session is None: + for s in self._sidebar._all_sessions_cache: + if s.get('id') == session_id: + session = s + break + + if session is None: + return + + msgs = session.get('messages', []) + title = session.get('title', 'Chat') + created = session.get('created_at', '') + kind = session.get('kind', 'text') + + # Switch the active session context to the one being viewed so that + # if the user types a follow-up, it goes to the right session. + self._current_session_id = session_id + self._session_created_at = created + self._current_session_kind = kind + self._session_title_override = title if title != "Chat" else None + self.chat_history = list(msgs) + self._retry_history = list(msgs) + self._last_user_text = next( + (m[5:].strip() for m in reversed(msgs) if m.startswith("User:")), "" + ) + + # Restore image store from session so follow-ups can re-send images + saved_images = session.get("images", {}) + self._images_store = saved_images + self._last_image_paths = [] # original paths are gone; base64 stored instead + + html = WELCOME_MESSAGE + html += ( + '' + '
' + '
' + f'Viewing saved chat: {_escape_text_preserve_breaks(title[:50])}' + f'  ·  {_escape_text_preserve_breaks(created)}' + f'  ·  {kind}' + '
' + 'Scroll down to see full conversation' + '

' + ) + + self._bot_responses = {} + local_counter = 0 + + # Build a flat list of saved image thumbnails in order for replay + all_saved_imgs = [] + for key in sorted(saved_images.keys()): + all_saved_imgs.extend(saved_images[key]) + img_replay_idx = 0 + + for line in msgs: + if line.startswith("User:"): + text = line[5:].strip() + # If this line is an image-analysis request, show the thumbnail + if text.startswith("[Image analysis request:"): + # Show saved thumbnails for this entry + while img_replay_idx < len(all_saved_imgs): + fname, b64 = all_saved_imgs[img_replay_idx] + html += _image_thumbnail_html(b64, fname) + img_replay_idx += 1 + # Only consume images for this request + if img_replay_idx >= len(all_saved_imgs): + break + # Also show any user text after the image tag + user_text_part = text.split("\n", 1)[-1].strip() + if user_text_part and not user_text_part.startswith("[Image"): + html += _user_bubble(user_text_part, "") + else: + html += _user_bubble(text, "") + elif line.startswith("Bot:"): + text = line[4:].strip() + self._bot_responses[local_counter] = text + html += _bot_bubble(text, "", local_counter) + local_counter += 1 + self._response_counter = local_counter + + self.chat_display.setHtml(html) + QTimer.singleShot(120, lambda: self.chat_display.verticalScrollBar().setValue( + self.chat_display.verticalScrollBar().maximum() + )) + + # Load the session's messages into chat_history so follow-up questions + # have full context, and update the session ID so any new messages save + # to the correct file rather than the previous live session. + self.chat_history = list(msgs) + self._retry_history = list(msgs) + self._current_session_id = session_id + self._session_created_at = session.get('created_at', datetime.now().strftime("%Y-%m-%d %H:%M")) + self._current_session_kind = kind + self._session_title_override = session.get('title', None) + self._last_user_text = next( + (m[5:].strip() for m in reversed(msgs) if m.startswith("User:")), "" + ) + self._viewing_past_session = True + + def _abort_worker(self): + """ + Stop the active worker immediately and discard its response. + Use _suspend_worker() instead when switching sessions so the + generation can finish silently in the background. + """ + if hasattr(self, 'worker') and self.worker.isRunning(): + self.worker.stop() + try: + self.worker.response_signal.disconnect() + self.worker.status_signal.disconnect() + except Exception: + pass + self.worker.wait(300) + self._stop_thinking() + + def _sidebar_upsert_from_signal(self, session: dict): + """Slot — always called on the main thread via _background_session_saved.""" + self._sidebar.upsert_session(session) + + def _suspend_worker(self, session_id: str, history: list, + session_kind: str, images_store: dict): + """ + Detach the running worker from the UI and let it finish in the + background. When it completes, the bot reply is appended to the + session file on disk so the user sees the full conversation the + next time they open that chat from the sidebar. + """ + if not (hasattr(self, 'worker') and self.worker.isRunning()): + self._stop_thinking() + return + + # Snapshot everything the callback needs before self.* moves on. + _sid = session_id + _history = list(history) + _kind = session_kind + _images = dict(images_store) + _worker = self.worker + _signal = self._background_session_saved # Qt signal, safe to emit from thread + + def _on_background_response(bot_response: str): + """ + Called from the worker thread when generation finishes. + Saves the response to disk, then emits a signal so the sidebar + update happens on the main thread (direct QWidget calls from + worker threads cause crashes on some platforms). + """ + try: + _history.append(f"Bot: {bot_response}") + path = os.path.join(_SESSIONS_DIR, f"{_sid}.json") + + if os.path.exists(path): + with open(path, encoding="utf-8") as fp: + session = json.load(fp) + else: + # Session file doesn't exist yet — build it from the snapshot. + session = { + "id": _sid, + "title": next( + (m[5:].strip()[:50] for m in _history + if m.startswith("User:")), "Chat" + ), + "created_at": datetime.now().strftime("%Y-%m-%d %H:%M"), + "kind": _kind, + "images": _images, + } + + session["messages"] = _history[-40:] + session["updated_at"] = datetime.now().strftime("%Y-%m-%d %H:%M") + session["kind"] = _kind + + os.makedirs(_SESSIONS_DIR, exist_ok=True) + with open(path, "w", encoding="utf-8") as fp: + json.dump(session, fp, ensure_ascii=False, indent=2) + + # Emit signal — the connected slot runs on the main thread. + _signal.emit(session) + except Exception: + pass + + try: + _worker.response_signal.disconnect() + _worker.status_signal.disconnect() + except Exception: + pass + + _worker.response_signal.connect(_on_background_response) + self._stop_thinking() + + def _new_chat(self): + # Stop the debounce timer and flush the current session to disk NOW, + # before anything is reset, so the file is written under the correct ID. + self._save_debounce_timer.stop() + self._save_pending = False + if self._is_generating: + # Generation is running — detach it so it finishes silently and + # saves its reply into the current session file when it completes. + self._suspend_worker( + session_id=self._current_session_id, + history=self.chat_history, + session_kind=self._current_session_kind, + images_store=self._images_store, + ) + else: + # Save current session synchronously so it lands on disk before + # we move on. _save_current_session() is a no-op if chat_history + # is empty, so clicking New Chat on a blank window is safe. + self._save_current_session() + + # Reset UI and state for the new blank session WITHOUT calling + # clear_session() — that method deletes the session file, which + # would erase the chat we just saved above. + self.chat_display.setHtml(WELCOME_MESSAGE) + self.chat_history = [] + self._retry_history = [] + self._bot_responses = {} + self._response_counter = 0 + self._last_user_text = "" + self._viewing_past_session = False + self._clear_staged_images() + self._images_store = {} + self._last_image_paths = [] + self._current_session_kind = "text" + self._session_title_override = None + self._current_session_id = str(uuid.uuid4()) + self._session_created_at = datetime.now().strftime("%Y-%m-%d %H:%M") + + try: + if os.path.exists(_HISTORY_FILE): + os.remove(_HISTORY_FILE) + except Exception: + pass + + self._sidebar.populate() + self._current_session_kind = "text" + self._session_title_override = None + self._sidebar.populate() + + def _on_session_deleted(self, deleted_id: str): + if deleted_id == self._current_session_id or self._viewing_past_session: + self._abort_worker() + + # Cancel any pending debounced save so the deleted session + # file cannot be re-created by a timer that was already running. + self._save_debounce_timer.stop() + self._save_pending = False + + self._current_session_id = str(uuid.uuid4()) + self._session_created_at = datetime.now().strftime("%Y-%m-%d %H:%M") + self._current_session_kind = "text" + self._session_title_override = None + self._viewing_past_session = False + self.chat_history = [] + self._retry_history = [] + self._bot_responses = {} + self._response_counter = 0 + self._last_user_text = "" + self._images_store = {} + self._last_image_paths = [] + try: + if os.path.exists(_HISTORY_FILE): + os.remove(_HISTORY_FILE) + except Exception: + pass + self.chat_display.setHtml(WELCOME_MESSAGE) + + # ── Export ──────────────────────────────────────────────────────── + + def _export_current_chat(self): + if not self.chat_history: + self.status_label.setText("Nothing to export.") + QTimer.singleShot(2500, lambda: self.status_label.setText("")) + return + + path, _ = QFileDialog.getSaveFileName( + self, + "Export Chat", + os.path.join(os.path.expanduser("~"), "chat_export.txt"), + "Text Files (*.txt);;Markdown Files (*.md)" + ) + if not path: + return + + try: + with open(path, "w", encoding="utf-8") as f: + for line in self.chat_history: + f.write(line.strip() + "\n\n") + self.status_label.setText("✅ Chat exported.") + QTimer.singleShot(2500, lambda: self.status_label.setText("")) + except Exception as e: + self.status_label.setText(f"❌ Export failed: {e}") + QTimer.singleShot(3500, lambda: self.status_label.setText("")) + + # ── Ollama status ──────────────────────────────────────────────── + + def _update_ollama_status(self): + self._status_worker = OllamaStatusWorker() + self._status_worker.result_signal.connect(self._on_status_result) + self._status_worker.start() + + def _on_status_result(self, running: bool): + if running: + self.ollama_status_label.setText("🟢 Live") + self.ollama_status_label.setStyleSheet(""" + QLabel { + font-size:11px; font-weight:bold; padding:2px 10px; + border-radius:12px; background-color:#e6f9ee; + color:#1a7f3c; border:1px solid #a3d9b5; + } + """) + if self._was_ollama_offline: + self._was_ollama_offline = False + self._populate_models() + else: + self.ollama_status_label.setText("🔴 Offline") + self.ollama_status_label.setStyleSheet(""" + QLabel { + font-size:11px; font-weight:bold; padding:2px 10px; + border-radius:12px; background-color:#fdecea; + color:#b71c1c; border:1px solid #f5c0bc; + } + """) + self._was_ollama_offline = True + + # ── Typing bubble ───────────────────────────────────────────────── + + # ── Typing bubble (window-switch safe) ────────────────────────── + # + # (_typing_start_pos) and used it to select-and-replace the animated + # dots on every timer tick. When the user switches away from the + # chatbot window Qt reflows the QTextBrowser's HTML document, which + # shifts character positions. On the next timer tick the cursor + # landed in the wrong place and deleted real chat content. + # + # New approach: insert a sentinel anchor tag with a unique id + # ("_typing_anchor_") right before the bubble HTML. To update or + # remove the bubble we search the document for that anchor using + # QTextDocument.find() — which is position-independent and survives + # any reflow — then select from the match to the end of the document. + # The sentinel itself is a zero-width invisible link so it never + # appears in the rendered output. + + _TYPING_ANCHOR = '' + + def _find_typing_anchor_cursor(self): + """Return a cursor positioned at the typing-bubble sentinel, + or None if the sentinel is not in the document. + + PyQt5 exposes anchor names via QTextCharFormat.anchorNames() + (returns a list) not .anchorName() -- we handle both spellings + defensively so the code works across PyQt5 versions. + """ + doc = self.chat_display.document() + block = doc.begin() + while block.isValid(): + it = block.begin() + while not it.atEnd(): + frag = it.fragment() + if frag.isValid(): + fmt = frag.charFormat() + # PyQt5 uses anchorNames() -> list[str] + # Some builds also have anchorName() -> str + # We try both so it works regardless of version. + try: + names = fmt.anchorNames() # PyQt5 standard + matched = "_typing_anchor_" in (names or []) + except AttributeError: + try: + matched = fmt.anchorName() == "_typing_anchor_" + except AttributeError: + matched = False + if matched: + cursor = QTextCursor(doc) + cursor.setPosition(frag.position()) + return cursor + it += 1 + block = block.next() + return None + + def _show_typing_bubble(self): + self._typing_frame = 0 + cursor = QTextCursor(self.chat_display.document()) + cursor.movePosition(QTextCursor.End) + # Insert sentinel anchor + bubble in one operation so they form + # a contiguous block that can be fully removed later. + cursor.insertHtml(self._TYPING_ANCHOR + _typing_bubble(0)) + self._scroll_to_bottom() + self._typing_anim_timer.start(400) + + def _animate_typing_bubble(self): + self._typing_frame = (self._typing_frame + 1) % 3 + anchor_cursor = self._find_typing_anchor_cursor() + if anchor_cursor is None: + # Sentinel gone — stop the timer defensively + self._typing_anim_timer.stop() + return + # Select from the sentinel to the end of the document and replace. + # This is immune to any reflow that happened while the window was + # in the background because we locate by anchor name, not position. + anchor_cursor.movePosition(QTextCursor.End, QTextCursor.KeepAnchor) + anchor_cursor.insertHtml(self._TYPING_ANCHOR + _typing_bubble(self._typing_frame)) + # Only auto-scroll if the user is already near the bottom so we + # don't hijack their scroll position while they read earlier msgs. + sb = self.chat_display.verticalScrollBar() + if sb.maximum() - sb.value() < 60: + self._scroll_to_bottom() + + def _remove_typing_bubble(self): + self._typing_anim_timer.stop() + anchor_cursor = self._find_typing_anchor_cursor() + if anchor_cursor is not None: + anchor_cursor.movePosition(QTextCursor.End, QTextCursor.KeepAnchor) + anchor_cursor.removeSelectedText() + # Legacy guard: if somehow _typing_start_pos path left stale state + self._typing_start_pos = -1 + + # ── Links ──────────────────────────────────────────────────────── + + def _handle_link_click(self, url): + scheme, parts = _parse_custom_url(url) + + if scheme == 'copy': + if not parts: + return + try: + idx = int(parts[-1]) + except ValueError: + return + text = self._bot_responses.get(idx, "") + if text: + QApplication.clipboard().setText(text) + self._show_copy_toast() + + elif scheme == 'retry': + if not parts: + return + try: + idx = int(parts[-1]) + except ValueError: + return + self._retry_response(idx) + + elif scheme == 'clear': + self.clear_session() + + def _show_copy_toast(self): + self._toast.setText(" ✅ Copied! ") + chat_rect = self.chat_display.geometry() + tw, th = 110, 30 + x = chat_rect.x() + (chat_rect.width() - tw) // 2 + y = chat_rect.y() + chat_rect.height() - th - 16 + self._toast.setGeometry(x, y, tw, th) + self._toast.show() + self._toast.raise_() + QTimer.singleShot(1600, self._toast.hide) + + # ── Images ─────────────────────────────────────────────────────── + + def _pick_image(self): + paths, _ = QFileDialog.getOpenFileNames(self, "Select Images", "", _IMG_FILTER) + self._stage_image_paths(paths) + + def _stage_image_paths(self, paths): + added = 0 + for path in paths: + if path and _is_image_file(path) and path not in self._staged_images: + self._staged_images.append(path) + added += 1 + if added: + self._refresh_staging_strip() + + def _refresh_staging_strip(self): + while self._thumb_row.count() > 1: + item = self._thumb_row.takeAt(0) + if item.widget(): + item.widget().deleteLater() + + for path in self._staged_images: + self._thumb_row.insertWidget(self._thumb_row.count() - 1, self._make_thumbnail(path)) + + self._staging_area.setVisible(bool(self._staged_images)) + + def _make_thumbnail(self, image_path: str) -> QWidget: + from PyQt5.QtGui import QPixmap + + card = QWidget() + card.setFixedSize(80, 64) + card.setStyleSheet(""" + QWidget { + background:#ffffff; + border:1px solid #d0d8f0; + border-radius:10px; + } + """) + card_layout = QVBoxLayout(card) + card_layout.setContentsMargins(4, 4, 4, 2) + card_layout.setSpacing(2) + + thumb_lbl = QLabel() + thumb_lbl.setAlignment(Qt.AlignCenter) + thumb_lbl.setFixedHeight(36) + pix = QPixmap(image_path) + if not pix.isNull(): + orig_w, orig_h = pix.width(), pix.height() + card.setToolTip( + f"{os.path.basename(image_path)}\n{orig_w} × {orig_h} px" + ) + pix = pix.scaled(68, 36, Qt.KeepAspectRatio, Qt.SmoothTransformation) + thumb_lbl.setPixmap(pix) + else: + thumb_lbl.setText("🖼") + thumb_lbl.setStyleSheet("font-size:20px;background:transparent;") + card_layout.addWidget(thumb_lbl) + + fname = os.path.basename(image_path) + name_lbl = QLabel(fname[:10] + ("…" if len(fname) > 10 else "")) + name_lbl.setAlignment(Qt.AlignCenter) + name_lbl.setStyleSheet("font-size:9px;color:#555;background:transparent;") + card_layout.addWidget(name_lbl) + + remove_btn = QPushButton("✕", card) + remove_btn.setFixedSize(16, 16) + remove_btn.move(62, 2) + remove_btn.setStyleSheet(""" + QPushButton { + font-size:9px; font-weight:bold; + background:#ff3b30; color:white; + border:none; border-radius:8px; + padding:0; + } + QPushButton:hover { background:#cc2a22; } + """) + remove_btn.clicked.connect(lambda checked=False, p=image_path: self._remove_staged_image(p)) + return card + + def _remove_staged_image(self, path: str): + if path in self._staged_images: + self._staged_images.remove(path) + self._refresh_staging_strip() + + def _clear_staged_images(self): + self._staged_images.clear() + self._refresh_staging_strip() + + def _warn_or_switch_to_vision_model(self) -> bool: + """ + Ensure a vision-capable model is selected before sending images. + + Returns True if it is safe to proceed (a vision model is active), + or False if no vision model is installed and the request should be + blocked. Sending images to a text-only model causes it to fabricate + completely wrong answers because it cannot actually see the image. + """ + current = self.model_combo.currentText() + vision_keywords = ["llava", "bakllava", "vision", "moondream", "qwen2-vl", "minicpm-v"] + + # Already on a vision model — good to go. + if any(k in current.lower() for k in vision_keywords): + return True + + # Try to auto-switch to any vision model the user has installed. + preferred_order = ["moondream", "llava:7b", "llava", "bakllava", "llava:13b"] + for i in range(self.model_combo.count()): + name = self.model_combo.itemText(i) + if any(k in name.lower() for k in vision_keywords): + self.model_combo.setCurrentIndex(i) + self.chat_display.append(_system_bubble( + f"Switched to vision model: {name}" + )) + self._scroll_to_bottom() + return True + + # No vision model found — block the request and explain clearly. + self.chat_display.append(_system_bubble( + "⚠️ No vision model installed. Image analysis is not possible with the " + "current model — a text-only model cannot see images and will give " + "completely wrong answers.\n\n" + "Install a vision model by running this in a terminal:\n" + " ollama pull llava\n\n" + "Then restart eSim and select llava from the model dropdown." + )) + self._scroll_to_bottom() + return False + + # ── Mic ────────────────────────────────────────────────────────── + + def _on_temp_changed(self, value: int): + self._temperature = round(value / 100, 2) + self._temp_label.setText(f"Precision {self._temperature:.2f}") + + def _on_tok_changed(self, value: int): + self._num_predict = value * 128 + self._tok_label.setText(f"Max tokens {self._num_predict}") + + def _reset_settings(self): + self._temperature = 0.35 + self._num_predict = 1024 + self._temp_slider.setValue(35) + self._tok_slider.setValue(8) + + def _update_mic_tooltip(self): + backend = get_stt_backend() + tips = { + "whisper": "Speak your question\n✅ Offline STT active (faster-whisper)", + "vosk": "Speak your question\n✅ Offline STT active (vosk)", + "google": "Speak your question\n⚠ Online STT only (Google)", + "none": "Speak your question\n❌ No STT installed", + } + self.mic_button.setToolTip(tips.get(backend, "Speak your question")) + + def _on_mic_clicked(self): + if self._mic_active: + return + self._mic_active = True + self.mic_button.setEnabled(False) + self.status_label.setText("🎤 Starting microphone…") + self._mic_worker = MicWorker() + self._mic_worker.text_signal.connect(self._on_mic_text) + self._mic_worker.error_signal.connect(self._on_mic_error) + self._mic_worker.status_signal.connect(self._on_mic_status) + self._mic_worker.start() + + def _on_mic_status(self, msg: str): + self.status_label.setText(msg) + + def _on_mic_text(self, text: str): + self._reset_mic_button() + self.status_label.setText("") + self.user_input.setText(text) + self.user_input.setFocus() + + def _on_mic_error(self, msg: str): + self._reset_mic_button() + self.status_label.setText(msg) + QTimer.singleShot(3500, lambda: self.status_label.setText("")) + + def _reset_mic_button(self): + self._mic_active = False + self.mic_button.setEnabled(True) + + # ── Netlist analysis ───────────────────────────────────────────── + + def analyse_netlist(self, netlist_path: str): + if not os.path.exists(netlist_path): + self.chat_display.append( + f'
' + f'❌ Netlist file not found: {_escape_text_preserve_breaks(netlist_path)}
' + ) + return + + self._current_session_kind = "netlist" + + ts = _get_time() + filename = os.path.basename(netlist_path) + self.chat_display.append(_netlist_header_bubble(filename, ts)) + self._scroll_to_bottom() + + try: + with open(netlist_path, 'r', errors='replace') as f: + raw_lines = f.readlines() + except Exception as e: + self.chat_display.append( + f'
' + f'❌ Could not read file: {_escape_text_preserve_breaks(str(e))}
' + ) + return + + components, nodes, directives = [], set(), [] + for line in raw_lines: + s = line.strip() + if not s or s.startswith('*'): + continue + first = s[0].upper() + if first in 'RCLVIDQMEFGHJKTUWXZ': + components.append(s) + parts = s.split() + if len(parts) >= 3: + nodes.update([parts[1], parts[2]]) + elif first == '.': + directives.append(s) + + summary = ( + f"Netlist file: {filename}\n" + f"Total lines: {len(raw_lines)}\n" + f"Components ({len(components)}): " + f"{', '.join(components[:15])}{'...' if len(components) > 15 else ''}\n" + f"Unique nodes: {', '.join(sorted(nodes)[:20])}\n" + f"SPICE directives: {', '.join(directives[:10])}\n\n" + f"Full netlist:\n{''.join(raw_lines[:80])}" + f"{'[truncated]' if len(raw_lines) > 80 else ''}" + ) + + prompt = ( + f"Analyse this NgSpice netlist for me.\n\n{summary}\n\n" + "Please: (1) identify all components and their roles, " + "(2) describe what circuit this is and what it does, " + "(3) highlight any potential simulation issues, " + "(4) suggest any improvements." + ) + + self.chat_history = (self.chat_history + [f"User: {prompt}"])[-20:] + self._retry_history = list(self.chat_history) + self._last_user_text = prompt + self._start_thinking() + + self.worker = OllamaWorker( + self.chat_history, + model=self.model_combo.currentText(), + temperature=self._temperature, + num_predict=self._num_predict, + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + + # ── Topic switch ───────────────────────────────────────────────── + + def _check_topic_switch(self, new_text: str) -> bool: + switched = detect_topic_switch(self._last_user_text, new_text) + if switched and self.chat_history: + self.chat_history = self.chat_history[-2:] + self.chat_display.append(_topic_reset_banner()) + self._scroll_to_bottom() + # Clear image follow-up context when topic changes + self._last_image_paths = [] + return switched + + # ── Persistence ────────────────────────────────────────────────── + + def _save_history(self): + """ + Schedules a debounced disk write so saves are batched rather than + firing synchronously after every message, preventing UI freezes. + """ + self._save_pending = True + # Restart the timer so the window slides forward from the last change. + # If the user sends multiple messages quickly, only the final state is + # written, avoiding redundant I/O. + if not self._save_debounce_timer.isActive(): + self._save_debounce_timer.start(_SAVE_DEBOUNCE_MS) + + def _flush_save(self): + """Perform the actual disk write when the debounce timer fires.""" + if not self._save_pending: + return + self._save_pending = False + try: + os.makedirs(os.path.dirname(_HISTORY_FILE), exist_ok=True) + with open(_HISTORY_FILE, 'w', encoding='utf-8') as f: + json.dump(self.chat_history[-20:], f, ensure_ascii=False, indent=2) + self._save_current_session() + self._refresh_sidebar_if_open() + except Exception: + pass + + def _save_current_session(self): + if not self.chat_history: + return + try: + os.makedirs(_SESSIONS_DIR, exist_ok=True) + session = { + "id": self._current_session_id, + "title": self._derive_session_title(), + "created_at": self._session_created_at, + "updated_at": datetime.now().strftime("%Y-%m-%d %H:%M"), + "messages": self.chat_history[-40:], + "kind": self._current_session_kind, + "images": self._images_store, + "settings": { + "temperature": self._temperature, + "num_predict": self._num_predict, + }, + } + path = os.path.join(_SESSIONS_DIR, f"{self._current_session_id}.json") + with open(path, 'w', encoding='utf-8') as f: + json.dump(session, f, ensure_ascii=False, indent=2) + # Keep the sidebar in-memory cache in sync so the chat appears + # immediately without requiring a full populate() from disk. + self._sidebar.upsert_session(session) + except Exception: + pass + + def _load_history(self): + """ + On startup: if a leftover history file exists, archive it into the + sidebar sessions directory so the user can access it from the sidebar, + then delete the file. The chat window always opens fresh. + """ + if not os.path.exists(_HISTORY_FILE): + return + try: + with open(_HISTORY_FILE, 'r', encoding='utf-8') as f: + saved = json.load(f) + if isinstance(saved, list) and saved: + title = next( + (m[5:].strip()[:50] for m in saved if m.startswith("User:")), + "Previous session" + ) + old_session = { + "id": self._current_session_id, + "title": title, + "created_at": self._session_created_at, + "updated_at": datetime.now().strftime("%Y-%m-%d %H:%M"), + "messages": saved[-40:], + "kind": "text", + "settings": { + "temperature": self._temperature, + "num_predict": self._num_predict, + }, + } + os.makedirs(_SESSIONS_DIR, exist_ok=True) + sess_path = os.path.join( + _SESSIONS_DIR, f"{self._current_session_id}.json" + ) + with open(sess_path, 'w', encoding='utf-8') as f: + json.dump(old_session, f, ensure_ascii=False, indent=2) + except Exception: + pass + finally: + try: + os.remove(_HISTORY_FILE) + except Exception: + pass + # New session ID so nothing from the old chat bleeds into the new one + self._current_session_id = str(uuid.uuid4()) + self._session_created_at = datetime.now().strftime("%Y-%m-%d %H:%M") + + # ── Models ─────────────────────────────────────────────────────── + + def _populate_models(self): + self.model_combo.clear() + self.model_combo.addItem("Loading models…") + self.model_combo.setEnabled(False) + self._model_worker = ModelFetchWorker() + self._model_worker.result_signal.connect(self._on_models_fetched) + self._model_worker.start() + + def _on_models_fetched(self, model_names: list): + self.model_combo.clear() + for name in model_names: + self.model_combo.addItem(name) + + preferred_order = [ + 'qwen2.5-coder:3b', + 'llava:13b', + 'llava:7b', + 'llava', + 'bakllava', + ] + chosen_idx = -1 + for preferred in preferred_order: + idx = self.model_combo.findText(preferred) + if idx >= 0: + chosen_idx = idx + break + if chosen_idx >= 0: + self.model_combo.setCurrentIndex(chosen_idx) + + self.model_combo.setEnabled(True) + + # ── Thinking / retry / regenerate ──────────────────────────────── + + def _animate_thinking(self): + pass + + def _start_thinking(self): + self._is_generating = True + self.user_input.setEnabled(False) + self.attach_button.setEnabled(False) + self.mic_button.setEnabled(False) + self._staging_area.setEnabled(False) + self.send_button.hide() + self.stop_button.show() + self.clear_button.setEnabled(False) + self._show_typing_bubble() + + def _stop_thinking(self): + self._is_generating = False + self._remove_typing_bubble() + self.status_label.setText("") + self.user_input.setEnabled(True) + self.attach_button.setEnabled(True) + self.mic_button.setEnabled(True) + self._staging_area.setEnabled(True) + self.stop_button.hide() + self.send_button.show() + self.clear_button.setEnabled(True) + + def _scroll_to_bottom(self): + self.chat_display.verticalScrollBar().setValue( + self.chat_display.verticalScrollBar().maximum() + ) + + def _stop_generating(self): + if hasattr(self, 'worker') and self.worker.isRunning(): + self.worker.stop() + + def _retry_response(self, response_idx: int): + """ + Retry the bot response at response_idx. + Trims chat_history back to just before that response, + rebuilds the UI cleanly, then re-fires the worker so the + new answer replaces the old one with no duplicate bubbles. + """ + if self._is_generating: + return + + # Walk chat_history counting Bot: entries to find the target, + # then slice everything from that point forward off. + bot_count = 0 + trim_to = None + for i, line in enumerate(self.chat_history): + if line.startswith("Bot:"): + if bot_count == response_idx: + trim_to = i + break + bot_count += 1 + + if trim_to is None: + # Fallback: trim the last bot entry + for i in range(len(self.chat_history) - 1, -1, -1): + if self.chat_history[i].startswith("Bot:"): + trim_to = i + break + + if trim_to is None or not any( + l.startswith("User:") for l in self.chat_history[:trim_to] + ): + self.status_label.setText("Nothing to retry.") + QTimer.singleShot(2000, lambda: self.status_label.setText("")) + return + + # Trim history then rebuild UI so the stale bubble is gone + # before the new response is appended. + self.chat_history = self.chat_history[:trim_to] + self._retry_history = list(self.chat_history) + self._rebuild_chat_html_from_history() + self._start_thinking() + + # Re-use vision worker if the last user turn included images. + last_user = next( + (l for l in reversed(self.chat_history) if l.startswith("User:")), "" + ) + followup_paths = [p for p in self._last_image_paths if os.path.exists(p)] + if followup_paths and "[Image analysis request:" in last_user: + prompt = last_user.split("\n", 1)[-1].strip() if "\n" in last_user else "" + self.worker = OllamaVisionWorker( + image_paths=followup_paths, + extra_prompt=prompt, + model=self.model_combo.currentText(), + ) + else: + self.worker = OllamaWorker( + self._retry_history, + model=self.model_combo.currentText(), + temperature=self._temperature, + num_predict=self._num_predict, + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + + def _retry_last(self): + """Legacy shim kept so any external callers don't break.""" + if self.chat_history: + self._retry_response(self._response_counter - 1) + + def _regenerate_last_response(self): + if not self.chat_history: + return + + # Remove trailing bot response if present + if self.chat_history and self.chat_history[-1].startswith("Bot:"): + self.chat_history.pop() + + # Find last user prompt + if not self.chat_history or not self.chat_history[-1].startswith("User:"): + self.status_label.setText("No previous user prompt to regenerate.") + QTimer.singleShot(2500, lambda: self.status_label.setText("")) + return + + # Rebuild UI from trimmed history and retry from same state + self._retry_history = list(self.chat_history) + self._rebuild_chat_html_from_history() + self._start_thinking() + + self.worker = OllamaWorker( + self._retry_history, + model=self.model_combo.currentText(), + temperature=self._temperature, + num_predict=self._num_predict, + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + + def _on_status_update(self, msg: str): + self.status_label.setText(msg) + # Only show as chat bubble for major state changes, not every progress tick + if "Starting Ollama" in msg or "Ollama started" in msg: + self.chat_display.append(_system_bubble(msg)) + self._scroll_to_bottom() + + # ── Main chat logic ────────────────────────────────────────────── + + def ask_ollama(self): + user_text = self.user_input.text().strip() + staged_paths = list(self._staged_images) + + if not user_text and not staged_paths: + return + + if self._is_generating: + return + + if self._viewing_past_session: + # chat_history was already synced when the session was loaded, + # so no rebuild is needed — just clear the read-only flag. + self._viewing_past_session = False + + ts = _get_time() + + if staged_paths: + self._current_session_kind = "image" + if not self._warn_or_switch_to_vision_model(): + # No vision model available — clear staged images and abort. + self._clear_staged_images() + return + + fnames = [os.path.basename(p) for p in staged_paths] + + if user_text: + self.user_input.add_to_history(user_text) + self.user_input.clear() + + # Pass the user's text directly to the vision worker. + # chatbot_thread._build_schematic_vision_prompt() handles both + # cases: if user_text is empty it requests a general analysis; + # if it contains a question that question drives the response. + vision_extra_prompt = user_text + + if user_text: + user_history_text = ( + f"[Image analysis request: {', '.join(fnames)}]\n{user_text}" + ) + else: + user_history_text = ( + f"[Image analysis request: {', '.join(fnames)}]" + ) + + self.chat_history = (self.chat_history + [f"User: {user_history_text}"])[-20:] + self._retry_history = list(self.chat_history) + self._last_user_text = user_text if user_text else "image analysis" + + # Read and encode images before displaying so thumbnails appear + # in the chat bubble immediately when the user sends. + img_key = ts + "_" + self._current_session_id + b64_list = [] + for p in staged_paths: + try: + with open(p, "rb") as f_img: + raw = f_img.read() + # Downscale for storage (reuse PIL if available) + try: + from PIL import Image as _PI + import io as _io2 + img_obj = _PI.open(_io2.BytesIO(raw)) + img_obj.thumbnail((320, 240)) + if img_obj.mode not in ("RGB", "L"): + img_obj = img_obj.convert("RGB") + buf = _io2.BytesIO() + img_obj.save(buf, format="JPEG", quality=75) + raw = buf.getvalue() + except Exception: + pass + b64_list.append((os.path.basename(p), base64.b64encode(raw).decode())) + except Exception: + pass + if b64_list: + self._images_store[img_key] = b64_list + + # Show image thumbnails inline so the user can see what was sent. + if b64_list: + for fname, b64 in b64_list: + self.chat_display.append(_image_thumbnail_html(b64, fname)) + else: + # Fallback to filename badges if encoding failed for all images + self.chat_display.append(_staged_images_bubble(fnames, ts)) + + if user_text: + self.chat_display.append(_user_bubble(user_text, ts)) + self._scroll_to_bottom() + + # Keep paths for follow-up context + self._last_image_paths = list(staged_paths) + + self._clear_staged_images() + self._start_thinking() + + self.worker = OllamaVisionWorker( + image_paths=staged_paths, + extra_prompt=vision_extra_prompt, + model=self.model_combo.currentText(), + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + return + + self._current_session_kind = "text" + self._check_topic_switch(user_text) + self.chat_history = (self.chat_history + [f"User: {user_text}"])[-20:] + self.chat_display.append(_user_bubble(user_text, ts)) + self._scroll_to_bottom() + + self.user_input.add_to_history(user_text) + self.user_input.clear() + self._last_user_text = user_text + self._retry_history = list(self.chat_history) + self._start_thinking() + + # If the user is following up on an image session, re-send the last + # images so the model has visual context for its answer. + followup_image_paths = [ + p for p in self._last_image_paths if os.path.exists(p) + ] + if followup_image_paths and self._current_session_kind in ("image", "text"): + self.worker = OllamaVisionWorker( + image_paths=followup_image_paths, + extra_prompt=user_text, + model=self.model_combo.currentText(), + ) + else: + self.worker = OllamaWorker( + self.chat_history, + model=self.model_combo.currentText(), + temperature=self._temperature, + num_predict=self._num_predict, + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + + # ── Window / response / clear ──────────────────────────────────── + + def move_to_bottom_right(self): + # in Qt 6. Use QApplication.primaryScreen().availableGeometry() instead. + screen = QApplication.primaryScreen().availableGeometry() + widget = self.geometry() + x = screen.width() - widget.width() - 10 + y = screen.height() - widget.height() - 50 + self.move(x, y) + + def display_response(self, bot_response: str): + self._stop_thinking() + ts = _get_time() + idx = self._response_counter + self._response_counter += 1 + self._bot_responses[idx] = bot_response + + self.chat_display.append(_bot_bubble(bot_response, ts, idx)) + self.chat_history.append(f"Bot: {bot_response}") + self._scroll_to_bottom() + self._update_ollama_status() + + # Push a lightweight session entry into the sidebar immediately so + # the new chat appears at the top as soon as the first reply lands, + # without waiting for the debounced disk save (up to 5 seconds). + self._sidebar.upsert_session({ + "id": self._current_session_id, + "title": self._derive_session_title(), + "created_at": self._session_created_at, + "updated_at": datetime.now().strftime("%Y-%m-%d %H:%M"), + "messages": self.chat_history[-40:], + "kind": self._current_session_kind, + }) + + self._save_history() + + # (Retry is now an inline link in every bot bubble; + # the old navbar retry_button has been removed.) + + def clear_session(self): + # Cancel any pending debounced save so _flush_save() can't + # resurrect the session file after we delete it below. + self._save_debounce_timer.stop() + self._save_pending = False + + # Remove session file so it never reappears in the sidebar. + session_file = os.path.join(_SESSIONS_DIR, f"{self._current_session_id}.json") + try: + if os.path.exists(session_file): + os.remove(session_file) + except Exception: + pass + + self.chat_display.setHtml(WELCOME_MESSAGE) + self.chat_history = [] + self._retry_history = [] + self._bot_responses = {} + self._response_counter = 0 + self._last_user_text = "" + self._viewing_past_session = False + self._clear_staged_images() + self._images_store = {} + self._last_image_paths = [] + self._viewing_past_session = False + self._current_session_kind = "text" + self._session_title_override = None + + # Assign a fresh session ID so the next conversation starts clean + self._current_session_id = str(uuid.uuid4()) + self._session_created_at = datetime.now().strftime("%Y-%m-%d %H:%M") + + try: + if os.path.exists(_HISTORY_FILE): + os.remove(_HISTORY_FILE) + except Exception: + pass + + # Refresh sidebar so the cleared session disappears immediately + self._refresh_sidebar_if_open() + + # ── Debug helpers ──────────────────────────────────────────────── + + def debug_ollama(self): + self._current_session_kind = "simulation_error" + self.chat_display.append( + '' + '
' + '
' + '⚠️ Simulation Failed — Analyzing error log…' + '
' + ) + self._scroll_to_bottom() + self._retry_history = list(self.chat_history) + self._start_thinking() + self.worker = OllamaWorker( + self.chat_history, + model=self.model_combo.currentText(), + temperature=self._temperature, + num_predict=self._num_predict, + ) + self.worker.response_signal.connect(self.display_response) + self.worker.status_signal.connect(self._on_status_update) + self.worker.start() + self.user_input.clear() + + def debug_error(self, log): + self.setWindowFlags(self.windowFlags()) + self.show() + self.raise_() + self.activateWindow() + + self.chat_history = [] + self._current_session_kind = "simulation_error" + + if os.path.exists(log): + with open(log, "r") as f: + lines = [ln for ln in f.readlines() if ln.strip()] + + no_compat_index = next( + (i for i, ln in enumerate(lines) if "No compatibility mode selected!" in ln), None + ) + circuit_index = next((i for i, ln in enumerate(lines) if "Circuit:" in ln), None) + total_cpu_index = next( + (i for i, ln in enumerate(lines) if "Total CPU time (seconds)" in ln), None + ) + + before_no_compat = lines[:no_compat_index] if no_compat_index else [] + between = ( + lines[circuit_index + 1:total_cpu_index] + if circuit_index is not None and total_cpu_index is not None + else [] + ) + filtered_lines = before_no_compat + between + # before sending to the model. NgSpice logs can be 10-50 KB; sending + # all of it blows past num_ctx: 2048 and makes the model ignore the + # actual error. The most actionable errors always appear at the end. + if len(filtered_lines) > _MAX_ERROR_LOG_LINES: + truncated_notice = [ + f"[Log truncated: showing last {_MAX_ERROR_LOG_LINES} " + f"of {len(filtered_lines)} lines]\n" + ] + filtered_lines = truncated_notice + filtered_lines[-_MAX_ERROR_LOG_LINES:] + + combined_text = "".join(filtered_lines) + # QLineEdit); display a compact summary label in the status bar instead. + self.status_label.setText( + f"🔍 Analysing error log ({len(filtered_lines)} lines)…" + ) + + self.obj_appconfig = Appconfig() + self.projDir = self.obj_appconfig.current_project["ProjectName"] + output_file = os.path.join(self.projDir, "erroroutput.txt") + with open(output_file, "w") as f: + f.writelines(filtered_lines) + + self.chat_history.append( + f"User: I got a simulation error. Here is the log:\n{combined_text}" + ) + self.debug_ollama() \ No newline at end of file