diff --git a/py/ai.py b/py/ai.py index f78b3b7..a8c5537 100644 --- a/py/ai.py +++ b/py/ai.py @@ -1,6 +1,5 @@ from mistralai import Mistral from openai import OpenAI -import google.generativeai as genai import anthropic import ollama @@ -73,31 +72,3 @@ class AI: for text in stream.text_stream: with return_class.ai_response_lock: return_class.ai_response[access_token] += text - - @staticmethod - def process_google(model, messages, return_class, access_token, api_key): - - message = messages[-1]['content'] - messages.pop() - - system = None - if messages and messages[0]['role'] == 'system': - system = messages[0]['content'] - messages.pop(0) - - for msg in messages: - msg['parts'] = msg.pop('content') - - for msg in messages: - if msg['role'] == 'assistant': - msg['role'] = 'model' - - genai.configure(api_key=api_key) - model = genai.GenerativeModel("gemini-1.5-flash") - chat = model.start_chat( - system_instruction=system, - history=[ - {"role": "user", "parts": "Hello"}, - {"role": "model", "parts": "Great to meet you. What would you like to know?"}, - ] - ) diff --git a/py/api.py b/py/api.py index 6c2d243..be44561 100644 --- a/py/api.py +++ b/py/api.py @@ -42,22 +42,19 @@ class API: return jsonify({'status': 200}) elif model_type == "mistral": api_key = data.get('api_key') - thread = threading.Thread(target=self.ai.process_mistralai, - args=(ai_model, messages, self, access_token, api_key)) + thread = threading.Thread(target=self.ai.process_mistralai, args=(ai_model, messages, self, access_token, api_key)) thread.start() thread.join() return jsonify({'status': 200}) elif model_type == "openai": api_key = data.get('api_key') - thread = threading.Thread(target=self.ai.process_openai, - args=(ai_model, messages, self, access_token, api_key)) + thread = threading.Thread(target=self.ai.process_openai, args=(ai_model, messages, self, access_token, api_key)) thread.start() thread.join() return jsonify({'status': 200}) elif model_type == "anthropic": api_key = data.get('api_key') - thread = threading.Thread(target=self.ai.process_anthropic, - args=(ai_model, messages, self, access_token, api_key)) + thread = threading.Thread(target=self.ai.process_anthropic, args=(ai_model, messages, self, access_token, api_key)) thread.start() thread.join() return jsonify({'status': 200}) @@ -121,7 +118,3 @@ class API: ssl_context = ("cert.pem", "key.pem") self.app.run(debug=True, host='0.0.0.0', port=5000) - - -api = API() -api.run() diff --git a/py/requirements.txt b/py/requirements.txt index bd93ad7..8a79dc1 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -4,4 +4,5 @@ ollama mistralai openai anthropic -pyOpenSSL \ No newline at end of file +pyOpenSSL +pywebview \ No newline at end of file diff --git a/py/run.py b/py/run.py new file mode 100644 index 0000000..612d72e --- /dev/null +++ b/py/run.py @@ -0,0 +1,7 @@ +import os +import webview + +os.system("python api.py") +webview.create_window('Hello world', 'http://localhost:3000') +webview.start() + diff --git a/py/webapp.py b/py/webapp.py new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/py/webapp.py @@ -0,0 +1 @@ +