React-Group/interstellar_ai
1
1
Fork 2
Code Issues 8 Pull requests Projects Releases Packages Wiki Activity Actions

removed security to "fix" a bug.

Browse source
This commit is contained in:
Patrick_Pluto 2024-10-08 16:23:21 +02:00
parent 6aabfd52d4
commit 6baa3d5fdb
1 changed files with 86 additions and 57 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

143
py/api.py
View file

@ -9,11 +9,11 @@ from db import DB
from weather import Weather from weather import Weather
from voice import VoiceRecognition from voice import VoiceRecognition
from tts import TTS from tts import TTS
class API: class API:
def __init__(self): def __init__(self):
self.crypt_size = 4096 self.crypt_size = 64
self.app = Flask(__name__) self.app = Flask(__name__)
self.ai_response = {} self.ai_response = {}
self.ai = AI() self.ai = AI()
@ -26,7 +26,7 @@ class API:
CORS(self.app) CORS(self.app)
def run(self): def run(self):
@self.app.route('/interstellar_ai/api/ai_create', methods=['GET']) @self.app.route("/interstellar_ai/api/ai_create", methods=["GET"])
def create_ai(): def create_ai():
access_token = secrets.token_urlsafe(self.crypt_size) access_token = secrets.token_urlsafe(self.crypt_size)
@ -34,112 +34,141 @@ class API:
access_token = secrets.token_urlsafe(self.crypt_size) access_token = secrets.token_urlsafe(self.crypt_size)
self.ai_response[access_token] = "" self.ai_response[access_token] = ""
return jsonify({'status': 200, 'access_token': access_token}) return jsonify({"status": 200, "access_token": access_token})
@self.app.route('/interstellar_ai/api/ai_send', methods=['POST']) @self.app.route("/interstellar_ai/api/ai_send", methods=["POST"])
def send_ai(): def send_ai():
data = request.get_json() data = request.get_json()
messages = data.get('messages') messages = data.get("messages")
model_type = data.get('model_type') model_type = data.get("model_type")
ai_model = data.get('ai_model') ai_model = data.get("ai_model")
access_token = data.get('access_token') access_token = data.get("access_token")
if access_token not in self.ai_response: if access_token not in self.ai_response:
return jsonify({'status': 401, 'error': 'Invalid access token'}) return jsonify({"status": 401, "error": "Invalid access token"})
if model_type == "local": if model_type == "local":
thread = threading.Thread(target=self.ai.process_local, args=(ai_model, messages, self, access_token)) thread = threading.Thread(
target=self.ai.process_local,
args=(ai_model, messages, self, access_token),
)
thread.start() thread.start()
thread.join() thread.join()
sleep(0.5) sleep(0.5)
return jsonify({'status': 200}) return jsonify({"status": 200})
elif model_type == "mistral": elif model_type == "mistral":
api_key = data.get('api_key') api_key = data.get("api_key")
thread = threading.Thread(target=self.ai.process_mistralai, thread = threading.Thread(
args=(ai_model, messages, self, access_token, api_key)) target=self.ai.process_mistralai,
args=(ai_model, messages, self, access_token, api_key),
)
thread.start() thread.start()
thread.join() thread.join()
sleep(0.5) sleep(0.5)
return jsonify({'status': 200}) return jsonify({"status": 200})
elif model_type == "openai": elif model_type == "openai":
api_key = data.get('api_key') api_key = data.get("api_key")
thread = threading.Thread(target=self.ai.process_openai, thread = threading.Thread(
args=(ai_model, messages, self, access_token, api_key)) target=self.ai.process_openai,
args=(ai_model, messages, self, access_token, api_key),
)
thread.start() thread.start()
thread.join() thread.join()
sleep(0.5) sleep(0.5)
return jsonify({'status': 200}) return jsonify({"status": 200})
elif model_type == "anthropic": elif model_type == "anthropic":
api_key = data.get('api_key') api_key = data.get("api_key")
thread = threading.Thread(target=self.ai.process_anthropic, thread = threading.Thread(
args=(ai_model, messages, self, access_token, api_key)) target=self.ai.process_anthropic,
args=(ai_model, messages, self, access_token, api_key),
)
thread.start() thread.start()
thread.join() thread.join()
sleep(0.5) sleep(0.5)
return jsonify({'status': 200}) return jsonify({"status": 200})
elif model_type == "google": elif model_type == "google":
api_key = data.get('api_key') api_key = data.get("api_key")
thread = threading.Thread(target=self.ai.process_google, thread = threading.Thread(
args=(ai_model, messages, self, access_token, api_key)) target=self.ai.process_google,
args=(ai_model, messages, self, access_token, api_key),
)
thread.start() thread.start()
thread.join() thread.join()
sleep(0.5) sleep(0.5)
return jsonify({'status': 200}) return jsonify({"status": 200})
return jsonify({'status': 401, 'error': 'Invalid AI model type'}) return jsonify({"status": 401, "error": "Invalid AI model type"})
@self.app.route('/interstellar_ai/api/ai_get', methods=['GET']) @self.app.route("/interstellar_ai/api/ai_get", methods=["GET"])
def get_ai(): def get_ai():
data = request.args.get('access_token') data = request.args.get("access_token")
if data not in self.ai_response: if data not in self.ai_response:
return jsonify({'status': 401, 'error': 'Invalid access token'}) return jsonify({"status": 401, "error": "Invalid access token"})
return jsonify({'status': 200, 'response': self.ai_response[data]}) return jsonify({"status": 200, "response": self.ai_response[data]})
@self.app.route('/interstellar_ai/db', methods=['POST']) @self.app.route("/interstellar_ai/db", methods=["POST"])
def db_manipulate(): def db_manipulate():
sent_data = request.get_json() sent_data = request.get_json()
print(sent_data) print(sent_data)
action = sent_data.get('action') action = sent_data.get("action")
if action == "create_account": if action == "create_account":
return jsonify({'status': 200, 'response': self.db.add_user(sent_data)}) return jsonify({"status": 200, "response": self.db.add_user(sent_data)})
elif action == "change_password": elif action == "change_password":
return jsonify({'status': 200, 'response': self.db.update_password(sent_data)}) return jsonify(
{"status": 200, "response": self.db.update_password(sent_data)}
)
elif action == "get_settings": elif action == "get_settings":
return jsonify({'status': 200, 'response': self.db.get_settings(sent_data)}) return jsonify(
{"status": 200, "response": self.db.get_settings(sent_data)}
)
elif action == "change_settings": elif action == "change_settings":
return jsonify({'status': 200, 'response': self.db.change_settings(sent_data)}) return jsonify(
{"status": 200, "response": self.db.change_settings(sent_data)}
)
elif action == "get_history": elif action == "get_history":
return jsonify({'status': 200, 'response': self.db.get_history(sent_data)}) return jsonify(
{"status": 200, "response": self.db.get_history(sent_data)}
)
elif action == "change_history": elif action == "change_history":
return jsonify({'status': 200, 'response': self.db.change_history(sent_data)}) return jsonify(
{"status": 200, "response": self.db.change_history(sent_data)}
)
elif action == "check_credentials": elif action == "check_credentials":
return jsonify({'status': 200, 'response': self.db.check_credentials(sent_data)}) return jsonify(
{"status": 200, "response": self.db.check_credentials(sent_data)}
)
elif action == "delete_account": elif action == "delete_account":
return jsonify({'status': 200, 'response': self.db.delete_user(sent_data)}) return jsonify(
{"status": 200, "response": self.db.delete_user(sent_data)}
)
elif action == "get_email": elif action == "get_email":
return jsonify({'status': 200, 'response': self.db.get_email(sent_data)}) return jsonify(
{"status": 200, "response": self.db.get_email(sent_data)}
)
elif action == "get_name": elif action == "get_name":
return jsonify({'status': 200, 'response': self.db.get_name(sent_data)}) return jsonify({"status": 200, "response": self.db.get_name(sent_data)})
return jsonify({'status': 401, 'response': "Invalid action"}) return jsonify({"status": 401, "response": "Invalid action"})
@self.app.route('/interstellar_ai/api/voice_recognition', methods=['POST']) @self.app.route("/interstellar_ai/api/voice_recognition", methods=["POST"])
def voice_recognition(): def voice_recognition():
audio = request.files.get('audio') audio = request.files.get("audio")
text = self.voice.recognition(audio) text = self.voice.recognition(audio)
return jsonify({'status': 200, 'response': text}) return jsonify({"status": 200, "response": text})
@self.app.route('/interstellar_ai/api/weather', methods=['POST']) @self.app.route("/interstellar_ai/api/weather", methods=["POST"])
def get_weather(): def get_weather():
unit_type = request.args.get('unit_type') unit_type = request.args.get("unit_type")
city = request.args.get('city') city = request.args.get("city")
return jsonify({'status': 200, 'response': self.weather.getweather(unit_type, city)}) return jsonify(
{"status": 200, "response": self.weather.getweather(unit_type, city)}
)
self.app.run(debug=True, host='0.0.0.0', port=5000) self.app.run(debug=True, host="0.0.0.0", port=5000)
@self.app.route('/interstellar_ai/api/tts', methods=['POST']) @self.app.route("/interstellar_ai/api/tts", methods=["POST"])
def tts(): def tts():
text = request.args.get('text') text = request.args.get("text")
return jsonify({'status': 200, 'response': self.tts.gen_tts(text)}) return jsonify({"status": 200, "response": self.tts.gen_tts(text)})
api = API() api = API()