Update app.py

app.py

@@ -27,6 +27,29 @@ if not HF_TOKEN:
     st.error("HF_TOKEN not found in environment variables. Please set it in your Hugging Face Space secrets.")
     st.stop()
 
+
+
+class ProgressHook:
+    def __init__(self, status, progress_bar):
+        self.status = status
+        self.progress_bar = progress_bar
+        self.total = 0
+        self.completed = 0
+
+    def __call__(self, *args, **kwargs):
+        if 'completed' in kwargs and 'total' in kwargs:
+            self.completed = kwargs['completed']
+            self.total = kwargs['total']
+        elif len(args) == 2:
+            self.completed, self.total = args
+        
+        if self.total > 0:
+            progress_percentage = min(self.completed / self.total, 1.0)
+            self.status.update(label=f"Processing: {progress_percentage:.1%} complete", state="running")
+            self.progress_bar.progress(progress_percentage)
+
+
+
 def check_versions():
     st.info("Checking package versions...")
     
@@ -154,11 +177,7 @@ with tab1:
             with st.status("Processing audio...", expanded=True) as status:
                 progress_bar = st.progress(0)
                 
-                def progress_hook(**kwargs):
-                    if 'completed' in kwargs and 'total' in kwargs:
-                        progress_percentage = min(kwargs['completed'] / kwargs['total'], 1.0)
-                        status.update(label=f"Processing: {progress_percentage:.1%} complete", state="running")
-                        progress_bar.progress(progress_percentage)
+                progress_hook = ProgressHook(status, progress_bar)
 
                 # Run the pipeline on the audio file
                 diarization_args = {
@@ -194,6 +213,7 @@ with tab1:
         # Clean up the temporary file
         os.unlink(tmp_path)
 
+
 with tab2:
     if 'diarization' in locals():
         st.subheader("Diarization Results")