rizwarrior commited on
Commit
65066a0
·
1 Parent(s): 22826d0

Fix NumPy compatibility issue and improve filename handling

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -1
  2. server.py +12 -3
requirements.txt CHANGED
@@ -5,4 +5,5 @@ Werkzeug==2.3.6
5
  gunicorn==21.2.0
6
  soundfile==0.12.1
7
  torch==2.0.1
8
- torchaudio==2.0.2
 
 
5
  gunicorn==21.2.0
6
  soundfile==0.12.1
7
  torch==2.0.1
8
+ torchaudio==2.0.2
9
+ numpy<2.0
server.py CHANGED
@@ -66,8 +66,17 @@ def separate_audio():
66
  temp_dir = tempfile.mkdtemp(prefix=f"demucs_{session_id}_")
67
 
68
  try:
69
- # Save uploaded file to temp directory
70
- filename = secure_filename(file.filename)
 
 
 
 
 
 
 
 
 
71
  file_path = os.path.join(temp_dir, filename)
72
  file.save(file_path)
73
 
@@ -138,7 +147,7 @@ def separate_audio():
138
  # Store session data in memory
139
  active_sessions[session_id] = {
140
  'tracks': track_data,
141
- 'original_filename': filename,
142
  'created_at': time.time(),
143
  'temp_dir': temp_dir
144
  }
 
66
  temp_dir = tempfile.mkdtemp(prefix=f"demucs_{session_id}_")
67
 
68
  try:
69
+ # Save uploaded file to temp directory with clean filename
70
+ original_filename = secure_filename(file.filename)
71
+ # Remove any problematic characters and ensure proper extension
72
+ base_name = os.path.splitext(original_filename)[0]
73
+ extension = os.path.splitext(original_filename)[1].lower()
74
+
75
+ # Clean the base name to avoid issues
76
+ import re
77
+ clean_base = re.sub(r'[^\w\-_\.]', '_', base_name)
78
+ filename = f"{clean_base}{extension}"
79
+
80
  file_path = os.path.join(temp_dir, filename)
81
  file.save(file_path)
82
 
 
147
  # Store session data in memory
148
  active_sessions[session_id] = {
149
  'tracks': track_data,
150
+ 'original_filename': original_filename, # Use original filename for display
151
  'created_at': time.time(),
152
  'temp_dir': temp_dir
153
  }