Update app.py

app.py

@@ -60,62 +60,64 @@ st.set_page_config(
 st.title('DIFF-SVC Render')
 
 ###CKPT LOADER
-# File uploader
-ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
+# Create the temporary directory in the desired location
+with tempfile.TemporaryDirectory(dir=os.path.expanduser("~/app")) as temp_dir:
+    # CKPT LOADER
+    # File uploader
+    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)
+    # Check if user uploaded a CKPT file
+    if ckpt is not None:
+      #TEMP FUNCTION
+      with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False, dir=temp_dir) 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))
+    # 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')
+    # 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)
+    # Check if user uploaded a config file
+    if config is not None:
+      #TEMP FUNCTION
+      with tempfile.NamedTemporaryFile(mode="w", suffix='.yaml', delete=False, dir=temp_dir) as temp:
+        # Get the file contents as a string
+        str_data = config.read()
+        # Write the string to the temporary file
+        temp.write(str_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))
+    # Display the file path
+    if "config_temp_file" in locals():
+        st.success("File saved to: {}".format(config_temp_file))
 
-# File uploader
-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:
+    # Check if user uploaded an audio file
+    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, dir=temp_dir) as temp:
     # Get the file contents as bytes
-    bytes_data = audio.getvalue()
+       bytes_data = audio.getvalue()
     # Write the bytes to the temporary file
-    temp.write(bytes_data)
-    audio_temp_file = temp.name
+       temp.write(bytes_data)
+       audio_temp_file = temp.name
     # Print the temporary file name
-    print(temp.name)
+       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