Added route to support stop running transcription

ASR_Server.py

@@ -22,6 +22,9 @@ import timeit
 cpu_score = timeit.timeit("sum(range(1000000))", number=5)
 print(f"🧠 CPU benchmark score: {cpu_score:.2f}")
 
+# Vars for stopping running transcription:
+stop_transcription_flag = {"active": False}
+current_thread = {"thread": None}
 
 job_status = {
     "running": False,
@@ -56,6 +59,7 @@ def generateTranscript(ASR_model):
     from datetime import datetime
     import traceback
 
+    stop_transcription_flag["active"] = False
     job_status.update({
         "running": True,
         "model": ASR_model,
@@ -125,6 +129,21 @@ def generateTranscript(ASR_model):
         rtfx_score = []
 
         for idx, row in tqdm.tqdm(df.iterrows(), total=len(df)):
+            #------------------------------------------
+            if stop_transcription_flag.get("active", False):
+                job_status.update({
+                    "running": False,
+                    "model": None,
+                    "message": "Transcription cancelled by user.",
+                    "error": None,
+                    "error_trace": None,
+                    "completed": idx,
+                    "%_completed": (idx + 1) * 100 / total,
+                    "total": total
+                })
+                print("🛑 Transcription cancelled by user request.")
+                return
+            #----------------------------------------------------------
             filename = row[filename_column] + ".wav"
 
             if filename in dataset_map:
@@ -246,6 +265,13 @@ def asr_models():
 def get_status():
     return jsonify(job_status)
 
+@app.route('/api/stop', methods=['GET'])
+def stop_transcription():
+    global stop_transcription_flag
+    stop_transcription_flag["active"] = True
+    return jsonify({"message": "Stop requested. The running transcription will terminate soon or as soon as possible."})
+
+
 @app.route('/api', methods=['GET'])
 def api():
     model = request.args.get('ASR_model', default="", type=str)
@@ -291,7 +317,7 @@ def api():
             job_status["error_trace"] = None
 
             return jsonify({
-                'message': "Transcription failed.",
+                'message': f"Transcription failed (Kindly do not run this model again, untill we fix the issue): {error_msg}",
                 'error': error_msg,
                 'error_trace': error_trace,
                 'status': job_status,
@@ -305,6 +331,7 @@ def api():
         # Start the transcript generation in a separate thread
         # thread = threading.Thread(target=generateTranscript, args=(model,), daemon=True)
         thread = threading.Thread(target=generateTranscript, args=(model,))
+        current_thread["thread"] = thread
         thread.start()
 
         return response