Update app.py

app.py

@@ -67,34 +67,17 @@ def empty_output_folder(output_dir):
             print(f"Error deleting file {file_path}: {e}")
 
 # Function to create a temporary file with string content
-def create_temp_file(content, suffix=".txt"):
-    content = content.strip()
-
-    # Case 1: If there's no block marker `[]`, treat it as a single-line input
-    if not re.search(r"\[\w+\]", content):
-        # Normalize single-line input to avoid extra newlines
-        formatted_content = " ".join(content.split())
-    else:
-        # Case 2: If there are block markers, ensure blocks are separated by two newlines
-        blocks = re.split(r"(?=\[\w+\])", content)  # Keeps headers in the split parts
-        cleaned_blocks = [block.strip() for block in blocks if block.strip()]
-        formatted_content = "\n\n".join(cleaned_blocks)  # Ensure exactly two newlines
-
-    # Create temp file
-    fd, path = tempfile.mkstemp(suffix=suffix)
+def create_temp_file(content, prefix, suffix=".txt"):
+    temp_file = tempfile.NamedTemporaryFile(delete=False, mode="w", prefix=prefix, suffix=suffix)
+    temp_file.write(content)
+    temp_file.close()
     
-    try:
-        # Open file in write mode and write the formatted content
-        with os.fdopen(fd, "w", encoding="utf-8") as f:
-            f.write(formatted_content)
-    except Exception as e:
-        print(f"Error writing file: {e}")
-        return None
+    # Debug: Print file contents
+    print(f"\nContent written to {prefix}{suffix}:")
+    print(content)
+    print("---")
     
-    # Ensure file permissions allow reading
-    os.chmod(path, 0o644)  # Read & write permissions for owner, read for others
-
-    return path
+    return temp_file.name
 
 def get_last_mp3_file(output_dir):
     # List all files in the output directory
@@ -118,8 +101,8 @@ def get_last_mp3_file(output_dir):
 
 def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
     # Create temporary files
-    genre_txt_path = create_temp_file(genre_txt_content, ".txt")
-    lyrics_txt_path = create_temp_file(lyrics_txt_content, ".txt")
+    genre_txt_path = create_temp_file(genre_txt_content, prefix="genre_")
+    lyrics_txt_path = create_temp_file(lyrics_txt_content, prefix="lyrics_")
 
     print(f"Genre TXT path: {genre_txt_path}")
     print(f"Lyrics TXT path: {lyrics_txt_path}")
@@ -249,8 +232,7 @@ with gr.Blocks() as demo:
 In the quiet of the evening, shadows start to fall
 Whispers of the night wind echo through the hall
 Lost within the silence, I hear your gentle voice
-Guiding me back homeward, making my heart rejoice
-
+Guiding me back homeward, making my heart rejoice\n\n
 [chorus]
 Don't let this moment fade, hold me close tonight
 With you here beside me, everything's alright
@@ -267,8 +249,7 @@ Got my team beside me, no room for fear
 Walking through the streets, beats inside my head
 Every step I take, closer to the bread
 People passing by, they don't understand
-Building up my future with my own two hands
-
+Building up my future with my own two hands\n\n
 [chorus]
 This is my life, and I'm aiming for the top
 Never gonna quit, no, I'm never gonna stop