Rainbow fix
This commit is contained in:
Lasse Studion
parent
227061e44a
commit
6a0897449d
215
app.py
215
app.py
@ -1,9 +1,115 @@
|
||||
from io import StringIO
|
||||
from time import sleep
|
||||
|
||||
from datetime import datetime
|
||||
import streamlit as st
|
||||
from bs4 import BeautifulSoup
|
||||
|
||||
|
||||
def find_start_seconds(region):
|
||||
''' Returns start second for region. '''
|
||||
if len(region["Start"]) > 9:
|
||||
start_time = datetime.strptime(region["Start"], "%H:%M:%S.%f")
|
||||
elif len(region["Start"]) in [8, 9]:
|
||||
start_time = datetime.strptime(region["Start"], "%M:%S.%f")
|
||||
elif len(region["Start"]) < 8:
|
||||
start_time = datetime.strptime(region["Start"], "%S.%f")
|
||||
|
||||
t = start_time.time()
|
||||
return (t.hour * 60 + t.minute) * 60 + t.second
|
||||
|
||||
def find_lenght_seconds(region):
|
||||
''' Returns lenght of region in seconds. '''
|
||||
if len(region["Length"]) > 9:
|
||||
lenght = datetime.strptime(region["Length"], "%H:%M:%S.%f")
|
||||
elif len(region["Length"]) in [8, 9]:
|
||||
lenght = datetime.strptime(region["Length"], "%M:%S.%f")
|
||||
elif len(region["Length"]) < 8:
|
||||
lenght = datetime.strptime(region["Length"], "%S.%f")
|
||||
t = lenght.time()
|
||||
return (t.hour * 60 + t.minute) * 60 + t.second
|
||||
|
||||
def colorize(soup):
|
||||
# Get colors from clips on timeline.
|
||||
colors = {}
|
||||
n_regions = len(timline.find_all("Region"))
|
||||
for region in timline.find_all("Region"):
|
||||
try:
|
||||
colors[str(region["Ref"])] = int(region["Colour"])
|
||||
except:
|
||||
pass
|
||||
|
||||
# Give colors to clips in clipboard
|
||||
for group in soup.find_all("Group"):
|
||||
new_group_tag = group
|
||||
for region in group.find_all("Region"):
|
||||
try:
|
||||
with s:
|
||||
st.markdown(f':green[{region["Name"]}]')
|
||||
sleep(1/(n_regions/3))
|
||||
except KeyError:
|
||||
pass
|
||||
new_region_tag = region
|
||||
try:
|
||||
new_region_tag["Colour"] = colors[region["Ref"]]
|
||||
except KeyError:
|
||||
new_region_tag["Colour"] = "0"
|
||||
new_group_tag.append(new_region_tag)
|
||||
new_clipboard_tag.append(new_group_tag)
|
||||
|
||||
soup.Session.Clipboard.replace_with(new_clipboard_tag)
|
||||
|
||||
# Give colors to clips on timeline.
|
||||
new_tracks_tag = soup.find("Tracks")
|
||||
tracks = soup.find("Tracks")
|
||||
for track in tracks.find_all("Track"):
|
||||
try:
|
||||
with s:
|
||||
st.markdown(f':green[{region["Name"]}]')
|
||||
sleep(1/(n_regions/3))
|
||||
except KeyError:
|
||||
pass
|
||||
new_track_tag = track
|
||||
for region in track.find_all("Region"):
|
||||
new_region_tag = region
|
||||
try:
|
||||
new_region_tag["Colour"] = colors[region["Ref"]]
|
||||
except KeyError:
|
||||
new_region_tag["Colour"] = "0"
|
||||
new_track_tag.append(new_region_tag)
|
||||
new_tracks_tag.append(new_track_tag)
|
||||
soup.Session.Tracks.replace_with(new_tracks_tag)
|
||||
return soup
|
||||
|
||||
def make_rainbow(soup):
|
||||
''' Returns a tracks tag colored like a rainbow. '''
|
||||
|
||||
new_tracks_tag = soup.find("Tracks")
|
||||
tracks = soup.find("Tracks")
|
||||
regions = tracks.find_all("Region")
|
||||
min_start = min([find_start_seconds(i) for i in regions])
|
||||
max_end = max([int(find_start_seconds(i) + find_lenght_seconds(i)) for i in regions])
|
||||
|
||||
lenght = int(max_end-min_start)
|
||||
|
||||
# Give colors to clips on timeline.
|
||||
new_tracks_tag = soup.find("Tracks")
|
||||
tracks = soup.find("Tracks")
|
||||
for track in tracks.find_all("Track"):
|
||||
new_track_tag = track
|
||||
for region in track.find_all("Region"):
|
||||
new_region_tag = region
|
||||
try:
|
||||
middle = find_start_seconds(region) + find_lenght_seconds(region)/2
|
||||
new_region_tag["Colour"] = int(((middle)/lenght)*360)
|
||||
except KeyError:
|
||||
new_region_tag["Colour"] = "0"
|
||||
new_track_tag.append(new_region_tag)
|
||||
new_tracks_tag.append(new_track_tag)
|
||||
|
||||
return new_tracks_tag
|
||||
|
||||
print_sleep = False
|
||||
|
||||
st.title(":green[Colorize] your :red[Hindenburg] project")
|
||||
|
||||
st.markdown(
|
||||
@ -36,64 +142,59 @@ if uploaded_file:
|
||||
project_name = soup.find("AudioPool")["Path"].replace(" Files", "")
|
||||
new_clipboard_tag = soup.new_tag("Clipboard")
|
||||
|
||||
# Get colors from clips on timeline.
|
||||
colors = {}
|
||||
timline = soup.find("Tracks")
|
||||
n_regions = len(timline.find_all("Region"))
|
||||
for region in timline.find_all("Region"):
|
||||
try:
|
||||
colors[str(region["Ref"])] = int(region["Colour"])
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
# Area to print things.
|
||||
s = st.empty()
|
||||
c1, c2 = st.columns(2)
|
||||
with c1:
|
||||
s = st.empty()
|
||||
|
||||
# Give colors to clips in clipboard
|
||||
for group in soup.find_all("Group"):
|
||||
new_group_tag = group
|
||||
for region in group.find_all("Region"):
|
||||
try:
|
||||
s.markdown(f':green[{region["Name"]}]')
|
||||
sleep(1/n_regions)
|
||||
except KeyError:
|
||||
pass
|
||||
new_region_tag = region
|
||||
try:
|
||||
new_region_tag["Colour"] = colors[region["Ref"]]
|
||||
except KeyError:
|
||||
new_region_tag["Colour"] = "0"
|
||||
new_group_tag.append(new_region_tag)
|
||||
new_clipboard_tag.append(new_group_tag)
|
||||
|
||||
soup.Session.Clipboard.replace_with(new_clipboard_tag)
|
||||
|
||||
# Give colors to clips on timeline.
|
||||
new_tracks_tag = soup.find("Tracks")
|
||||
tracks = soup.find("Tracks")
|
||||
for track in tracks.find_all("Track"):
|
||||
try:
|
||||
s.markdown(f':green[{region["Name"]}]')
|
||||
sleep(1/n_regions)
|
||||
except KeyError:
|
||||
pass
|
||||
new_track_tag = track
|
||||
for region in track.find_all("Region"):
|
||||
new_region_tag = region
|
||||
try:
|
||||
new_region_tag["Colour"] = colors[region["Ref"]]
|
||||
except KeyError:
|
||||
new_region_tag["Colour"] = "0"
|
||||
new_track_tag.append(new_region_tag)
|
||||
new_tracks_tag.append(new_track_tag)
|
||||
soup.Session.Tracks.replace_with(new_tracks_tag)
|
||||
if 'soup' not in st.session_state:
|
||||
soup = colorize(soup)
|
||||
st.session_state['soup'] = soup
|
||||
print_sleep = True # Sleep while with print if first run.
|
||||
else:
|
||||
soup = st.session_state['soup']
|
||||
|
||||
# Allow the user to download file.
|
||||
with s:
|
||||
st.markdown(":green[Exporting...]")
|
||||
sleep(0.7)
|
||||
st.download_button(
|
||||
"Download file",
|
||||
soup.encode("utf-8"),
|
||||
file_name=f"{project_name}_colors.nhsx",
|
||||
)
|
||||
|
||||
with c1:
|
||||
with s:
|
||||
if print_sleep:
|
||||
st.markdown(":green[Exporting...]")
|
||||
sleep(0.7)
|
||||
st.download_button(
|
||||
"🎈 Download file",
|
||||
soup.encode("utf-8"),
|
||||
file_name=f"{project_name}_colors.nhsx",
|
||||
)
|
||||
|
||||
with c2:
|
||||
e = st.empty()
|
||||
with e:
|
||||
rainbow = st.button('🌈 Turn timeline into a rainbow! 🌈')
|
||||
if rainbow:
|
||||
for _ in range(0, 4):
|
||||
with e:
|
||||
st.markdown('🌧️')
|
||||
sleep(0.4)
|
||||
with e:
|
||||
st.markdown('')
|
||||
sleep(0.15)
|
||||
with e:
|
||||
st.markdown('🌦️')
|
||||
sleep(1)
|
||||
with e:
|
||||
rainbows = []
|
||||
for _ in range(0,10):
|
||||
rainbows.append('🌈')
|
||||
st.markdown(''.join(rainbows))
|
||||
sleep(0.09)
|
||||
new_tracks_tag = make_rainbow(soup)
|
||||
soup.Session.Tracks.replace_with(new_tracks_tag)
|
||||
with e:
|
||||
st.download_button(
|
||||
"🌈 Download rainbow project",
|
||||
soup.encode("utf-8"),
|
||||
file_name=f"{project_name}_rainbow.nhsx",
|
||||
)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user