Update app.py

app.py

@@ -63,14 +63,14 @@ class TrainingState:
         self.status = "idle"
         self.progress = 100
 
-    def get_status(self):
-        m = {
-            "idle": "✅ Ready",
-            "training": "🏋️ Training",
-            "fine-tuning": "🎯 Fine-tuning"
-        }
-        return m.get(self.status, "❓ Unknown") + (f" - {self.progress}%" if self.status != "idle" else "")
-
+    with gr.Tab("📊 Status"):
+        with gr.Row():
+            status_box = gr.Textbox(label="Current Status", interactive=False)
+            progress_bar = gr.Slider(minimum=0, maximum=1, value=0, step=0.01, interactive=False, label="Progress")
+        log_output = gr.Textbox(label="Logs", lines=10, interactive=False)
+        refresh_btn = gr.Button("🔄 Refresh Status")
+        refresh_btn.click(get_current_status, outputs=[status_box, progress_bar, log_output])
+        
 state = TrainingState()
 
 def test_model(text):