Java调用Python脚本：单次导入TensorFlow/spaCy及交互故障排查

问题说明 #

需要通过Java调用Python脚本，由于TensorFlow、spaCy等库导入耗时久，希望仅执行一次导入操作，保持Python进程存活并多次交互处理输入；但当前代码无法得到预期的三次输入输出结果。

解决方案 #

1. 调整Python脚本：实现单次导入+持续交互 #

将库导入语句放在循环外部，确保仅加载一次；同时处理EOF信号，避免进程异常退出：

import sys
import tensorflow as tf
import spacy

# 提前加载所需模型/资源，仅执行一次
nlp = spacy.load("en_core_web_sm")

while True:
    try:
        # 读取Java传入的输入
        input_str = input().strip()
        if not input_str:
            continue
        
        # 这里替换为实际业务处理逻辑
        output_str = "Processed input: " + input_str
        
        # 输出结果并强制刷新缓冲区，确保Java能立即读取
        print(output_str)
        sys.stdout.flush()
    except EOFError:
        # 捕获Java关闭输入流的信号，正常退出循环
        break

2. 修复Java代码：解决阻塞问题，实现逐行交互 #

原代码中reader.lines()会持续阻塞直到Python进程结束，导致第一次循环后无法继续执行。改用readLine()逐行读取输出，适配交互式场景：

import java.io.*;
import java.util.Map;
import java.util.Scanner;

public class RunPythonScriptInteractive {
    public static void main(String[] args) throws Exception {
        try {
            ProcessBuilder builder = new ProcessBuilder();
            Map<String, String> env = builder.environment();
            // 添加conda环境路径，确保Python能找到对应库
            env.put("PATH", env.get("PATH") + ":/home/parth/miniconda3/bin");
            builder.command("/bin/bash", "-c", "python pythontry.py");

            // 将错误流重定向到输出流，方便统一读取
            builder.redirectErrorStream(true);
            Process process = builder.start();
            Scanner userInput = new Scanner(System.in);
            
            BufferedWriter pythonInput = new BufferedWriter(new OutputStreamWriter(process.getOutputStream()));
            BufferedReader pythonOutput = new BufferedReader(new InputStreamReader(process.getInputStream()));

            for (int i = 0; i < 3; i++) {
                System.out.print("输入内容: ");
                String input = userInput.nextLine();
                
                // 向Python发送输入，换行+刷新缓冲区确保对方能接收
                pythonInput.write(input);
                pythonInput.newLine();
                pythonInput.flush();

                // 读取Python返回的一行结果
                String output = pythonOutput.readLine();
                if (output != null) {
                    System.out.println(output);
                }
            }
            
            // 关闭输入流，触发Python的EOFError，使其正常退出
            pythonInput.close();
            // 读取Python进程退出前的剩余输出
            String remainingLine;
            while ((remainingLine = pythonOutput.readLine()) != null) {
                System.out.println(remainingLine);
            }
            pythonOutput.close();

            // 等待Python进程完全结束，获取退出状态
            int exitCode = process.waitFor();
            System.out.println("Python进程已退出，状态码: " + exitCode);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

关键修复点说明 #

• Python端：库导入和模型加载放在循环外，避免重复耗时操作；添加EOF异常处理，保证进程优雅退出。
• Java端：用readLine()替代lines()，避免因等待流关闭导致的阻塞；每次发送输入后强制刷新缓冲区，确保Python能立即读取到内容；最后关闭输入流并读取剩余输出，保证交互完整性。

内容的提问来源于stack exchange，提问作者Parth Gupta