Update app.py

app.py

@@ -1,35 +1,70 @@
 import streamlit as st
 import os
-password = os.environ.get("password")
-# Define a function to hide a Streamlit tab
-def hide_tab():
-    hide_streamlit_style = """
-            <style>
-            #MainMenu {visibility: hidden;}
-            footer {visibility: hidden;}
-            </style>
-            """
-    st.markdown(hide_streamlit_style, unsafe_allow_html=True)
+st.set_page_config(
+    page_title="DiffSVC Render",
+    page_icon="🧊",
+    initial_sidebar_state="expanded",
+)
+############
+st.title('DIFF-SVC Render')
 
-# Define a function to show a Streamlit tab
-def show_tab():
-    show_streamlit_style = """
-            <style>
-            #MainMenu {visibility: visible;}
-            footer {visibility: visible;}
-            </style>
-            """
-    st.markdown(show_streamlit_style, unsafe_allow_html=True)
-
-# Hide the Streamlit tab by default
-hide_tab()
-
-# Create a button to show the hidden tab
-if st.button('Show tab'):
-    show_tab()
-    # Add content to the tab
-    st.write('This is a hidden tab. You can now see it!')
-    st.write(f"Password: {password}")
-    # Create a button to hide the tab again
-    if st.button('Hide tab'):
-        hide_tab()
+###CKPT LOADER
+ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
+# Check if user uploaded a CKPT file
+if ckpt is not None:
+  #TEMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False) as temp:
+    # Get the file contents as bytes
+    bytes_data = ckpt.getvalue()
+    # Write the bytes to the temporary file
+    temp.write(bytes_data)
+    ckpt_temp_file = temp.name
+    # Print the temporary file name
+    print(temp.name)
+# Display the file path
+if "ckpt_temp_file" in locals():
+    st.success("File saved to: {}".format(ckpt_temp_file))
+# File uploader
+config = st.file_uploader("Choose your config", type= 'yaml')
+# Check if user uploaded a config file
+if config is not None:
+  #TEMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False) as temp:
+    # Get the file contents as bytes
+    bytes_data = config.getvalue()
+    # Write the bytes to the temporary file
+    temp.write(bytes_data)
+    config_temp_file = temp.name
+    # Print the temporary file name
+    print(temp.name)
+# Display the file path
+if "config_temp_file" in locals():
+    st.success("File saved to: {}".format(config_temp_file))
+audio = st.file_uploader("Choose your audio", type=["wav"])
+# Check if user uploaded an audio file
+if audio is not None:
+#EMP FUNCTION
+  with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False) as temp:
+# Get the file contents as bytes
+    bytes_data = audio.getvalue()
+# Write the bytes to the temporary file
+    temp.write(bytes_data)
+    audio_temp_file = temp.name
+# Print the temporary file name
+    print(temp.name)
+# Display the file path
+if "audio_temp_file" in locals():
+    st.success("File saved to: {}".format(audio_temp_file))
+# Add a text input for the title with a default value of 0
+title = st.text_input("Key", value="0")
+title2 = st.text_input("Speedup", value="20")
+if password = st.text_input("Enter password")
+    correct_password = os.environ.get("gatepassword")
+# Add a button to start the rendering process
+# Add a button to start the rendering process
+if st.button("Render audio"):
+    if password == correct_password:
+        render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2)
+  
+    else:
+        st.error("Incorrect password")