claudecode/AvianVisitors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reshuffle code to make server.py importable

Browse Source
This commit is contained in:
frederik
2023-12-23 10:46:44 +01:00
parent 563ee57e19
commit 3df8290e67
1 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/server.py
+16 -11
View File
@@ -37,16 +37,19 @@ DISCONNECT_MESSAGE = "!DISCONNECT"
userDir = os.path.expanduser('~') userDir = os.path.expanduser('~')
DB_PATH = userDir + '/BirdNET-Pi/scripts/birds.db' DB_PATH = userDir + '/BirdNET-Pi/scripts/birds.db'
INTERPRETER, INCLUDE_LIST, EXCLUDE_LIST = (None, None, None)
PREDICTED_SPECIES_LIST = [] PREDICTED_SPECIES_LIST = []
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
server.bind(ADDR) def bind_port():
except BaseException: try:
print("Waiting on socket") server.bind(ADDR)
time.sleep(5) except BaseException:
print("Waiting on socket")
time.sleep(5)
# Open most recent Configuration and grab DB_PWD as a python variable # Open most recent Configuration and grab DB_PWD as a python variable
@@ -271,9 +274,11 @@ def predict(sample, sensitivity):
return p_sorted[:human_cutoff] return p_sorted[:human_cutoff]
def analyzeAudioData(chunks, lat, lon, week, sensitivity, overlap,): def analyzeAudioData(chunks, lat, lon, week, sens, overlap,):
global INTERPRETER global INTERPRETER
sensitivity = max(0.5, min(1.0 - (sens - 1.0), 1.5))
detections = {} detections = {}
start = time.time() start = time.time()
print('ANALYZING AUDIO...', end=' ', flush=True) print('ANALYZING AUDIO...', end=' ', flush=True)
@@ -454,10 +459,8 @@ def handle_client(conn, addr):
week_number = int(now.strftime("%V")) week_number = int(now.strftime("%V"))
week = max(1, min(week_number, 48)) week = max(1, min(week_number, 48))
sensitivity = max(0.5, min(1.0 - (args.sensitivity - 1.0), 1.5))
# Process audio data and get detections # Process audio data and get detections
detections = analyzeAudioData(audioData, args.lat, args.lon, week, sensitivity, args.overlap) detections = analyzeAudioData(audioData, args.lat, args.lon, week, args.sensitivity, args.overlap)
# Write detections to output file # Write detections to output file
min_conf = max(0.01, min(args.min_conf, 0.99)) min_conf = max(0.01, min(args.min_conf, 0.99))
@@ -619,8 +622,9 @@ def handle_client(conn, addr):
def start(): def start():
bind_port()
# Load model # Load model
global INTERPRETER, INCLUDE_LIST, EXCLUDE_LIST global INTERPRETER
INTERPRETER = loadModel() INTERPRETER = loadModel()
server.listen() server.listen()
# print(f"[LISTENING] Server is listening on {SERVER}") # print(f"[LISTENING] Server is listening on {SERVER}")
@@ -632,4 +636,5 @@ def start():
# print("[STARTING] server is starting...") # print("[STARTING] server is starting...")
start() if __name__ == '__main__':
start()