interstellar_ai/py/api.py

178 lines
6.7 KiB
Python
Raw Normal View History

2024-10-01 08:18:14 +02:00
from time import sleep
2024-09-27 13:59:27 +02:00
from flask import Flask, request, jsonify
2024-09-19 12:46:52 +02:00
from flask_cors import CORS
import secrets
2024-09-23 11:01:39 +02:00
import threading
2024-09-20 09:03:46 +02:00
from ai import AI
2024-09-20 11:15:42 +02:00
from db import DB
2024-09-24 14:18:57 +02:00
from weather import Weather
from voice import VoiceRecognition
2024-09-25 16:34:02 +02:00
from tts import TTS
2024-10-08 16:23:21 +02:00
2024-09-19 09:41:00 +02:00
class API:
def __init__(self):
2024-10-08 16:23:21 +02:00
self.crypt_size = 64
2024-09-19 09:41:00 +02:00
self.app = Flask(__name__)
2024-09-19 12:46:52 +02:00
self.ai_response = {}
2024-09-19 09:41:00 +02:00
self.ai = AI()
2024-09-20 11:15:42 +02:00
self.db = DB()
2024-09-24 14:18:57 +02:00
self.weather = Weather()
self.voice = VoiceRecognition()
2024-09-25 16:34:02 +02:00
self.tts = TTS()
2024-09-20 16:06:40 +02:00
self.db.load_database()
2024-09-23 11:01:39 +02:00
self.ai_response_lock = threading.Lock()
2024-09-19 12:46:52 +02:00
CORS(self.app)
2024-09-19 09:41:00 +02:00
def run(self):
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/ai_create", methods=["GET"])
2024-09-19 09:41:00 +02:00
def create_ai():
2024-09-20 14:40:16 +02:00
access_token = secrets.token_urlsafe(self.crypt_size)
2024-09-24 14:18:57 +02:00
2024-09-24 16:42:36 +02:00
while access_token in self.ai_response:
access_token = secrets.token_urlsafe(self.crypt_size)
2024-09-24 14:18:57 +02:00
2024-09-24 16:42:36 +02:00
self.ai_response[access_token] = ""
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200, "access_token": access_token})
2024-09-19 09:41:00 +02:00
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/ai_send", methods=["POST"])
2024-09-19 09:41:00 +02:00
def send_ai():
data = request.get_json()
2024-10-08 16:23:21 +02:00
messages = data.get("messages")
model_type = data.get("model_type")
ai_model = data.get("ai_model")
access_token = data.get("access_token")
2024-10-09 22:04:06 +02:00
print(model_type)
2024-09-19 12:46:52 +02:00
if access_token not in self.ai_response:
2024-10-08 16:23:21 +02:00
return jsonify({"status": 401, "error": "Invalid access token"})
if model_type == "local":
2024-10-08 16:23:21 +02:00
thread = threading.Thread(
target=self.ai.process_local,
args=(ai_model, messages, self, access_token),
)
2024-09-23 11:01:39 +02:00
thread.start()
thread.join()
2024-10-01 08:18:14 +02:00
sleep(0.5)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200})
2024-09-23 11:01:39 +02:00
elif model_type == "mistral":
2024-10-09 22:04:06 +02:00
print(model_type)
2024-10-08 16:23:21 +02:00
api_key = data.get("api_key")
thread = threading.Thread(
target=self.ai.process_mistralai,
args=(ai_model, messages, self, access_token, api_key),
)
2024-09-23 11:01:39 +02:00
thread.start()
thread.join()
2024-10-01 08:18:14 +02:00
sleep(0.5)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200})
2024-09-23 11:57:16 +02:00
elif model_type == "openai":
2024-10-08 16:23:21 +02:00
api_key = data.get("api_key")
thread = threading.Thread(
target=self.ai.process_openai,
args=(ai_model, messages, self, access_token, api_key),
)
2024-09-23 11:57:16 +02:00
thread.start()
thread.join()
2024-10-01 08:18:14 +02:00
sleep(0.5)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200})
2024-09-23 11:57:16 +02:00
elif model_type == "anthropic":
2024-10-08 16:23:21 +02:00
api_key = data.get("api_key")
thread = threading.Thread(
target=self.ai.process_anthropic,
args=(ai_model, messages, self, access_token, api_key),
)
2024-09-23 11:57:16 +02:00
thread.start()
thread.join()
2024-10-01 08:18:14 +02:00
sleep(0.5)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200})
elif model_type == "google":
2024-10-08 16:23:21 +02:00
api_key = data.get("api_key")
thread = threading.Thread(
target=self.ai.process_google,
args=(ai_model, messages, self, access_token, api_key),
)
thread.start()
thread.join()
2024-10-01 08:18:14 +02:00
sleep(0.5)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200})
2024-09-23 11:01:39 +02:00
2024-10-08 16:23:21 +02:00
return jsonify({"status": 401, "error": "Invalid AI model type"})
2024-09-19 09:41:00 +02:00
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/ai_get", methods=["GET"])
2024-09-19 09:41:00 +02:00
def get_ai():
2024-10-08 16:23:21 +02:00
data = request.args.get("access_token")
2024-09-19 12:46:52 +02:00
if data not in self.ai_response:
2024-10-08 16:23:21 +02:00
return jsonify({"status": 401, "error": "Invalid access token"})
return jsonify({"status": 200, "response": self.ai_response[data]})
2024-09-19 09:41:00 +02:00
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/db", methods=["POST"])
2024-09-20 11:15:42 +02:00
def db_manipulate():
2024-09-30 15:26:31 +02:00
sent_data = request.get_json()
2024-10-08 16:23:21 +02:00
action = sent_data.get("action")
2024-09-20 11:15:42 +02:00
if action == "create_account":
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200, "response": self.db.add_user(sent_data)})
elif action == "change_password":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.update_password(sent_data)}
)
2024-10-02 14:14:20 +02:00
elif action == "get_settings":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.get_settings(sent_data)}
)
2024-10-02 14:14:20 +02:00
elif action == "change_settings":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.change_settings(sent_data)}
)
2024-10-02 14:14:20 +02:00
elif action == "get_history":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.get_history(sent_data)}
)
2024-10-02 14:14:20 +02:00
elif action == "change_history":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.change_history(sent_data)}
)
elif action == "check_credentials":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.check_credentials(sent_data)}
)
2024-09-30 09:35:58 +02:00
elif action == "delete_account":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.delete_user(sent_data)}
)
2024-10-01 15:34:43 +02:00
elif action == "get_email":
2024-10-08 16:23:21 +02:00
return jsonify(
{"status": 200, "response": self.db.get_email(sent_data)}
)
2024-10-01 15:34:43 +02:00
elif action == "get_name":
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200, "response": self.db.get_name(sent_data)})
2024-10-08 16:23:21 +02:00
return jsonify({"status": 401, "response": "Invalid action"})
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/voice_recognition", methods=["POST"])
2024-09-24 16:42:36 +02:00
def voice_recognition():
2024-10-08 16:23:21 +02:00
audio = request.files.get("audio")
2024-09-26 11:01:15 +02:00
text = self.voice.recognition(audio)
2024-10-08 16:23:21 +02:00
return jsonify({"status": 200, "response": text})
2024-09-20 11:15:42 +02:00
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/weather", methods=["POST"])
2024-09-24 14:18:57 +02:00
def get_weather():
2024-10-09 18:35:38 +02:00
sent_data = request.get_json()
unit_type = sent_data.get("unit_type")
city = sent_data.get("city")
weather_data = self.weather.getweather(unit_type, city)
return jsonify({"status": 200, "response": weather_data})
2024-09-24 14:18:57 +02:00
2024-10-08 16:23:21 +02:00
self.app.run(debug=True, host="0.0.0.0", port=5000)
2024-09-24 09:24:31 +02:00
2024-10-08 16:23:21 +02:00
@self.app.route("/interstellar_ai/api/tts", methods=["POST"])
2024-09-25 16:34:02 +02:00
def tts():
2024-10-08 16:23:21 +02:00
text = request.args.get("text")
return jsonify({"status": 200, "response": self.tts.gen_tts(text)})
2024-09-25 16:34:02 +02:00
2024-09-24 09:24:31 +02:00
api = API()
api.run()