Update app.py

app.py

@@ -11,11 +11,12 @@ import json
 import psutil
 import sys
 from pathlib import Path
+import glob
 
 # Workaround for torch.classes and Streamlit compatibility
-# Exclude torch modules from Streamlit's path watcher
 st._is_running_with_streamlit = True
-sys.modules['torch'].__path__ = []
+if 'torch' in sys.modules and hasattr(sys.modules['torch'], '__path__'):
+    sys.modules['torch'].__path__ = []
 
 st.set_page_config(layout="wide")
 
@@ -214,6 +215,7 @@ footer {
     color: #ffffff;
     padding: 3rem 2rem;
     margin-top: 3rem;
+Desk
     border-radius: 1rem 1rem 0 0;
 }
 .footer-container {
@@ -335,7 +337,7 @@ def transcribe_audio(audio, sr, processor, model, device, start_time, language,
             input_features = input_features.half()
         generate_kwargs = {
             "task": task,
-            "language": "urdu" if langage == "ur" else language,
+            "language": "urdu" if language == "ur" else language,
             "max_new_tokens": 128,
             "return_timestamps": True,
             "do_sample": False
@@ -357,7 +359,10 @@ def process_chunks(chunks, sr, processor, model, device, language, chunk_duratio
     progress = TranscriptionProgress()
     progress.init_progress()
     if os.path.exists(transcript_file):
-        os.remove(transcript_file)
+        try:
+            os.remove(transcript_file)
+        except Exception as e:
+            st.warning(f"Failed to remove {transcript_file}: {str(e)}")
     for i, chunk in enumerate(chunks):
         progress.update((i + 1) / total_chunks, f"Processing chunk {i+1}/{total_chunks}...")
         try:
@@ -387,7 +392,9 @@ def summarize_text(text, tokenizer, model, device, summarizer_type='bart'):
     try:
         inputs = tokenizer(text, return_tensors="pt", truncation=False)
         input_ids = inputs["input_ids"].to(device)
-        attention_mask = inputs.get("attention_mask").to(device)
+        attention_mask = inputs.get("attention_mask")
+        if attention_mask is not None:
+            attention_mask = attention_mask.to(device)
         num_tokens = input_ids.shape[1]
         st.write(f"Number of tokens in input: {num_tokens}")
         if num_tokens < 50:
@@ -464,10 +471,10 @@ def merge_intervals(intervals):
     return merged
 
 def create_edited_video(video_path, transcript, keep_indices):
+    temp_files = []
     try:
         intervals_to_keep = [(transcript[i][1], transcript[i][2]) for i in keep_indices]
         merged_intervals = merge_intervals(intervals_to_keep)
-        temp_files = []
         for j, (start, end) in enumerate(merged_intervals):
             temp_file = f"temp_{j}.mp4"
             ffmpeg.input(video_path, ss=start, to=end).output(temp_file, c='copy').run(overwrite_output=True, quiet=True)
@@ -484,9 +491,15 @@ def create_edited_video(video_path, transcript, keep_indices):
     finally:
         for temp_file in temp_files:
             if os.path.exists(temp_file):
-                os.remove(temp_file)
+                try:
+                    os.remove(temp_file)
+                except Exception as e:
+                    st.warning(f"Failed to remove {temp_file}: {str(e)}")
         if os.path.exists("list.txt"):
-            os.remove("list.txt")
+            try:
+                os.remove("list.txt")
+            except Exception as e:
+                st.warning(f"Failed to remove list.txt: {str(e)}")
 
 def generate_srt(transcript, include_timeframe=True):
     srt_content = ""
@@ -505,13 +518,16 @@ def cleanup_temp_files():
         if os.path.exists(temp_file):
             try:
                 os.remove(temp_file)
-            except:
-                pass
+                st.info(f"Removed temporary file: {temp_file}")
+            except Exception as e:
+                st.warning(f"Failed to remove {temp_file}: {str(e)}")
     for temp_file in glob.glob("temp_*.mp4"):
-        try:
-            os.remove(temp_file)
-        except:
-            pass
+        if os.path.exists(temp_file):
+            try:
+                os.remove(temp_file)
+                st.info(f"Removed temporary file: {temp_file}")
+            except Exception as e:
+                st.warning(f"Failed to remove {temp_file}: {str(e)}")
 
 # Main Function
 def main():
@@ -570,7 +586,7 @@ def main():
         st.markdown("<div id='upload'></div>", unsafe_allow_html=True)
         st.markdown("<h3 style='text-align: center; color: black;'>Upload Your Video</h3>", unsafe_allow_html=True)
         with st.form(key="upload_form"):
-            uploaded_file = st.file_uploader("Choose a video file", type=["mp4"], label_visibility="collapsed")
+            uploaded_file = st.file_uploader(" Succeeded uploading file: 1
             if st.form_submit_button("Upload") and uploaded_file:
                 video_path = save_uploaded_file(uploaded_file)
                 if video_path:
@@ -732,9 +748,17 @@ def main():
     if st.session_state.get('video_path') and st.button("Reset"):
         cleanup_temp_files()
         if st.session_state['video_path'] and os.path.exists(st.session_state['video_path']):
-            os.remove(st.session_state['video_path'])
+            try:
+                os.remove(st.session_state['video_path'])
+                st.info(f"Removed video file: {st.session_state['video_path']}")
+            except Exception as e:
+                st.warning(f"Failed to remove video file: {str(e)}")
         if st.session_state['edited_video_path'] and os.path.exists(st.session_state['edited_video_path']):
-            os.remove(st.session_state['edited_video_path'])
+            try:
+                os.remove(st.session_state['edited_video_path'])
+                st.info(f"Removed edited video file: {st.session_state['edited_video_path']}")
+            except Exception as e:
+                st.warning(f"Failed to remove edited video file: {str(e)}")
         st.session_state.clear()
         st.rerun()