Update app.py
Browse files
app.py
CHANGED
|
@@ -18,20 +18,16 @@ def write_log_to_file(message):
|
|
| 18 |
|
| 19 |
# Hàm chạy file engine và cập nhật log
|
| 20 |
def run_engine(input_param, log_state):
|
| 21 |
-
# Khởi tạo log nếu log_state là None, đọc từ file nếu có
|
| 22 |
if log_state is None:
|
| 23 |
log_state = read_log_from_file()
|
| 24 |
|
| 25 |
-
# Đường dẫn tới file engine
|
| 26 |
engine_path = "./engine.bin"
|
| 27 |
|
| 28 |
-
# Kiểm tra và cấp quyền thực thi cho file engine
|
| 29 |
if not os.path.exists(engine_path):
|
| 30 |
log_state += "Lỗi: File engine.bin không tồn tại!\n"
|
| 31 |
write_log_to_file("Lỗi: File engine.bin không tồn tại!")
|
| 32 |
return log_state, log_state
|
| 33 |
|
| 34 |
-
# Cấp quyền thực thi (chmod 755)
|
| 35 |
try:
|
| 36 |
os.chmod(engine_path, 0o755)
|
| 37 |
log_state += "Đã cấp quyền thực thi cho engine.bin.\n"
|
|
@@ -41,25 +37,21 @@ def run_engine(input_param, log_state):
|
|
| 41 |
write_log_to_file(f"Lỗi khi cấp quyền thực thi: {str(e)}")
|
| 42 |
return log_state, log_state
|
| 43 |
|
| 44 |
-
# Ghi log bắt đầu quá trình
|
| 45 |
start_message = f"Bắt đầu chạy engine với tham số: {input_param}\n"
|
| 46 |
log_state += start_message
|
| 47 |
write_log_to_file(start_message)
|
| 48 |
|
| 49 |
try:
|
| 50 |
-
# Chạy file engine với tham số đầu vào
|
| 51 |
process = subprocess.Popen([engine_path, input_param], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
| 52 |
|
| 53 |
-
# Đọc log theo thời gian thực từ stdout và stderr
|
| 54 |
while process.poll() is None:
|
| 55 |
output = process.stdout.readline()
|
| 56 |
if output:
|
| 57 |
log_state += output
|
| 58 |
write_log_to_file(output.strip())
|
| 59 |
yield log_state, log_state
|
| 60 |
-
time.sleep(0.1)
|
| 61 |
|
| 62 |
-
# Đọc phần còn lại của output sau khi process hoàn tất
|
| 63 |
stdout, stderr = process.communicate()
|
| 64 |
if stdout:
|
| 65 |
log_state += stdout
|
|
@@ -88,10 +80,8 @@ def clear_log(log_state):
|
|
| 88 |
|
| 89 |
# Tạo giao diện Gradio
|
| 90 |
with gr.Blocks(title="Engine Runner on Hugging Face Space") as demo:
|
| 91 |
-
# State để lưu log
|
| 92 |
log_state = gr.State(value=None)
|
| 93 |
|
| 94 |
-
# Giao diện người dùng
|
| 95 |
gr.Markdown("## Chạy Engine Binary và Hiển thị Log")
|
| 96 |
input_param = gr.Textbox(label="Tham số đầu vào", placeholder="Nhập tham số cho engine")
|
| 97 |
with gr.Row():
|
|
@@ -99,20 +89,18 @@ with gr.Blocks(title="Engine Runner on Hugging Face Space") as demo:
|
|
| 99 |
clear_button = gr.Button("Xóa Log")
|
| 100 |
log_output = gr.Textbox(label="Log", lines=10, interactive=False, value=read_log_from_file())
|
| 101 |
|
| 102 |
-
# Sự kiện khi nhấn nút "Chạy Engine"
|
| 103 |
run_button.click(
|
| 104 |
fn=run_engine,
|
| 105 |
inputs=[input_param, log_state],
|
| 106 |
-
outputs=[log_output, log_state]
|
| 107 |
-
_js=None
|
| 108 |
)
|
| 109 |
|
| 110 |
-
# Sự kiện khi nhấn nút "Xóa Log"
|
| 111 |
clear_button.click(
|
| 112 |
fn=clear_log,
|
| 113 |
inputs=[log_state],
|
| 114 |
-
outputs=[log_output, log_state]
|
| 115 |
-
_js=None
|
| 116 |
)
|
| 117 |
|
| 118 |
# Chạy ứng dụng
|
|
|
|
| 18 |
|
| 19 |
# Hàm chạy file engine và cập nhật log
|
| 20 |
def run_engine(input_param, log_state):
|
|
|
|
| 21 |
if log_state is None:
|
| 22 |
log_state = read_log_from_file()
|
| 23 |
|
|
|
|
| 24 |
engine_path = "./engine.bin"
|
| 25 |
|
|
|
|
| 26 |
if not os.path.exists(engine_path):
|
| 27 |
log_state += "Lỗi: File engine.bin không tồn tại!\n"
|
| 28 |
write_log_to_file("Lỗi: File engine.bin không tồn tại!")
|
| 29 |
return log_state, log_state
|
| 30 |
|
|
|
|
| 31 |
try:
|
| 32 |
os.chmod(engine_path, 0o755)
|
| 33 |
log_state += "Đã cấp quyền thực thi cho engine.bin.\n"
|
|
|
|
| 37 |
write_log_to_file(f"Lỗi khi cấp quyền thực thi: {str(e)}")
|
| 38 |
return log_state, log_state
|
| 39 |
|
|
|
|
| 40 |
start_message = f"Bắt đầu chạy engine với tham số: {input_param}\n"
|
| 41 |
log_state += start_message
|
| 42 |
write_log_to_file(start_message)
|
| 43 |
|
| 44 |
try:
|
|
|
|
| 45 |
process = subprocess.Popen([engine_path, input_param], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
| 46 |
|
|
|
|
| 47 |
while process.poll() is None:
|
| 48 |
output = process.stdout.readline()
|
| 49 |
if output:
|
| 50 |
log_state += output
|
| 51 |
write_log_to_file(output.strip())
|
| 52 |
yield log_state, log_state
|
| 53 |
+
time.sleep(0.1)
|
| 54 |
|
|
|
|
| 55 |
stdout, stderr = process.communicate()
|
| 56 |
if stdout:
|
| 57 |
log_state += stdout
|
|
|
|
| 80 |
|
| 81 |
# Tạo giao diện Gradio
|
| 82 |
with gr.Blocks(title="Engine Runner on Hugging Face Space") as demo:
|
|
|
|
| 83 |
log_state = gr.State(value=None)
|
| 84 |
|
|
|
|
| 85 |
gr.Markdown("## Chạy Engine Binary và Hiển thị Log")
|
| 86 |
input_param = gr.Textbox(label="Tham số đầu vào", placeholder="Nhập tham số cho engine")
|
| 87 |
with gr.Row():
|
|
|
|
| 89 |
clear_button = gr.Button("Xóa Log")
|
| 90 |
log_output = gr.Textbox(label="Log", lines=10, interactive=False, value=read_log_from_file())
|
| 91 |
|
| 92 |
+
# Sự kiện khi nhấn nút "Chạy Engine" (xóa _js)
|
| 93 |
run_button.click(
|
| 94 |
fn=run_engine,
|
| 95 |
inputs=[input_param, log_state],
|
| 96 |
+
outputs=[log_output, log_state]
|
|
|
|
| 97 |
)
|
| 98 |
|
| 99 |
+
# Sự kiện khi nhấn nút "Xóa Log" (xóa _js)
|
| 100 |
clear_button.click(
|
| 101 |
fn=clear_log,
|
| 102 |
inputs=[log_state],
|
| 103 |
+
outputs=[log_output, log_state]
|
|
|
|
| 104 |
)
|
| 105 |
|
| 106 |
# Chạy ứng dụng
|