Removing stop option

app.py

@@ -2,7 +2,6 @@
 
 import os
 import asyncio
-from huggingface_hub import HfFolder
 
 import gradio as gr
 from synthgenai import DatasetConfig, DatasetGeneratorConfig, LLMConfig, InstructionDatasetGenerator, PreferenceDatasetGenerator,RawDatasetGenerator,SentimentAnalysisDatasetGenerator, SummarizationDatasetGenerator, TextClassificationDatasetGenerator
@@ -143,14 +142,8 @@ def generate_synthetic_dataset(
     else:
         return "Invalid dataset type"
 
-    stop_event.clear()
-
     async def generate():
-        if stop_event.is_set():
-            return "Dataset generation stopped."
         dataset = await generator.agenerate_dataset()
-        if stop_event.is_set():
-            return "Dataset generation stopped."
         dataset.save_dataset(hf_repo_name=hf_repo_name)
         return "Dataset generated and saved successfully."
 
@@ -244,20 +237,11 @@ def ui_main():
             )
 
         generate_button = gr.Button("Generate Dataset")
-        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(
             on_generate_click,
             inputs=[
@@ -278,11 +262,6 @@ def ui_main():
             outputs=output,
         )
 
-        stop_button.click(
-            on_stop_click,
-            outputs=output,
-        )
-
     demo.launch(inbrowser=True, favicon_path=None)