Update app.py

app.py

@@ -35,6 +35,17 @@ from transformers import (
 # Medical knowledge validation
 import re
 
+import subprocess
+import gradio as gr
+
+def train_model():
+    result = subprocess.run(["python", "train_flan_t5.py"], capture_output=True, text=True)
+    if result.returncode == 0:
+        return "✅ Model training complete!"
+    else:
+        return f"❌ Training failed:\n{result.stderr}"
+
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -843,6 +854,9 @@ def create_optimized_gradio_interface():
                             variant="secondary",
                             scale=1
                         )
+                        train_btn = gr.Button("Train")
+                        train_output = gr.Textbox(label="Training Status", lines=10)
+                        train_btn.click(train_model, outputs=train_output)
             
             with gr.Column(scale=1):
                 # Quick Access