Update app.py

app.py

@@ -8,7 +8,7 @@ import tempfile
 import shutil
 import requests
 from pathlib import Path
-temp_dir = tempfile.TemporaryDirectory()
+temp_dir = os.path.expanduser("~/app")
 global ckpt_temp_file
 global audio_temp_file
 global config_temp_file
@@ -62,32 +62,31 @@ st.set_page_config(
 st.title('DIFF-SVC Render')
 
 ###CKPT LOADER
-
-ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
+with tempfile.TemporaryDirectory(dir=os.path.expanduser("~/app")) as temp_dir:
+    ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
     # Check if user uploaded a CKPT file
-if ckpt is not None:
+    if ckpt is not None:
       #TEMP FUNCTION
-    with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False) as temp:
+      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)
-        config_temp_file = temp.name
+        ckpt_temp_file = temp.name
         # Print the temporary file name
         print(temp.name)
-    st.success("File saved to: {}".format(ckpt_temp_file))
 
     # Display the file path
-if "ckpt_temp_file" in locals():
-    st.success("File saved to: {}".format(ckpt_temp_file))
+    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')
+    config = st.file_uploader("Choose your config", type= 'yaml')
 
     # Check if user uploaded a config file
-if config is not None:
+    if config is not None:
       #TEMP FUNCTION
-    with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False) as temp:
+      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
@@ -97,15 +96,15 @@ if config is not None:
         print(temp.name)
 
     # Display the file path
-if "config_temp_file" in locals():
-    st.success("File saved to: {}".format(config_temp_file))
+    if "config_temp_file" in locals():
+        st.success("File saved to: {}".format(config_temp_file))
 
-audio = st.file_uploader("Choose your audio", type=["wav", "mp3"])
+    audio = st.file_uploader("Choose your audio", type=["wav", "mp3"])
 
     # Check if user uploaded an audio file
-if audio is not None:
+    if audio is not None:
   #TEMP FUNCTION
-    with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False) as temp:
+      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
@@ -115,8 +114,8 @@ if audio is not None:
         print(temp.name)
 
 # Display the file path
-if "audio_temp_file" in locals():
-    st.success("File saved to: {}".format(audio_temp_file))
+    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")
 # Add a button to start the rendering process