Update app.py

app.py

@@ -15,7 +15,7 @@ from pretty_midi import PrettyMIDI
 import numpy as np
 import pydub
 from PIL import Image
-
+import pytube
 
 st.set_page_config(
     page_title="TastyPiano",
@@ -38,6 +38,13 @@ def centered_button(func, text, args=None):
         return func(text)
 
 def setup_streamlite():
+    path = '/'.join(pytube.__file__.split('/')[:-1]) + '/cipher.py'
+    with open(path, 'r') as f:
+        cipher = f.read().split('\n')
+    cipher[286] = cipher[286].replace('\\', '')
+    with open(path, 'w') as f:
+        f.write('\n'.join(cipher))
+    os.path.exists(path)
     setup_translation_models()
     image = Image.open('./data/pianocktail.jpg')
     st.image(image, caption='Pianocktail by Compagnie la Rumeur')
@@ -72,14 +79,16 @@ def setup_streamlite():
         midi = st.file_uploader("Upload it here (.mid)", type=['.mid'])
         generate_audio_from_midi = st.checkbox('Generate audio? Untick if the song is too long (>10min)', value=True)
 
+    #url = "https://www.youtube.com/watch?v=UGK70IkP830"
+    #unit# = 'mL'
+
     def run(unit):
         setup_and_run(unit=unit, url=url, midi=midi, audio=audio, generate_audio_from_midi=generate_audio_from_midi, extra_code=None)
-    # url = "https://youtu.be/VUCI-1vIbUo"
-    # unit = 'mL'
-    # run(unit)
+    #run(unit)
     st.markdown('##')
     unit = st.radio('Pick the units (before pressing "Taste it!", default mL)', ['mL', 'oz'], index=0)
     button = centered_button(st.button, 'Taste it!')
+    # print(url)
     if button:
         run(unit)
 
@@ -129,7 +138,7 @@ def setup_and_run(unit='mL', url=None, midi=None, audio=None, generate_audio_fro
         assert False
     st.subheader('Synesthesia')
     now = datetime.now()
-    folder_name = f'{now.year}-{now.month}-{now.day}_{now.hour}:{now.minute}:{now.second}'
+    folder_name = f'date_{now.year}_{now.month}_{now.day}_time_{now.hour}_{now.minute}_{now.second}'
     folder_path = synestesia_path + folder_name
     if extra_code is not None:
         folder_path += '_' + extra_code