Update app.py

app.py

@@ -204,6 +204,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
                 feedback_submit = gr.Button("Submit Feedback")
                 feedback_output = gr.Textbox(label="Feedback Status")
 
+    with gr.Row():
+        with gr.Column():
+            refine_btn = gr.Button("Run Iterative Refinement (Stage 4)", variant="primary")
+            refine_status = gr.Textbox(label="Refinement Status", placeholder="Status will appear here...")
+
     def handle_classification(comment, history):
         if history is None:
             history = []
@@ -289,6 +294,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
 
         return f"Thank you for your feedback: {feedback}\nAdditional comment: {additional_comment if additional_comment else 'None'}\nFeedback has been saved."
 
+    def run_refinement():
+        try:
+            from refine_paraphrases import main
+            main()
+            return "Refinement complete. Results saved to iterated_paraphrases.csv and pushed to JanviMl/toxi_iterated_paraphrases."
+        except Exception as e:
+            return f"Error running refinement: {str(e)}"
+
     submit_btn.click(
         fn=lambda: (
             "Classifying... <span class='loading'>⏳</span>", 0, "", None, "", "", 
@@ -342,6 +355,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
         ]
     )
 
+    refine_btn.click(
+        fn=run_refinement,
+        inputs=[],
+        outputs=[refine_status]
+    )
+
     gr.Markdown(
         """
         ---