include updates after adding sensor

main
vitrinekast 6 months ago
parent a64753bf74
commit 239c6decd1

@ -2,7 +2,7 @@ from flask import Flask, render_template, request, flash, redirect, url_for
from werkzeug.utils import secure_filename from werkzeug.utils import secure_filename
from PIL import Image from PIL import Image
import os import os
import ffmpeg # import ffmpeg
import datetime; import datetime;
@ -54,20 +54,19 @@ def upload_file():
return redirect(url_for('submitted')) return redirect(url_for('submitted'))
elif ext in ['mp4', 'avi', 'mov', 'flv']: elif ext in ['mp4', 'avi', 'mov', 'flv']:
out_path = os.path.join(app.config['UPLOAD_FOLDER'], filename.rsplit('.', 1)[0].lower() + '.mp4') out_path = os.path.join(app.config['UPLOAD_FOLDER'], filename.rsplit('.', 1)[0].lower() + '.mp4')
file.save(temp_path) file.save(out_path)
stream = ffmpeg.input(temp_path) # stream = ffmpeg.input(temp_path)
stream = ffmpeg.hflip(stream) # stream = ffmpeg.hflip(stream)
stream = ffmpeg.output(stream, out_path) # stream = ffmpeg.output(stream, out_path)
ffmpeg.run(stream) # ffmpeg.run(stream)
print("ive outputted somethign")
# ffmpeg.input(temp_path).output(out_path,vf="scale=300:230", crf=24).run() # ffmpeg.input(temp_path).output(out_path,vf="scale=300:230", crf=24).run()
# os.remove(temp_path) # os.remove(temp_path)
return redirect(url_for('submitted')) return redirect(url_for('submitted'))
elif ext in ['mp3', 'wav', 'flac']: elif ext in ['mp3', 'wav', 'flac']:
out_name = filename.rsplit('.', 1)[0].lower() + '.mp3' out_name = filename.rsplit('.', 1)[0].lower() + '.mp3'
file.save(temp_path) file.save(os.path.join(app.config['UPLOAD_FOLDER'], out_name))
ffmpeg.input(temp_path).output(os.path.join(app.config['UPLOAD_FOLDER'], out_name)).run() # ffmpeg.input(temp_path).output(os.path.join(app.config['UPLOAD_FOLDER'], out_name)).run()
os.remove(temp_path) # os.remove(temp_path)
return redirect(url_for('submitted')) return redirect(url_for('submitted'))
else: else:
print("not an image!") print("not an image!")

@ -57,6 +57,13 @@ def resize_and_center_frame(frame):
window_background = np.zeros((WINDOW_HEIGHT, WINDOW_WIDTH, 3), dtype=np.uint8) window_background = np.zeros((WINDOW_HEIGHT, WINDOW_WIDTH, 3), dtype=np.uint8)
# Overlay the resized frame onto the background at the calculated offset # Overlay the resized frame onto the background at the calculated offset
print(y_offset)
print(y_offset)
print(new_height)
print(x_offset)
print(x_offset)
print(new_width)
window_background[ window_background[
y_offset : y_offset + new_height, x_offset : x_offset + new_width y_offset : y_offset + new_height, x_offset : x_offset + new_width
] = resized_frame ] = resized_frame
@ -67,16 +74,21 @@ def resize_and_center_frame(frame):
# Show an image on the canvas # Show an image on the canvas
def showImage(file): def showImage(file):
# Read the image # Read the image
current_image = cv2.imread(MEDIA_FOLDER + "/" + file) current_image = cv2.imread(file)
if current_image is None: if current_image is None:
print(f"Error loading image: {image_path}") print(f"Error loading image: {current_image}")
return return
a = cv2.cvtColor(current_image, cv2.COLOR_BGR2GRAY) a = cv2.cvtColor(current_image, cv2.COLOR_BGR2GRAY)
resized_frame = resize_and_center_frame(current_image)
cv2.imshow(WINDOW_NAME, resized_frame) # resized_frame = resize_and_center_frame(current_image)
cv2.namedWindow(WINDOW_NAME,cv2.WND_PROP_FULLSCREEN) cv2.imshow(WINDOW_NAME, current_image)
cv2.setWindowProperty(WINDOW_NAME, cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN) # cv2.namedWindow(WINDOW_NAME,cv2.WND_PROP_FULLSCREEN)
# cv2.setWindowProperty(WINDOW_NAME, cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)
cv2.waitKey(1)
while True: while True:
for i in range(12): for i in range(12):
@ -87,7 +99,7 @@ def showImage(file):
def play_audio(file): def play_audio(file):
clear_frame() clear_frame()
sound = sa.WaveObject.from_wave_file("media/" + file) sound = sa.WaveObject.from_wave_file(file)
play_obj = sound.play() play_obj = sound.play()
# TODO: check for sensor data change # TODO: check for sensor data change
@ -96,11 +108,20 @@ def play_audio(file):
time.sleep(0.1) time.sleep(0.1)
load_media() load_media()
def get_random_file(previous=None): def get_random_file(f):
random.shuffle(mediaFolder) print("getting random file from ")
file = mediaFolder[0] print(f)
if file == previous:
return get_random_file(previous) folder = os.listdir(f)
random.shuffle(folder)
file = folder[0]
if os.path.isdir(file):
print("oh no u gave me a folder, load the sub instead")
nF = os.listdir(f + "/" + file)
return get_random_file(nF)
return file return file
@ -108,27 +129,34 @@ def get_random_file(previous=None):
def load_media(region=None, override=None): def load_media(region=None, override=None):
global current_file global current_file
print("runningn load_media")
if region: if region:
print("i should now load a media from region ") print("i should now load a media from region ")
print(region) print(region)
else: else:
print("region is not defined, just do something!") print("region is not defined, just do something!")
folder = os.listdir(MEDIA_FOLDER)[region]
print(folder)
if(override): if(override):
file = override file = override
else: else:
file = get_random_file(current_file) file = get_random_file(MEDIA_FOLDER + "/" + folder)
current_file = file current_file = file
file_path = MEDIA_FOLDER + "/" + folder + "/" + file
print(file_path)
# Check if the file is an image or a video # Check if the file is an image or a video
if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"): if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
# Read and resize the image # Read and resize the image
print("ive loaded an image: " + file) print("ive loaded an image: " + file)
showImage(file) showImage(file_path)
elif file.endswith(".mov") or file.endswith(".mp4") or file.endswith(".avi"): elif file.endswith(".mov") or file.endswith(".mp4") or file.endswith(".avi"):
# Create a VideoCapture object and read from input file # Create a VideoCapture object and read from input file
cap = cv2.VideoCapture("media/" + file) cap = cv2.VideoCapture(file_path)
print("ive loaded a video" + file) print("ive loaded a video" + file)
# show the input video file # show the input video file
@ -156,7 +184,7 @@ def load_media(region=None, override=None):
break break
elif file.endswith(".wav") or file.endswith(".mp3"): elif file.endswith(".wav") or file.endswith(".mp3"):
print("ive loaded an audio file: " + file) print("ive loaded an audio file: " + file)
play_audio(file) play_audio(file_path)
else: else:
print("Unsupported file format: " + file) print("Unsupported file format: " + file)
@ -164,4 +192,4 @@ def load_media(region=None, override=None):
print("start script ") print("start script ")
load_media() load_media(2)

Loading…
Cancel
Save