Update app.py

app.py

@@ -12,6 +12,12 @@ import time
 
 console = Console()
 
+def fade_text(text, duration=0.5):
+    for i in range(10):
+        opacity = i / 10
+        yield f"<div style='opacity: {opacity}; transition: opacity 0.05s;'>{text}</div>"
+        time.sleep(duration / 10)
+
 @spaces.GPU
 def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass: bool, other: bool, mp3: bool, mp3_bitrate: int) -> Tuple[str, List[str], gr.HTML]:
     log_messages = []
@@ -19,16 +25,17 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     def stream_log(message, style=""):
         formatted_message = f"[{model_name}] {message}"
         log_messages.append(formatted_message)
-        return gr.HTML(f"<pre style='margin-bottom: 0;{style}'>{formatted_message}</pre>")
+        for frame in fade_text(f"<pre style='margin-bottom: 0;{style}'>{formatted_message}</pre>"):
+            yield None, None, gr.HTML(frame)
 
-    yield None, None, stream_log("Initializing Demucs...", "color: #4CAF50; font-weight: bold;")
+    yield from stream_log("Initializing Demucs...", "color: #4CAF50; font-weight: bold;")
     time.sleep(1)  # Simulate initialization time
 
-    yield None, None, stream_log("Loading audio file...", "color: #2196F3;")
+    yield from stream_log("Loading audio file...", "color: #2196F3;")
     time.sleep(0.5)  # Simulate loading time
     
     if audio_file is None:
-        yield None, None, stream_log("Error: No audio file provided", "color: #F44336;")
+        yield from stream_log("Error: No audio file provided", "color: #F44336;")
         raise gr.Error("Please upload an audio file")
 
     # Use absolute paths
@@ -44,7 +51,7 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
         audio_file
     ]
 
-    yield None, None, stream_log("Preparing separation process...", "color: #FF9800;")
+    yield from stream_log("Preparing separation process...", "color: #FF9800;")
     time.sleep(0.5)  # Simulate preparation time
 
     try:
@@ -68,65 +75,65 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
 
         if process.returncode != 0:
             error_output = process.stderr.read()
-            yield None, None, stream_log(f"Error: Separation failed", "color: #F44336;")
+            yield from stream_log(f"Error: Separation failed", "color: #F44336;")
             raise gr.Error(f"Demucs separation failed. Check the logs for details.")
 
     except Exception as e:
-        yield None, None, stream_log(f"Unexpected error: {str(e)}", "color: #F44336;")
+        yield from stream_log(f"Unexpected error: {str(e)}", "color: #F44336;")
         raise gr.Error(f"An unexpected error occurred: {str(e)}")
 
-    yield None, None, stream_log("Separation completed successfully!", "color: #4CAF50; font-weight: bold;")
+    yield from stream_log("Separation completed successfully!", "color: #4CAF50; font-weight: bold;")
     time.sleep(0.5)  # Pause for effect
 
-    yield None, None, stream_log("Processing stems...", "color: #9C27B0;")
+    yield from stream_log("Processing stems...", "color: #9C27B0;")
     time.sleep(0.5)  # Simulate processing time
 
     # Change the stem search directory using full path
     stem_search_dir = os.path.join(base_output_dir, model_name, os.path.splitext(os.path.basename(audio_file))[0])
-    yield None, None, stream_log(f"Searching for stems in: {stem_search_dir}")
+    yield from stream_log(f"Searching for stems in: {stem_search_dir}")
 
     stems: Dict[str, str] = {}
     for stem in ["vocals", "drums", "bass", "other"]:
         stem_path = os.path.join(stem_search_dir, f"{stem}.wav")
-        yield None, None, stream_log(f"Checking for {stem} stem at: {stem_path}")
+        yield from stream_log(f"Checking for {stem} stem at: {stem_path}")
         if os.path.exists(stem_path):
             stems[stem] = stem_path
-            yield None, None, stream_log(f"Found {stem} stem")
+            yield from stream_log(f"Found {stem} stem")
         else:
-            yield None, None, stream_log(f"Warning: {stem} stem not found")
+            yield from stream_log(f"Warning: {stem} stem not found")
 
     if not stems:
-        yield None, None, stream_log("Error: No stems found. Checking alternative directory...")
+        yield from stream_log("Error: No stems found. Checking alternative directory...")
         stem_search_dir = os.path.join(base_output_dir, model_name)
         for stem in ["vocals", "drums", "bass", "other"]:
             stem_path = os.path.join(stem_search_dir, f"{stem}.wav")
-            yield None, None, stream_log(f"Checking for {stem} stem at: {stem_path}")
+            yield from stream_log(f"Checking for {stem} stem at: {stem_path}")
             if os.path.exists(stem_path):
                 stems[stem] = stem_path
-                yield None, None, stream_log(f"Found {stem} stem")
+                yield from stream_log(f"Found {stem} stem")
             else:
-                yield None, None, stream_log(f"Warning: {stem} stem not found")
+                yield from stream_log(f"Warning: {stem} stem not found")
 
-    yield None, None, stream_log(f"All found stems: {list(stems.keys())}")
+    yield from stream_log(f"All found stems: {list(stems.keys())}")
 
     selected_stems: List[str] = []
     for stem, selected in zip(["vocals", "drums", "bass", "other"], [vocals, drums, bass, other]):
         if selected:
-            yield None, None, stream_log(f"{stem} is selected by user")
+            yield from stream_log(f"{stem} is selected by user")
             if stem in stems:
                 selected_stems.append(stems[stem])
-                yield None, None, stream_log(f"Selected {stem} stem for mixing")
+                yield from stream_log(f"Selected {stem} stem for mixing")
             else:
-                yield None, None, stream_log(f"Warning: {stem} was selected but not found")
+                yield from stream_log(f"Warning: {stem} was selected but not found")
 
-    yield None, None, stream_log(f"Final selected stems: {selected_stems}")
+    yield from stream_log(f"Final selected stems: {selected_stems}")
 
     if not selected_stems:
-        yield None, None, stream_log("Error: No stems selected for mixing", "color: #F44336;")
+        yield from stream_log("Error: No stems selected for mixing", "color: #F44336;")
         raise gr.Error("Please select at least one stem to mix and ensure it was successfully separated.")
 
     output_file: str = os.path.join(output_dir, "mixed.wav")
-    yield None, None, stream_log("Mixing selected stems...", "color: #FF5722;")
+    yield from stream_log("Mixing selected stems...", "color: #FF5722;")
     time.sleep(0.5)  # Simulate mixing time
     
     mixed_audio: AudioSegment = AudioSegment.empty()
@@ -135,20 +142,19 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     mixed_audio.export(output_file, format="wav")
 
     if mp3:
-        yield None, None, stream_log(f"Converting to MP3...", "color: #795548;")
+        yield from stream_log(f"Converting to MP3...", "color: #795548;")
         time.sleep(0.5)  # Simulate conversion time
         mp3_output_file: str = os.path.splitext(output_file)[0] + ".mp3"
         mixed_audio.export(mp3_output_file, format="mp3", bitrate=str(mp3_bitrate) + "k")
         output_file = mp3_output_file
 
-    yield None, None, stream_log("Process completed successfully!", "color: #4CAF50; font-weight: bold;")
-    yield output_file, list(stems.values()), gr.HTML(
-        Panel.fit(
-            Text("Separation and mixing completed successfully!", style="bold green"),
-            title="Demucs Result",
-            border_style="green"
-        ).render()
+    yield from stream_log("Process completed successfully!", "color: #4CAF50; font-weight: bold;")
+    final_message = Panel(
+        Text("Separation and mixing completed successfully!", style="bold green"),
+        title="Demucs Result",
+        border_style="green"
     )
+    yield output_file, list(stems.values()), gr.HTML(console.export_html(final_message))
 
 # Define the Gradio interface
 with gr.Blocks() as iface: