Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import time
 from transformers import pipeline
 import torch
-import ffmpeg
-
+import ffmpeg  # Make sure it's ffmpeg-python
 
 # Check if GPU is available
 use_gpu = torch.cuda.is_available()
@@ -24,50 +23,33 @@ 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": "Please provide an audio file."}
-    
+        return {state_var: state, transcription_var: state}  # Return a meaningful message
+
     try:
-        # 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"]}
+        time.sleep(3)
+        text = p(audio, chunk_length_s= 50)["text"]
+        state += text + "\n"
+        return {state_var: state, transcription_var: state}
     except Exception as e:
-        return {"transcription_var": f"An error occurred during transcription: {e}", "state_var": state}
-    
+        return {transcription_var: "An error occurred during transcription.", state_var: state}  # Handle other exceptions
+
+# ... [most of your code remains unchanged]
+
 def reset_output(transcription, state):
     """Function to reset the state to an empty string."""
     return "", ""
 
-# Your setup continues here...
-
 with gr.Blocks() as demo:
-    state_var = gr.State({"transcription": "", "stop_requested": False})
-    
-   
+    state_var = gr.State("")
+
     with gr.Row():
         with gr.Column():
             microphone = gr.Audio(source="microphone", type="filepath", label="Microphone")
@@ -92,13 +74,5 @@ 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]
-    )
 
-# Launch the demo as usual
 demo.launch()