Adding stop button

app.py

@@ -149,11 +149,7 @@ def ui_main():
                 <img src="./assets/logo_header.png" alt="Header Image" style="display: block; margin-left: auto; margin-right: auto; width: 50%;"/>
                 <h1>SynthGenAI Dataset Generator</h1>
             </div>
-            """
-        )
 
-        gr.Markdown(
-            """
             ## Overview 🧐
             SynthGenAI is designed to be modular and can be easily extended to include different API providers for LLMs and new features.
 
@@ -161,6 +157,8 @@ def ui_main():
             Interest in synthetic data generation has surged recently, driven by the growing recognition of data as a critical asset in AI development. Synthetic data generation addresses challenges by allowing us to create diverse and useful datasets using current pre-trained Large Language Models (LLMs).
 
             [GitHub Repository](https://github.com/Shekswess/synthgenai/tree/main) | [Documentation](https://shekswess.github.io/synthgenai/)
+
+            For more information on which LLMs are allowed and how they can be used, please refer to the [documentation](https://shekswess.github.io/synthgenai/llm_providers/).
             """
         )
 
@@ -214,18 +212,24 @@ def ui_main():
                 placeholder="Comma-separated environment variables (e.g., KEY1=VALUE1, KEY2=VALUE2)",
                 value="HUGGINGFACE_API_KEY=hf_1234566789912345677889, OPENAI_API_KEY=sk-1234566789912345677889",
             )
-            gr.Markdown(
-                """
-                For more information on which LLMs are allowed and how they can be used, please refer to the [documentation](https://shekswess.github.io/synthgenai/llm_providers/).
-                """
-            )
 
         generate_button = gr.Button("Generate Dataset")
-        stop_button = gr.Button("Stop Generation")
+        stop_button = gr.Button("Stop Generation", visible=False)
         output = gr.Textbox(label="Operation Result", value="")
 
+        def on_generate_click(*args):
+            generate_button.visible = False
+            stop_button.visible = True
+            return generate_synthetic_dataset(*args)
+
+        def on_stop_click():
+            stop_generation()
+            generate_button.visible = True
+            stop_button.visible = False
+            return "Dataset generation stopped."
+
         generate_button.click(
-            generate_synthetic_dataset,
+            on_generate_click,
             inputs=[
                 llm_model,
                 temperature,
@@ -244,7 +248,7 @@ def ui_main():
         )
 
         stop_button.click(
-            stop_generation,
+            on_stop_click,
             outputs=output,
         )