Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 import subprocess
 import os
+import shutil
 
 logs = []
 inference_logs = []
+data_dir = "/home/user/data"
+
+os.makedirs(data_dir, exist_ok=True)  # تأكد من وجود المجلد
 
 def run_training():
     global logs
@@ -17,7 +21,7 @@ def run_training():
     )
     for line in process.stdout:
         logs.append(line)
-        yield "\n".join(logs[-50:])  # عرض آخر 50 سطر
+        yield "\n".join(logs[-50:])  # آخر 50 سطر فقط
 
 def run_inference():
     global inference_logs
@@ -36,7 +40,7 @@ def run_inference():
         output_lines.append(line)
 
     # حفظ النتائج
-    with open("/home/user/data/inference_results.txt", "a") as f:
+    with open(os.path.join(data_dir, "inference_results.txt"), "a") as f:
         f.write("=== New Inference Run ===\n")
         f.writelines(output_lines)
         f.write("\n")
@@ -44,15 +48,24 @@ def run_inference():
     yield "\n".join(inference_logs[-50:])
 
 def list_saved_files():
-    data_dir = "/home/user/data"
-    if not os.path.exists(data_dir):
-        return "📁 No saved files found."
-    
     files = os.listdir(data_dir)
     if not files:
-        return "📂 No files found in /data."
-    
-    return "\n".join(f"📄 {f}" for f in files)
+        return [], "📂 No files found in /data."
+    return files, "\n".join(f"📄 {f}" for f in files)
+
+def upload_files(file):
+    if file is None:
+        return "No file uploaded."
+    dest_path = os.path.join(data_dir, os.path.basename(file.name))
+    shutil.copy(file.name, dest_path)
+    return f"✅ Uploaded to {dest_path}"
+
+def download_file(filename):
+    file_path = os.path.join(data_dir, filename)
+    if os.path.exists(file_path):
+        return file_path
+    else:
+        return None
 
 with gr.Blocks(title="VRP Solver Interface") as demo:
     gr.Markdown("# 🚛 VRP Solver: Transformer + RL + OR-Tools")
@@ -63,13 +76,22 @@ with gr.Blocks(title="VRP Solver Interface") as demo:
         start_btn.click(fn=run_training, outputs=output)
 
     with gr.Tab("🔍 Inference"):
-        infer_btn = gr.Button("Run Inference")
+        with gr.Row():
+            infer_btn = gr.Button("Run Inference")
+            upload_input = gr.File(label="Upload File for Inference (.json or .pt)")
+            upload_status = gr.Textbox(label="Upload Status")
         infer_output = gr.Textbox(label="Inference Output", lines=15)
+
         infer_btn.click(fn=run_inference, outputs=infer_output)
+        upload_input.change(fn=upload_files, inputs=upload_input, outputs=upload_status)
 
     with gr.Tab("📁 Show Saved Files"):
         list_btn = gr.Button("List /data Files")
         file_output = gr.Textbox(label="Saved Files", lines=10)
-        list_btn.click(fn=list_saved_files, outputs=file_output)
+        file_dropdown = gr.Dropdown(choices=[], label="Select File to Download")
+        download_btn = gr.File(label="Download File")
+
+        list_btn.click(fn=list_saved_files, outputs=[file_dropdown, file_output])
+        file_dropdown.change(fn=download_file, inputs=file_dropdown, outputs=download_btn)
 
 demo.launch()