Update app.py

app.py

@@ -295,6 +295,11 @@ class TranscriptionProgress:
             self.progress_bar.progress(progress)
         if self.status_text is not None:
             self.status_text.text(status)
+    def clear(self):
+        if self.status_text is not None:
+            self.status_text.text("Processing complete!")
+        if self.progress_bar is not None:
+            self.progress_bar.empty()
 
 @st.cache_resource
 def load_model(language='en', summarizer_type='bart'):
@@ -344,12 +349,12 @@ def process_chunks(chunks, sr, processor, model, device, language, chunk_duratio
     transcript = []
     chunk_start = 0
     total_chunks = len(chunks)
-    progress_bar = st.progress(0)
-    status_text = st.empty()
+    progress = TranscriptionProgress()
+    progress.init_progress()
     if os.path.exists(transcript_file):
         os.remove(transcript_file)
     for i, chunk in enumerate(chunks):
-        status_text.text(f"Processing chunk {i+1}/{total_chunks}...")
+        progress.update((i + 1) / total_chunks, f"Processing chunk {i+1}/{total_chunks}...")
         try:
             memory = psutil.virtual_memory()
             st.write(f"Memory usage: {memory.percent}% (Chunk {i+1}/{total_chunks})")
@@ -358,12 +363,10 @@ def process_chunks(chunks, sr, processor, model, device, language, chunk_duratio
             with open(transcript_file, "w", encoding="utf-8") as f:
                 json.dump(transcript, f, ensure_ascii=False)
             chunk_start += chunk_duration
-            progress_bar.progress((i + 1) / total_chunks)
         except Exception as e:
             st.error(f"Error processing chunk {i+1}: {str(e)}")
             break
-    status_text.text("Processing complete!")
-    progress_bar.empty()
+    progress.clear()
     return transcript
 
 def summarize_text(text, tokenizer, model, device, summarizer_type='bart'):