Update app.py to stream training logs live

app.py

@@ -3,25 +3,27 @@ import subprocess
 
 def run_training():
     try:
-        print("✅ Training started.")
-        result = subprocess.run(["python", "train_abuse_model.py"], capture_output=True, text=True)
-        print("✅ Training finished.")
-        print("STDOUT:", result.stdout)
-        print("STDERR:", result.stderr)
-        return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
-    except Exception as e:
-        print("❌ Exception occurred:", e)
-        return f"Exception occurred:\n{str(e)}"
+        process = subprocess.Popen(
+            ["python", "train_abuse_model.py"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            text=True
+        )
 
+        output_lines = []
+        for line in process.stdout:
+            output_lines.append(line)
+            yield "".join(output_lines)
 
+    except Exception as e:
+        yield f"Exception occurred:\n{str(e)}"
 
-# Define a simple Gradio interface with one button
 demo = gr.Interface(
     fn=run_training,
     inputs=[],
-    outputs="text",
+    outputs=gr.Textbox(lines=25, label="Training Logs"),
     title="Run Model Training",
-    description="Click the button to execute train.py. This will use GPU if available."
+    description="Click the button to start training and see live logs below."
 )
 
 demo.launch()