Update app.py

app.py

@@ -575,32 +575,36 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             download_csv_file = gr.File(label="CSV File")
         download_csv_btn.click(fn=get_csv_file, outputs=download_csv_file)
 
-    with gr.Tab("🧪 GRPO / Dataset"):
-        gr.Markdown("""
-**GRPO Fine-tuning** (run offline or in a training Space):
-- Click **Export GRPO Preferences** to produce `data/grpo_prefs.jsonl` of (prompt, chosen, rejected).
-- Click **Write Trainer Script** to create `train/grpo_train.py`.
-- Then run:
-```bash
-pip install trl accelerate peft transformers datasets
-python train/grpo_train.py
-
-Set BASE_MODEL/OUTPUT_DIR env vars if you like.
-""")
-grpo_btn = gr.Button("📦 Export GRPO Preferences")
-grpo_status = gr.Markdown()
-grpo_btn.click(fn=export_grpo_preferences, outputs=[grpo_status])
-write_script_btn = gr.Button("📝 Write grpo_train.py")
-write_script_status = gr.Markdown()
-write_script_btn.click(fn=lambda: f"✅ Trainer script written to {_write_trainer_script()}", outputs=[write_script_status])
-gr.Markdown("---")
-gr.Markdown("### ⬇️ Download GRPO Dataset")
-
-with gr.Row():
-    download_grpo_btn = gr.Button("⬇️ Download GRPO Data (grpo_prefs.jsonl)")
-    download_grpo_file = gr.File(label="GRPO Dataset File")
+   with gr.Tab("🧪 GRPO / Dataset"):
+    gr.Markdown("""
+    **GRPO Fine-tuning** (run offline or in a training Space):
+    - Click **Export GRPO Preferences** to produce `data/grpo_prefs.jsonl` of (prompt, chosen, rejected).
+    - Click **Write Trainer Script** to create `train/grpo_train.py`.
+    - Then run:
+    ```bash
+    pip install trl accelerate peft transformers datasets
+    python train/grpo_train.py
+
+    Set BASE_MODEL/OUTPUT_DIR env vars if you like.
+    ```""")
+    
+    # These lines need to be indented to be inside the Tab's context
+    grpo_btn = gr.Button("📦 Export GRPO Preferences")
+    grpo_status = gr.Markdown()
+    grpo_btn.click(fn=export_grpo_preferences, outputs=[grpo_status])
+    
+    write_script_btn = gr.Button("📝 Write grpo_train.py")
+    write_script_status = gr.Markdown()
+    write_script_btn.click(fn=lambda: f"✅ Trainer script written to {_write_trainer_script()}", outputs=[write_script_status])
+    
+    gr.Markdown("---")
+    gr.Markdown("### ⬇️ Download GRPO Dataset")
+
+    with gr.Row():
+        download_grpo_btn = gr.Button("⬇️ Download GRPO Data (grpo_prefs.jsonl)")
+        download_grpo_file = gr.File(label="GRPO Dataset File")
     download_grpo_btn.click(fn=get_grpo_file, outputs=[download_grpo_file])
 
-
+# The `if __name__ == "__main__":` block should be at the top level
 if __name__ == "__main__":
     demo.queue(max_size=50).launch(share=True)