Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import subprocess
 
-logs = []  # لتخزين اللوجات
+logs = []
+inference_logs = []
 
 def run_training():
     global logs
@@ -13,16 +14,47 @@ def run_training():
         text=True,
         bufsize=1
     )
-
     for line in process.stdout:
         logs.append(line)
-        yield "\n".join(logs[:])
+        yield "\n".join(logs[-50:])  # عرض آخر 50 سطر
+
+def run_inference():
+    global inference_logs
+    inference_logs = []
+    output_lines = []
+
+    process = subprocess.Popen(
+        ["python", "inference.py"],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+        text=True,
+        bufsize=1
+    )
+    for line in process.stdout:
+        inference_logs.append(line)
+        output_lines.append(line)
+
+    # حفظ النتائج في ملف
+    with open("inference_results.txt", "a") as f:
+        f.write("=== New Inference Run ===\n")
+        f.writelines(output_lines)
+        f.write("\n")
+
+    yield "\n".join(inference_logs[-50:])  # عرض آخر 50 سطر
+
+with gr.Blocks(title="VRP Transformer Training & Inference") as demo:
+    gr.Markdown("# 🚛 Vehicle Routing Problem Solver with Transformer + Reinforcement Learning")
 
-with gr.Blocks() as demo:
-    with gr.Row():
-        btn = gr.Button("🚀 Start Training")
-    output = gr.Textbox(label="Logs", lines=25, interactive=False)
+    with gr.Tab("🚀 Start Training"):
+        with gr.Row():
+            start_btn = gr.Button("Start Training")
+        output = gr.Textbox(label="Training Logs", lines=25)
+        start_btn.click(fn=run_training, outputs=output)
 
-    btn.click(fn=run_training, outputs=output)
+    with gr.Tab("🔍 Run Inference"):
+        with gr.Row():
+            infer_btn = gr.Button("Run Inference on New Batch")
+        infer_output = gr.Textbox(label="Inference Results", lines=15)
+        infer_btn.click(fn=run_inference, outputs=infer_output)
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch()