Add files via upload

Author: Maxritz

frontend/config/__init__.py

@@ -0,0 +1,4 @@
+__init__.py
+from .settings import Settings
+
+__all__ = ['Settings']

frontend/config/settings.py

@@ -0,0 +1,31 @@
+# config/settings.py
+import json
+import os
+
+class Settings:
+    def __init__(self):
+        self.config_file = 'config/config.json'
+        self.default_settings = {
+            'api_base': 'http://localhost:11434/api',
+            'default_model': 'llama2-3.2-vision:latest',
+            'temperature': 0.7,
+            'max_tokens': 2000,
+            'db_path': 'data/ollama_gui.db',
+            'rag_settings': {
+                'chunk_size': 1000,
+                'chunk_overlap': 200,
+                'similarity_threshold': 0.7
+            }
+        }
+        self.current_settings = self.load_settings()
+
+    def load_settings(self):
+        if os.path.exists(self.config_file):
+            with open(self.config_file, 'r') as f:
+                return json.load(f)
+        return self.default_settings.copy()
+
+    def save_settings(self):
+        os.makedirs(os.path.dirname(self.config_file), exist_ok=True)
+        with open(self.config_file, 'w') as f:
+            json.dump(self.current_settings, f, indent=4)

frontend/database/__init__.py

@@ -0,0 +1,3 @@
+from .db_manager import DatabaseManager
+
+__all__ = ['DatabaseManager']

frontend/database/db_manager.py

@@ -0,0 +1,56 @@
+# database/db_manager.py
+import sqlite3
+from datetime import datetime
+
+class DatabaseManager:
+    def add_interaction_metrics(self, model_name, prompt, response, tokens_used, response_time, success_rate):
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('''
+                INSERT INTO model_metrics 
+                (timestamp, model_name, prompt_length, response_length, tokens_used, 
+                response_time, success_rate) 
+                VALUES (?, ?, ?, ?, ?, ?, ?)
+            ''', (
+                datetime.now(), model_name, len(prompt), len(response),
+                tokens_used, response_time, success_rate
+            ))
+
+    def init_database(self):
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            
+            # Create tables
+            cursor.execute('''
+                CREATE TABLE IF NOT EXISTS chat_history (
+                    id INTEGER PRIMARY KEY,
+                    timestamp DATETIME,
+                    model TEXT,
+                    message TEXT,
+                    response TEXT,
+                    tokens INTEGER,
+                    response_time FLOAT
+                )
+            ''')
+            
+            cursor.execute('''
+                CREATE TABLE IF NOT EXISTS model_metrics (
+                    id INTEGER PRIMARY KEY,
+                    model TEXT,
+                    date DATE,
+                    total_tokens INTEGER,
+                    avg_response_time FLOAT,
+                    total_conversations INTEGER
+                )
+            ''')
+            
+            conn.commit()
+
+    def add_chat_entry(self, model, message, response, tokens, response_time):
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.cursor()
+            cursor.execute('''
+                INSERT INTO chat_history (timestamp, model, message, response, tokens, response_time)
+                VALUES (?, ?, ?, ?, ?, ?)
+            ''', (datetime.now(), model, message, response, tokens, response_time))
+            conn.commit()

frontend/gui/__init__.py

@@ -0,0 +1,3 @@
+from .app_window import OllamaGUI
+
+__all__ = ['OllamaGUI']

frontend/gui/app_window.py

@@ -0,0 +1,29 @@
+import tkinter as tk
+from tkinter import ttk
+from .frames.model_frame import ModelFrame
+from .frames.chat_frame import ChatFrame
+from .frames.control_frame import ControlFrame
+
+class OllamaGUI:
+    def __init__(self, root):
+        self.root = root
+        self.root.title("Ollama Chat Interface")
+        self.root.geometry("1200x800")
+        
+        # Initialize main container
+        self.main_container = ttk.PanedWindow(root, orient=tk.HORIZONTAL)
+        self.main_container.pack(fill=tk.BOTH, expand=True)
+        
+        # Create frames
+        self.model_frame = ModelFrame(self.main_container, self)  # Pass self as controller
+        self.main_container.add(self.model_frame)
+        
+        # Create right pane
+        self.right_pane = ttk.PanedWindow(self.main_container, orient=tk.VERTICAL)
+        self.main_container.add(self.right_pane)
+        
+        self.chat_frame = ChatFrame(self.right_pane, self)  # Pass self as controller
+        self.right_pane.add(self.chat_frame)
+        
+        self.control_frame = ControlFrame(self.right_pane, self)  # Pass self as controller
+        self.right_pane.add(self.control_frame)

frontend/gui/frames/__init__.py

@@ -0,0 +1,3 @@
+from .chat_frame import ChatFrame
+from .model_frame import ModelFrame
+from .control_frame import ControlFrame