Update app.py

app.py

@@ -1,25 +1,39 @@
 import streamlit as st
 from moviepy.editor import VideoFileClip, CompositeVideoClip
-import ffmpeg  # Make sure to install ffmpeg-python
+import ffmpeg
 import tempfile
 import os
+import shutil
 
-def add_hard_subtitle_to_video(input_video, subtitle_file, subtitle_language):
-    video_input_stream = ffmpeg.input(input_video)
+def save_uploaded_file(uploaded_file):
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp_file:
+            shutil.copyfileobj(uploaded_file, tmp_file)
+            return tmp_file.name
+    except Exception as e:
+        st.error(f"Error saving file: {e}")
+        return None
+
+def add_hard_subtitle_to_video(input_video_path, subtitle_file_path):
     output_video = tempfile.mktemp(suffix=".mp4")
     stream = ffmpeg.output(
-        video_input_stream, output_video,
-        vf=f"subtitles={subtitle_file}:force_style='FontName=Arial,FontSize=24'"
+        ffmpeg.input(input_video_path),
+        ffmpeg.input(subtitle_file_path),
+        output_video,
+        vf=f"subtitles={subtitle_file_path}:force_style='FontName=Arial,FontSize=24'"
     )
     ffmpeg.run(stream, overwrite_output=True)
     return output_video
 
 def video_demo(video, subtitle):
-    if subtitle is not None:
-        processed_video_path = add_hard_subtitle_to_video(video.name, subtitle.name, "en")  # Assuming English subtitles
+    video_path = save_uploaded_file(video)
+    subtitle_path = save_uploaded_file(subtitle)
+    
+    if video_path and subtitle_path:
+        processed_video_path = add_hard_subtitle_to_video(video_path, subtitle_path)
         return processed_video_path
     else:
-        return video
+        return None
 
 # Streamlit UI
 st.title("Video Subtitler")
@@ -27,9 +41,11 @@ st.title("Video Subtitler")
 uploaded_video = st.file_uploader("Upload a video", type=["mp4", "avi", "mov"])
 uploaded_subtitle = st.file_uploader("Upload subtitle file", type=["srt", "vtt"])
 
-if uploaded_video is not None and uploaded_subtitle is not None:
+if uploaded_video and uploaded_subtitle:
     processed_video_path = video_demo(uploaded_video, uploaded_subtitle)
-    
-    st.video(processed_video_path)
+    if processed_video_path:
+        st.video(processed_video_path)
+    else:
+        st.write("An error occurred. Please try again.")
 else:
-    st.write("Please upload both a video and a subtitle file.")
+    st.write("Please upload both a video and a subtitle file.")