Update app.py

app.py

@@ -9,7 +9,7 @@ log_path = "train.log"
 model_dir = "trained_model"
 zip_file = "trained_model.zip"
 
-# Try loading the generator model once at the top (for better performance)
+# Try loading model for test tab
 try:
     generator = pipeline("text-generation", model=model_dir, tokenizer="distilgpt2")
 except:
@@ -32,9 +32,9 @@ def start_training():
 
     if os.path.exists(model_dir):
         shutil.make_archive("trained_model", "zip", model_dir)
-        return "✅ Training complete!", zip_file
+        return "✅ Training complete!", os.path.abspath(zip_file)
     else:
-        return "❌ Training failed.", None
+        return "❌ Training failed.", ""
 
 def read_logs():
     if os.path.exists(log_path):
@@ -51,7 +51,7 @@ def generate_response(prompt):
             max_length=256,
             do_sample=True,
             temperature=0.7,
-            truncation=True  # ✅ Fix warning and enforce consistent length
+            truncation=True
         )[0]["generated_text"]
         return result
     except Exception as e:
@@ -67,11 +67,12 @@ with gr.Blocks() as app:
 
         start_btn = gr.Button("🚀 Start Training")
         log_output = gr.Textbox(label="📜 Training Logs", lines=15)
-        download_btn = gr.File(label="📥 Download Trained Model", visible=False)
+        refresh_btn = gr.Button("🔁 Refresh Logs")
+        download_btn = gr.DownloadButton(label="📥 Download Trained Model", visible=False)
 
         upload_btn.click(fn=upload_file, inputs=file_input, outputs=status_box)
         start_btn.click(fn=start_training, outputs=[status_box, download_btn])
-        start_btn.click(fn=read_logs, outputs=log_output)
+        refresh_btn.click(fn=read_logs, outputs=log_output)
 
     with gr.Tab("🚀 Test AI"):
         gr.Markdown("## 💡 Try your trained Godot AI below")