#!/usr/bin/env python3
"""
FGD Stack Pro GUI – Grok-Only by Default + Safe LLM Switching
"""
from PyQt6.QtWidgets import (
QApplication,
QWidget,
QVBoxLayout,
QHBoxLayout,
QPushButton,
QLabel,
QLineEdit,
QComboBox,
QTextEdit,
QFileDialog,
QMessageBox,
QGroupBox,
)
from PyQt6.QtCore import QTimer, Qt
from PyQt6.QtGui import QFont, QTextCursor
from pathlib import Path
import sys
import yaml
import subprocess
import os
from dotenv import load_dotenv
# Load .env from project root
load_dotenv()
class FGDGUI(QWidget):
def __init__(self):
super().__init__()
self.setWindowTitle("FGD Stack Pro v2.4")
self.resize(960, 720)
self.layout = QVBoxLayout()
self.layout.setSpacing(18)
self.layout.setContentsMargins(28, 28, 28, 28)
self.setLayout(self.layout)
self.process = None
self.log_file = None
header = QLabel("FGD Stack Pro")
header.setAlignment(Qt.AlignmentFlag.AlignCenter)
header.setObjectName("header")
header.setFont(QFont("Segoe UI", 20, QFont.Weight.Bold))
self.layout.addWidget(header)
# Directory
dir_group = QGroupBox("Project Directory")
dir_layout = QVBoxLayout()
self.path_edit = QLineEdit()
browse = QPushButton("Browse")
browse.clicked.connect(self.browse)
h = QHBoxLayout()
h.addWidget(self.path_edit)
h.addWidget(browse)
dir_layout.addLayout(h)
dir_group.setLayout(dir_layout)
self.layout.addWidget(dir_group)
# Provider
provider_group = QGroupBox("LLM Provider")
provider_layout = QVBoxLayout()
self.provider = QComboBox()
self.provider.addItems(["grok", "openai", "claude", "ollama"])
self.provider.setCurrentText("grok") # GROK DEFAULT
provider_layout.addWidget(self.provider)
provider_group.setLayout(provider_layout)
self.layout.addWidget(provider_group)
# Start/Stop
control_group = QGroupBox("Server Control")
control_layout = QVBoxLayout()
self.start_btn = QPushButton("Start Server")
self.start_btn.clicked.connect(self.toggle_server)
control_layout.addWidget(self.start_btn)
self.status = QLabel("Status: Ready")
self.status.setObjectName("status")
control_layout.addWidget(self.status)
control_group.setLayout(control_layout)
self.layout.addWidget(control_group)
# Logs
logs_group = QGroupBox("Live Logs")
logs_layout = QVBoxLayout()
self.log_view = QTextEdit()
self.log_view.setReadOnly(True)
self.log_view.setFont(QFont("Courier", 10))
filters = QHBoxLayout()
self.level = QComboBox()
self.level.addItems(["All", "INFO", "WARNING", "ERROR"])
filters.addWidget(QLabel("Level:"))
filters.addWidget(self.level)
self.search = QLineEdit()
self.search.setPlaceholderText("Search...")
filters.addWidget(self.search)
clear = QPushButton("Clear")
clear.clicked.connect(self.clear_filters)
filters.addWidget(clear)
logs_layout.addLayout(filters)
logs_layout.addWidget(self.log_view)
logs_group.setLayout(logs_layout)
self.layout.addWidget(logs_group)
# Timer
self.timer = QTimer()
self.timer.timeout.connect(self.update_logs)
self.timer.start(1000)
self.apply_dark_mode(True)
def browse(self):
dir_path = QFileDialog.getExistingDirectory(self, "Select Project")
if dir_path:
self.path_edit.setText(dir_path)
def toggle_server(self):
if self.process and self.process.poll() is None:
self.process.terminate()
try:
self.process.wait(timeout=5)
except subprocess.TimeoutExpired:
self.process.kill()
self.process = None
self.status.setText("Status: Stopped")
self.start_btn.setText("Start Server")
else:
self.start_server()
def start_server(self):
dir_path = self.path_edit.text().strip()
if not dir_path or not Path(dir_path).exists():
self.status.setText("Status: Invalid directory")
return
provider = self.provider.currentText()
# BLOCK NON-GROK IF KEY MISSING
if provider != "grok":
key_map = {
"openai": "OPENAI_API_KEY",
"claude": "ANTHROPIC_API_KEY"
}
key_name = key_map.get(provider)
if key_name and not os.getenv(key_name):
reply = QMessageBox.question(
self, "Missing API Key",
f"{key_name} not found in .env\n\nUse Grok instead?",
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
)
if reply == QMessageBox.StandardButton.Yes:
self.provider.setCurrentText("grok")
provider = "grok"
else:
self.status.setText("Status: Start cancelled")
return
# Generate config
self.log_file = Path(dir_path) / "fgd_server.log"
self.log_file.write_text("")
config = {
"watch_dir": dir_path,
"memory_file": str(Path(dir_path) / ".fgd_memory.json"),
"log_file": str(self.log_file),
"context_limit": 20,
"scan": {"max_dir_size_gb": 2, "max_files_per_scan": 5, "max_file_size_kb": 250},
"llm": {
"default_provider": provider,
"providers": {
"grok": {"model": "grok-beta", "base_url": "https://api.x.ai/v1"},
"openai": {"model": "gpt-4o-mini", "base_url": "https://api.openai.com/v1"},
"claude": {"model": "claude-3-5-sonnet-20241022", "base_url": "https://api.anthropic.com/v1"},
"ollama": {"model": "llama3", "base_url": "http://localhost:11434/v1"}
}
}
}
config_path = Path(dir_path) / "fgd_config.yaml"
config_path.write_text(yaml.dump(config))
# Start subprocess with env
env = os.environ.copy()
try:
self.process = subprocess.Popen(
[sys.executable, "mcp_backend.py", str(config_path)],
cwd=dir_path,
env=env
)
except Exception as exc:
self.status.setText(f"Status: Failed to start ({exc})")
self.start_btn.setText("Start Server")
self.process = None
return
self.status.setText(f"Status: Running ({provider})")
self.start_btn.setText("Stop Server")
def update_logs(self):
if self.process and self.process.poll() is not None:
self.status.setText("Status: Server stopped")
self.start_btn.setText("Start Server")
self.process = None
if not self.log_file or not self.log_file.exists():
return
try:
lines = self.log_file.read_text().splitlines()
level = self.level.currentText()
search = self.search.text().lower()
filtered = []
for line in lines:
if level != "All" and level not in line:
continue
if search and search not in line.lower():
continue
filtered.append(line)
if len(filtered) > 500:
filtered = filtered[-500:]
self.log_view.clear()
for line in filtered:
cursor = self.log_view.textCursor()
cursor.movePosition(QTextCursor.MoveOperation.End)
self.log_view.setTextCursor(cursor)
if "ERROR" in line:
self.log_view.setTextColor(Qt.GlobalColor.red)
elif "WARNING" in line:
self.log_view.setTextColor(Qt.GlobalColor.yellow)
else:
self.log_view.setTextColor(Qt.GlobalColor.white)
self.log_view.insertPlainText(line + "\n")
self.log_view.setTextColor(Qt.GlobalColor.white)
except:
pass
def clear_filters(self):
self.level.setCurrentIndex(0)
self.search.clear()
def apply_dark_mode(self, dark):
if dark:
self.setStyleSheet(
"""
QWidget {
background-color: #0d1117;
color: #f0f6fc;
font-family: 'Segoe UI', sans-serif;
}
QGroupBox {
border: 1px solid #30363d;
border-radius: 8px;
margin-top: 10px;
padding: 12px;
font-weight: bold;
color: #f0f6fc;
}
QGroupBox::title {
subcontrol-origin: margin;
left: 12px;
padding: 0 4px 0 4px;
}
QPushButton {
background-color: #238636;
color: #ffffff;
padding: 8px 16px;
border-radius: 6px;
}
QPushButton:hover {
background-color: #2ea043;
}
QPushButton:disabled {
background-color: #161b22;
color: #8b949e;
}
QLineEdit, QComboBox {
background-color: #161b22;
border: 1px solid #30363d;
border-radius: 6px;
padding: 6px 8px;
color: #f0f6fc;
}
QTextEdit {
background-color: #010409;
border: 1px solid #30363d;
border-radius: 8px;
padding: 8px;
}
QLabel#status {
font-size: 14px;
}
QLabel#header {
color: #58a6ff;
}
"""
)
else:
self.setStyleSheet(
"""
QWidget {
background-color: #f7f9fc;
color: #24292f;
font-family: 'Segoe UI', sans-serif;
}
QGroupBox {
border: 1px solid #d0d7de;
border-radius: 8px;
margin-top: 10px;
padding: 12px;
font-weight: bold;
color: #24292f;
}
QGroupBox::title {
subcontrol-origin: margin;
left: 12px;
padding: 0 4px 0 4px;
}
QPushButton {
background-color: #0969da;
color: #ffffff;
padding: 8px 16px;
border-radius: 6px;
}
QPushButton:hover {
background-color: #218bff;
}
QPushButton:disabled {
background-color: #d0d7de;
color: #57606a;
}
QLineEdit, QComboBox {
background-color: #ffffff;
border: 1px solid #d0d7de;
border-radius: 6px;
padding: 6px 8px;
color: #24292f;
}
QTextEdit {
background-color: #ffffff;
border: 1px solid #d0d7de;
border-radius: 8px;
padding: 8px;
}
QLabel#status {
font-size: 14px;
}
QLabel#header {
color: #0969da;
}
"""
)
def closeEvent(self, event):
if self.process:
self.process.terminate()
try:
self.process.wait(timeout=5)
except subprocess.TimeoutExpired:
self.process.kill()
self.process = None
event.accept()
if __name__ == "__main__":
app = QApplication(sys.argv)
win = FGDGUI()
win.show()
sys.exit(app.exec())
