test stop button

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import time
 from transformers import pipeline
 import torch
-import ffmpeg  # Make sure it's ffmpeg-python
+import ffmpeg
+
 
 # Check if GPU is available
 use_gpu = torch.cuda.is_available()
@@ -23,33 +24,50 @@ def extract_audio_from_m3u8(url):
     except Exception as e:
         return f"An error occurred: {e}"
 
+def stop_transcription(state):
+    """Function to set a stop flag."""
+    state["stop_requested"] = True
+    return state
+
 def transcribe_function(audio, state, uploaded_audio, m3u8_url):
+    # Reset the stop request state at the start of transcription
+    state["stop_requested"] = False
+    
     if m3u8_url:
         audio = extract_audio_from_m3u8(m3u8_url)
-
+    
+    # Example of checking the stop state, could be integrated according to your logic
+    if state.get("stop_requested"):
+        return {"transcription_var": "Transcription stopped.", "state_var": state}
+    
     if uploaded_audio is not None:
         audio = uploaded_audio
-
+    
     if not audio:
-        return {state_var: state, transcription_var: state}  # Return a meaningful message
-
+        return {"state_var": state, "transcription_var": "Please provide an audio file."}
+    
     try:
-        time.sleep(3)
-        text = p(audio, chunk_length_s= 50)["text"]
-        state += text + "\n"
-        return {state_var: state, transcription_var: state}
+        # Simulate long-running task
+        for i in range(10):
+            if state.get("stop_requested"):
+                return {"transcription_var": "Transcription stopped.", "state_var": state}
+            time.sleep(1)  # Simulate work being done
+        text = p(audio, chunk_length_s=50)["text"]
+        state["transcription"] += text + "\n"
+        return {"state_var": state, "transcription_var": state["transcription"]}
     except Exception as e:
-        return {transcription_var: "An error occurred during transcription.", state_var: state}  # Handle other exceptions
-
-# ... [most of your code remains unchanged]
-
+        return {"transcription_var": f"An error occurred during transcription: {e}", "state_var": state}
+    
 def reset_output(transcription, state):
     """Function to reset the state to an empty string."""
     return "", ""
 
-with gr.Blocks() as demo:
-    state_var = gr.State("")
+# Your setup continues here...
 
+with gr.Blocks() as demo:
+    state_var = gr.State({"transcription": "", "stop_requested": False})
+    
+   
     with gr.Row():
         with gr.Column():
             microphone = gr.Audio(source="microphone", type="filepath", label="Microphone")
@@ -74,5 +92,13 @@ with gr.Blocks() as demo:
         [transcription_var, state_var],
         [transcription_var, state_var]
     )
+    
+    stop_button = gr.Button("Stop Transcription")
+    stop_button.click(
+        stop_transcription,
+        inputs=[state_var],
+        outputs=[state_var]
+    )
 
-demo.launch()
+# Launch the demo as usual
+demo.launch()