forked from React-Group/interstellar_ai
removed security to "fix" a bug.
This commit is contained in:
parent
6aabfd52d4
commit
6baa3d5fdb
1 changed files with 86 additions and 57 deletions
143
py/api.py
143
py/api.py
|
@ -9,11 +9,11 @@ from db import DB
|
|||
from weather import Weather
|
||||
from voice import VoiceRecognition
|
||||
from tts import TTS
|
||||
|
||||
|
||||
|
||||
class API:
|
||||
def __init__(self):
|
||||
self.crypt_size = 4096
|
||||
self.crypt_size = 64
|
||||
self.app = Flask(__name__)
|
||||
self.ai_response = {}
|
||||
self.ai = AI()
|
||||
|
@ -26,7 +26,7 @@ class API:
|
|||
CORS(self.app)
|
||||
|
||||
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():
|
||||
access_token = secrets.token_urlsafe(self.crypt_size)
|
||||
|
||||
|
@ -34,112 +34,141 @@ class API:
|
|||
access_token = secrets.token_urlsafe(self.crypt_size)
|
||||
|
||||
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():
|
||||
data = request.get_json()
|
||||
messages = data.get('messages')
|
||||
model_type = data.get('model_type')
|
||||
ai_model = data.get('ai_model')
|
||||
access_token = data.get('access_token')
|
||||
messages = data.get("messages")
|
||||
model_type = data.get("model_type")
|
||||
ai_model = data.get("ai_model")
|
||||
access_token = data.get("access_token")
|
||||
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":
|
||||
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.join()
|
||||
sleep(0.5)
|
||||
return jsonify({'status': 200})
|
||||
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))
|
||||
api_key = data.get("api_key")
|
||||
thread = threading.Thread(
|
||||
target=self.ai.process_mistralai,
|
||||
args=(ai_model, messages, self, access_token, api_key),
|
||||
)
|
||||
thread.start()
|
||||
thread.join()
|
||||
sleep(0.5)
|
||||
return jsonify({'status': 200})
|
||||
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))
|
||||
api_key = data.get("api_key")
|
||||
thread = threading.Thread(
|
||||
target=self.ai.process_openai,
|
||||
args=(ai_model, messages, self, access_token, api_key),
|
||||
)
|
||||
thread.start()
|
||||
thread.join()
|
||||
sleep(0.5)
|
||||
return jsonify({'status': 200})
|
||||
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))
|
||||
api_key = data.get("api_key")
|
||||
thread = threading.Thread(
|
||||
target=self.ai.process_anthropic,
|
||||
args=(ai_model, messages, self, access_token, api_key),
|
||||
)
|
||||
thread.start()
|
||||
thread.join()
|
||||
sleep(0.5)
|
||||
return jsonify({'status': 200})
|
||||
return jsonify({"status": 200})
|
||||
elif model_type == "google":
|
||||
api_key = data.get('api_key')
|
||||
thread = threading.Thread(target=self.ai.process_google,
|
||||
args=(ai_model, messages, self, access_token, api_key))
|
||||
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()
|
||||
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():
|
||||
data = request.args.get('access_token')
|
||||
data = request.args.get("access_token")
|
||||
if data not in self.ai_response:
|
||||
return jsonify({'status': 401, 'error': 'Invalid access token'})
|
||||
return jsonify({'status': 200, 'response': self.ai_response[data]})
|
||||
return jsonify({"status": 401, "error": "Invalid access token"})
|
||||
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():
|
||||
sent_data = request.get_json()
|
||||
print(sent_data)
|
||||
action = sent_data.get('action')
|
||||
action = sent_data.get("action")
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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":
|
||||
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():
|
||||
audio = request.files.get('audio')
|
||||
audio = request.files.get("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():
|
||||
unit_type = request.args.get('unit_type')
|
||||
city = request.args.get('city')
|
||||
return jsonify({'status': 200, 'response': self.weather.getweather(unit_type, city)})
|
||||
unit_type = request.args.get("unit_type")
|
||||
city = request.args.get("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():
|
||||
text = request.args.get('text')
|
||||
return jsonify({'status': 200, 'response': self.tts.gen_tts(text)})
|
||||
text = request.args.get("text")
|
||||
return jsonify({"status": 200, "response": self.tts.gen_tts(text)})
|
||||
|
||||
|
||||
api = API()
|
||||
|
|
Loading…
Reference in a new issue