ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2024-12-20 04:47:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced API

Browse Source
This commit is contained in:
saloui 2023-12-21 19:02:00 +01:00
parent e042ef49d9
commit 2efca58e57
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/__init__.py
View File

@ -255,11 +255,13 @@ class LoLLMsAPI(LollmsApplication):
@socketio.on('take_picture') @socketio.on('take_picture')
def take_picture(): def take_picture():
try: try:
self.info("Loading camera")
if not PackageManager.check_package_installed("cv2"): if not PackageManager.check_package_installed("cv2"):
PackageManager.install_package("opencv-python") PackageManager.install_package("opencv-python")
import cv2 import cv2
cap = cv2.VideoCapture(0) cap = cv2.VideoCapture(0)
n = time.time() n = time.time()
self.info("Stand by for taking a shot in 2s")
while(time.time()-n<2): while(time.time()-n<2):
_, frame = cap.read() _, frame = cap.read()
_, frame = cap.read() _, frame = cap.read()
@ -273,14 +275,19 @@ class LoLLMsAPI(LollmsApplication):
try: try:
cv2.imwrite(str(save_path), frame) cv2.imwrite(str(save_path), frame)
if not self.personality.processor is None: if not self.personality.processor is None:
self.info("Sending file to scripted persona")
self.personality.processor.add_file(save_path, partial(self.process_chunk, client_id = request.sid)) self.personality.processor.add_file(save_path, partial(self.process_chunk, client_id = request.sid))
# File saved successfully # File saved successfully
socketio.emit('picture_taken', {'status':True, 'progress': 100}) socketio.emit('picture_taken', {'status':True, 'progress': 100})
self.info("File sent to scripted persona")
else: else:
self.info("Sending file to persona")
self.personality.add_file(save_path, partial(self.process_chunk, client_id = request.sid)) self.personality.add_file(save_path, partial(self.process_chunk, client_id = request.sid))
# File saved successfully # File saved successfully
socketio.emit('picture_taken', {'status':True, 'progress': 100}) socketio.emit('picture_taken', {'status':True, 'progress': 100})
self.info("File sent to persona")
except Exception as e: except Exception as e:
trace_exception(e)
# Error occurred while saving the file # Error occurred while saving the file
socketio.emit('picture_taken', {'status':False, 'error': str(e)}) socketio.emit('picture_taken', {'status':False, 'error': str(e)})