Create app.py

app.py

@@ -0,0 +1,119 @@
+import gradio as gr
+import subprocess
+import os
+import time
+
+# Hàm đọc log từ file logger.txt
+def read_log_from_file():
+    log_file = "logger.txt"
+    if os.path.exists(log_file):
+        with open(log_file, "r", encoding="utf-8") as f:
+            return f.read()
+    return "Chưa có log nào trong file logger.txt.\n"
+
+# Hàm ghi log vào file logger.txt
+def write_log_to_file(message):
+    with open("logger.txt", "a", encoding="utf-8") as log_file:
+        log_file.write(f"{message}\n")
+
+# Hàm chạy file engine và cập nhật log
+def run_engine(input_param, log_state):
+    # Khởi tạo log nếu log_state là None, đọc từ file nếu có
+    if log_state is None:
+        log_state = read_log_from_file()
+
+    # Đường dẫn tới file engine
+    engine_path = "./engine.bin"
+    
+    # Kiểm tra và cấp quyền thực thi cho file engine
+    if not os.path.exists(engine_path):
+        log_state += "Lỗi: File engine.bin không tồn tại!\n"
+        write_log_to_file("Lỗi: File engine.bin không tồn tại!")
+        return log_state, log_state
+    
+    # Cấp quyền thực thi (chmod 755)
+    try:
+        os.chmod(engine_path, 0o755)
+        log_state += "Đã cấp quyền thực thi cho engine.bin.\n"
+        write_log_to_file("Đã cấp quyền thực thi cho engine.bin.")
+    except Exception as e:
+        log_state += f"Lỗi khi cấp quyền thực thi: {str(e)}\n"
+        write_log_to_file(f"Lỗi khi cấp quyền thực thi: {str(e)}")
+        return log_state, log_state
+
+    # Ghi log bắt đầu quá trình
+    start_message = f"Bắt đầu chạy engine với tham số: {input_param}\n"
+    log_state += start_message
+    write_log_to_file(start_message)
+    
+    try:
+        # Chạy file engine với tham số đầu vào
+        process = subprocess.Popen([engine_path, input_param], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+        # Đọc log theo thời gian thực từ stdout và stderr
+        while process.poll() is None:
+            output = process.stdout.readline()
+            if output:
+                log_state += output
+                write_log_to_file(output.strip())
+                yield log_state, log_state
+            time.sleep(0.1)  # Giảm tải CPU
+
+        # Đọc phần còn lại của output sau khi process hoàn tất
+        stdout, stderr = process.communicate()
+        if stdout:
+            log_state += stdout
+            write_log_to_file(stdout.strip())
+        if stderr:
+            log_state += "Lỗi: " + stderr
+            write_log_to_file("Lỗi: " + stderr.strip())
+        
+        log_state += "Hoàn tất quá trình chạy engine.\n"
+        write_log_to_file("Hoàn tất quá trình chạy engine.")
+    
+    except Exception as e:
+        error_message = f"Lỗi khi chạy engine: {str(e)}\n"
+        log_state += error_message
+        write_log_to_file(error_message)
+    
+    return log_state, log_state
+
+# Hàm xóa log
+def clear_log(log_state):
+    log_state = "Log đã được xóa.\n"
+    if os.path.exists("logger.txt"):
+        os.remove("logger.txt")
+    write_log_to_file("Log đã được xóa.")
+    return log_state, log_state
+
+# Tạo giao diện Gradio
+with gr.Blocks(title="Engine Runner on Hugging Face Space") as demo:
+    # State để lưu log
+    log_state = gr.State(value=None)
+    
+    # Giao diện người dùng
+    gr.Markdown("## Chạy Engine Binary và Hiển thị Log")
+    input_param = gr.Textbox(label="Tham số đầu vào", placeholder="Nhập tham số cho engine")
+    with gr.Row():
+        run_button = gr.Button("Chạy Engine")
+        clear_button = gr.Button("Xóa Log")
+    log_output = gr.Textbox(label="Log", lines=10, interactive=False, value=read_log_from_file())
+    
+    # Sự kiện khi nhấn nút "Chạy Engine"
+    run_button.click(
+        fn=run_engine,
+        inputs=[input_param, log_state],
+        outputs=[log_output, log_state],
+        _js=None
+    )
+    
+    # Sự kiện khi nhấn nút "Xóa Log"
+    clear_button.click(
+        fn=clear_log,
+        inputs=[log_state],
+        outputs=[log_output, log_state],
+        _js=None
+    )
+
+# Chạy ứng dụng
+demo.queue().launch()