fix: update the formatted name

app.py

@@ -28,16 +28,33 @@ def delete_input_files(input_dir):
         wav_file.unlink()
         print(f"Deleted {wav_file}")
 
-def analyze_filename_with_llm(filename):
-    prompt = f"Extract the artist and song title from the following filename and format it as 'Artist - Title':\n\n{filename}"
-    
-    # Generate a response using the local model
-    response = model(prompt, max_length=50, do_sample=False)[0]['generated_text']
-    
-    # Extract the first line of the response, which should be the "Artist - Title"
-    artist_title = response.strip().split('\n')[0]
-    
-    return artist_title
+def standardize_title(input_title):
+    title_cleaned = re.sub(r"[\(\[].*?[\)\]]", "", input_title)
+
+    unnecessary_words = ["official", "video", "hd", "4k", "lyrics", "music", "audio", "visualizer", "remix"]
+    title_cleaned = re.sub(r"\b(?:{})\b".format("|".join(unnecessary_words)), "", title_cleaned, flags=re.IGNORECASE)
+
+    parts = re.split(r"\s*-\s*|\s*,\s*", title_cleaned)
+
+    if len(parts) >= 2:
+        title_part = parts[-1].strip()
+        artist_part = ', '.join(parts[:-1]).strip()
+    else:
+        artist_part = "Unknown Artist"
+        title_part = title_cleaned.strip()
+
+    if "with" in input_title.lower() or "feat" in input_title.lower():
+        match = re.search(r"\((with|feat\.?) (.*?)\)", input_title, re.IGNORECASE)
+        if match:
+            additional_artist = match.group(2).strip()
+            artist_part = f"{artist_part}, {additional_artist}" if artist_part != "Unknown Artist" else additional_artist
+
+    artist_part = re.sub(r'\s+', ' ', artist_part).title()
+    title_part = re.sub(r'\s+', ' ', title_part).title()
+
+    standardized_output = f"{artist_part} - {title_part}"
+
+    return standardized_output.strip()
 
 def handle_file_upload(file):
     if file is None:
@@ -45,15 +62,13 @@ def handle_file_upload(file):
 
     filename = os.path.basename(file.name)
     
-    # Use LLM to analyze the filename and return the formatted title
-    formatted_title = analyze_filename_with_llm(filename)
+    formatted_title = standardize_title(filename)
 
     formatted_title = sanitize_filename(formatted_title.strip())
 
     input_path = os.path.join(INPUT_FOLDER, "wav", f"{formatted_title}.wav")
     os.makedirs(os.path.dirname(input_path), exist_ok=True)
 
-    # Copy the uploaded file to the input folder
     shutil.copy(file.name, input_path)
 
     return input_path, formatted_title