Update app.py

app.py

@@ -25,7 +25,7 @@ def run_training():
 def run_inference():
     global inference_logs
     inference_logs = []
-    output_lines = []
+    summary_line = ""
 
     process = subprocess.Popen(
         ["python", "inference.py"],
@@ -36,14 +36,18 @@ def run_inference():
     )
     for line in process.stdout:
         inference_logs.append(line)
-        output_lines.append(line)
+        # التحقق من وجود ملخص
+        if "Summary for UI" in line:
+            summary_line = line.strip()
 
-    with open(os.path.join(data_dir, "inference_results.txt"), "a") as f:
+    # حفظ النتائج
+    with open("/home/user/data/inference_results.txt", "a") as f:
         f.write("=== New Inference Run ===\n")
-        f.writelines(output_lines)
+        f.writelines(inference_logs)
         f.write("\n")
 
-    yield "\n".join(inference_logs[-50:])
+    return summary_line or "\n".join(inference_logs[-50:])
+
 
 def list_saved_files():
     files = os.listdir(data_dir)