sageTheDm/interstellar_ai
1
0
Fork 0
forked from React-Group/interstellar_ai
Code Pull requests Activity

Merge branch 'main' into main

Browse source
This commit is contained in:
sageTheDm 2024-10-08 16:50:00 +02:00
commit 9ea352cb4c
3 changed files with 121 additions and 57 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
app/components/History.tsx
View file

@ -3,11 +3,30 @@ import { useChatHistory } from '../hooks/useChatHistory';
const History: React.FC = () => {
const [chatHistory, setChatHistory, setSelectedIndex] = useChatHistory()
const [isEditing, setIsEditing] = useState(false);
const [inputValue, setInputValue] = useState<string>('');
const handleEditButtonClick = () => {
setIsEditing(true);
};
const handleInputChange = (e: React.ChangeEvent<HTMLInputElement>) => {
setInputValue(e.target.value);
};
const handleSaveButtonClick = () => {
setIsEditing(false);
chatHistory.chats.push({ name: inputValue, messages: [], timestamp: 5 })
setInputValue("")
};
const handleHistoryClick = (index: number) => {
setSelectedIndex(index)
console.log("index",index);
}
return (
<div className="history-background">
<div className="history">
@ -18,6 +37,21 @@ const History: React.FC = () => {
<a href="#" onClick={() => handleHistoryClick(index)}>{chatHistory.chats[index].name}</a>
</li>
))}
<li>
{isEditing ? (
<div>
<input
type="text"
value={inputValue}
onChange={handleInputChange}
placeholder="Enter text"
/>
<button onClick={handleSaveButtonClick}>Save</button>
</div>
) : (
<button onClick={handleEditButtonClick}>{'New Chat'}</button>
)}
</li>
</ul>
</div>
</div>

3
app/hooks/useChatHistory.tsx
View file

@ -19,7 +19,8 @@ interface GlobalChatHistory {
let globalChatHistory: GlobalChatHistory = {
chats: [
{ name: "Chat 1", messages: [], timestamp: 4 }
{ name: "Chat 1", messages: [{role:"system",content:"you are a helpful assistant"},{role:"assistant",content:"how can i help you"}], timestamp: 4 },
{ name: "Chat 2", messages: [{role:"system",content:"you are a helpful assistant"},{role:"assistant",content:"how can i help you"}], timestamp: 4 },
],
selectedIndex:0
}

141
py/api.py
View file

@ -9,7 +9,7 @@ from db import DB
from weather import Weather
from voice import VoiceRecognition
from tts import TTS
class API:
def __init__(self):
@ -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()