Update app.py

app.py

@@ -86,9 +86,10 @@ def infer(genre_txt_content, lyrics_txt_content):
     print(f"Output folder ensured at: {output_dir}")
 
     
-    # Command and arguments with additional parameters
+    # Command and arguments
     command = [
-        "python", "infer.py",
+        "python", "-u",  # Add -u for unbuffered output
+        "infer.py",
         "--stage1_model", "m-a-p/YuE-s1-7B-anneal-en-cot",
         "--stage2_model", "m-a-p/YuE-s2-1B-general",
         "--genre_txt", f"{genre_txt_path}",
@@ -98,38 +99,53 @@ def infer(genre_txt_content, lyrics_txt_content):
         "--output_dir", f"{output_dir}",
         "--cuda_idx", "0",
         "--max_new_tokens", "3000",
-        "--disable_offload_model"  # Add this to prevent GPU/CPU transfer issues
+        "--disable_offload_model"
     ]
     
+    # Execute the command
     try:
-        # Run with environment variables to control CUDA
-        env = os.environ.copy()
-        env["CUDA_VISIBLE_DEVICES"] = "0"
-        result = subprocess.run(
-            command, 
-            check=True,
-            env=env,
+        # Run process with real-time output
+        process = subprocess.Popen(
+            command,
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
-            text=True
+            text=True,
+            bufsize=1,
+            universal_newlines=True
         )
-        print("Command output:", result.stdout)
-        print("Command errors:", result.stderr)
+
+        # Print output in real-time
+        while True:
+            output = process.stdout.readline()
+            error = process.stderr.readline()
+            
+            if output:
+                print(f"Output: {output.strip()}")
+            if error:
+                print(f"Error: {error.strip()}")
+                
+            # Check if process has finished
+            if process.poll() is not None:
+                break
+        
+        # Get the return code
+        return_code = process.poll()
         
-        # Check output files
-        output_files = os.listdir(output_dir)
-        if output_files:
-            print("Output folder contents:")
-            for file in output_files:
-                print(f"- {file}")
-            # TODO: Return the generated audio file path
-            return None
+        if return_code == 0:
+            print("Command executed successfully!")
+            output_files = os.listdir(output_dir)
+            if output_files:
+                print("Output folder contents:")
+                for file in output_files:
+                    print(f"- {file}")
+            else:
+                print("Output folder is empty.")
         else:
-            print("Output folder is empty.")
-            return None
-    except subprocess.CalledProcessError as e:
+            print(f"Command failed with return code {return_code}")
+            
+        return None
+    except Exception as e:
         print(f"Error occurred: {e}")
-        print("Error output:", e.stderr)
         return None
     finally:
         # Clean up temporary files