Near to complete

Author: sudo-junaiddev

README.md

@@ -1,19 +1,115 @@
 # Simple AI UI
-Simple AI UI is a simple , minimalist and lightweight web GUI. It offers an server also to host your own AI in your local network or on Internet.
 
-### Features
-- Robust API infrastructure.
-- Simple , minimalist and lightweight UI
-- Privacy (Your data is stored in your computer)
+Simple AI UI is a minimalist and lightweight web GUI that allows you to host your own AI on your local network or the internet.
+
+---
+
+## Table of Contents
+
+1. [Features](#features)
+2. [Supported Platforms](#supported-platforms)
+3. [Flavors](#flavors)
+4. [APIs](#apis)
+5. [Server Configuration](#server-configuration)
+6. [Installation](#installation)
+7. [Credits](#credits)
+8. [License](#license)
+
+---
+
+## Features
+
+- Robust API infrastructure
+- Simple, minimalist, and lightweight UI
+- Privacy: Your data is stored on your computer
+- Chat History - [COMING SOON]
+- Memory for AI - [COMING SOON]
+- Admin Panel - [COMING SOON]
+- More APIs - [COMING SOON]
+
+---
+
+## Supported Platforms
 
-### Supported Platforms for Hosting
 - Windows
 - Mac
 - Linux
-**NOTE :** Ollama must be installed on the system.
 
-### Flavors
-- Python
-- Rust-lang (coming soon)
+**NOTE:** Ollama must be installed on the system.
+
+---
+
+## Flavors
+
+- **Python** - Code Name: Simple-AI-UI-EASY
+- **Rust-lang** - Code Name: Simple-AI-UI-ThunderBOLT [COMING SOON]
+
+---
+
+## APIs
+
+For detailed documentation, refer to `./docs/apis.md`.
+
+---
+
+## Server Configuration
+
+For detailed documentation, refer to `./docs/server-config.md`.
+
+---
+
+## Installation
+
+### For Python Lovers and Enthusiasts:
+
+1. **Clone the repository:**
+   ```bash
+   git clone https://github.com/junaidcodingmaster/SIMPLE-AI-UI
+   cd SIMPLE-AI-UI
+   ```
+
+2. **Create a virtual environment:**
+   ```bash
+   python3 -m venv venv
+   ```
+
+3. **Activate the virtual environment:**
+
+   **Windows:**
+   ```bash
+   .\venv\Scripts\activate
+   ```
+
+   **Linux or Mac:**
+   ```bash
+   source ./venv/bin/activate
+   ```
+
+4. **Install requirements:**
+   ```bash
+   pip install -r ./requirements.txt
+   ```
+
+5. **Run the application:**
+   ```bash
+   python app.py
+   ```
+
+### For Windows EXE - Coming Soon
+
+### For Linux Binaries - Coming Soon
+
+---
+
+## Credits
+Made By [Junaid](https://abujuni.dev)
+
+---
+
+## License
 
+This project is licensed under the MIT License. See the `LICENSE` file for more details.
 
+## Alert
+- Currently this project is not opensource.
+- I promise Soon it will be opensource.

app.py

@@ -3,7 +3,55 @@
 import threading
 import queue
 from ollamaClient import OllamaClient
+from dotenv import load_dotenv
 
+import hashlib
+
+# Load environment variables from .env file
+
+load_dotenv("./config.env")
+
+
+# Helper function to get configuration values
+
+def get_config(key, default=None):
+
+    value = os.getenv(key, default)
+
+    if value is None:
+
+        print(f"Error: The {key} is not set in the environment.")
+
+        exit(1)  # Exit with non-zero status to indicate an error occurred
+
+    return value
+
+
+# Configuration
+
+config = {
+
+    "host": get_config("SIMPLE_AI_UI_SERVER_HOST", "0.0.0.0"),
+
+    "port": int(get_config("SIMPLE_AI_UI_SERVER_PORT", "5000")),
+
+    "users": {
+
+        pair.split(":")[0]: pair.split(":")[1]
+
+        for pair in get_config("SIMPLE_AI_UI_AUTH_USERS", "joe:1234,apple:1234").split(",")
+
+        if ":" in pair
+
+    },
+
+    "open_stats": get_config("SIMPLE_AI_UI_AUTH_SHOW_USER_STATS", "true").lower() == "true",
+
+    "limit_of_requests": int(get_config("SIMPLE_AI_UI_API_LIMIT_OF_REQS_PER_DAY", "1000")),
+
+    "limit_of_devices": int(get_config("SIMPLE_AI_UI_API_NO_OF_DEVICES", "1000")),
+
+}
 # Initialize Flask app and Ollama client
 app = Flask(__name__)
 ai = OllamaClient()
@@ -33,6 +81,39 @@ def process_requests() -> None:
 # Start background worker thread
 threading.Thread(target=process_requests, daemon=True).start()
 
+def auth(username, password):
+    for user in config.get("users").keys():
+        if user == username:
+            if config.get("users").get(user) == password:
+                return "OK-12"
+            else:
+                return "BAD-2"
+        else:
+            return "BAD-12"
+
+def hashing(text):
+    content = str(uuid4)+text+str(uuid4)
+    hash_object = hashlib.sha256(content.encode("utf-8"))
+    hex_dig = hash_object.hexdigest()
+    return hex_dig
+
+def middlewares(username,password):
+    auth_status=auth(username, password)
+    if auth_status == "OK-12":
+        hash_obj=hashing(username+password)
+        request.cookies.add("auth":hash_obj)
+        ALLOW_HASHS.append(hash_obj)
+        return True
+    else: 
+        return render_template("error.html", error="YOU ARE NOT ALLOWED - AUTH ERROR")
+
+def api_chat_func():
+    data = request.json
+    response_queue: queue.Queue = queue.Queue()
+    request_queue.put((data, response_queue))
+    response = response_queue.get()  # Blocking wait for response
+    status = 200 if "response" in response else 400
+    return jsonify(response), status
 
 # Routes
 @app.route("/")
@@ -86,12 +167,7 @@ def api_connection_stats():
 
 @app.route("/api/chat", methods=["POST"])
 def api_chat():
-    data = request.json
-    response_queue: queue.Queue = queue.Queue()
-    request_queue.put((data, response_queue))
-    response = response_queue.get()  # Blocking wait for response
-    status = 200 if "response" in response else 400
-    return jsonify(response), status
+    middlewares_firewall=middlewares()
 
 
 if __name__ == "__main__":

config.env

@@ -0,0 +1,12 @@
+# Server Configuration
+SIMPLE_AI_UI_SERVER_HOST=localhost
+SIMPLE_AI_UI_SERVER_PORT=5000                   
+
+# Users
+SIMPLE_AI_UI_AUTH_USERS=joe:1234,apple:1234
+SIMPLE_AI_UI_AUTH_NO_OF_USERS_PER_DAY=1000
+SIMPLE_AI_UI_AUTH_SHOW_USER_STATS=true      
+
+# API Requests
+SIMPLE_AI_UI_API_LIMIT_OF_REQS_PER_DAY=1000     
+SIMPLE_AI_UI_API_NO_OF_DEVICES=1000             

docs/apis.md

@@ -0,0 +1,90 @@
+# API Documentation
+
+## Index
+1. [Base URL](#base-url)
+2. [Endpoints](#endpoints)
+   - [Check Connection](#1-check-connection)
+   - [Get Connection Stats](#2-get-connection-stats)
+   - [Chat with AI](#3-chat-with-ai)
+3. [Notes](#notes)
+
+---
+
+## Base URL
+```
+http://localhost:5000
+```
+
+---
+
+## Endpoints
+
+### 1. Check Connection
+**Endpoint:** `GET /api/connection`
+
+**Description:** Checks the connection to the local Ollama server.
+
+**Response:**
+- **200 OK** - Server is reachable.
+- **404 ERROR** - Server is unreachable.
+
+**Example Response:**
+```json
+{
+  "host": "localhost",
+  "port": 11434,
+  "status": "OK"
+}
+```
+
+---
+
+### 2. Get Connection Stats
+**Endpoint:** `GET /api/connection/stats`
+
+**Description:** Returns available and active AI models.
+
+**Response:**
+- **200 OK** - Returns list of models.
+- **404 ERROR** - No models found.
+
+**Example Response:**
+```json
+{
+  "available": ["model1", "model2"],
+  "active": ["model1"]
+}
+```
+
+---
+
+### 3. Chat with AI
+**Endpoint:** `POST /api/chat`
+
+**Description:** Sends a chat request to the AI model and returns a response. Ensure that the selected model is available before making a request.
+
+**Request Body:**
+```json
+{
+  "model": "model_name",
+  "prompt": "Hello AI!"
+}
+```
+
+**Response:**
+- **200 OK** - Successful response from AI.
+- **400 ERROR** - Invalid request data.
+
+**Example Response:**
+```json
+{
+  "response": "Hello, how can I assist you today?"
+}
+```
+
+---
+
+## Notes
+- The Ollama server is expected to be running on `http://localhost:11434`.
+- Requests to `/api/chat` are processed asynchronously using a queue.
+