hina19 commited on
Commit
467346b
·
verified ·
1 Parent(s): 1f5c103

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -13
app.py CHANGED
@@ -8,6 +8,7 @@ from scenedetect import open_video, SceneManager, ContentDetector
8
  from transformers import BlipProcessor, BlipForConditionalGeneration
9
  from openai import OpenAI
10
  import base64
 
11
  # Load AI models
12
  caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
13
  caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
@@ -125,8 +126,6 @@ if uploaded_file:
125
  st.audio('summary_audio.mp3')
126
 
127
 
128
- import moviepy.editor as mp
129
-
130
  def create_summary_video(image_folder, output_video):
131
  images = sorted([os.path.join(image_folder, img) for img in os.listdir(image_folder) if img.endswith(".jpg")])
132
  clips = [mp.ImageClip(img).set_duration(2) for img in images] # 2 sec per frame
@@ -137,17 +136,17 @@ if uploaded_file:
137
  # Example usage
138
  create_summary_video("scenes", "summary_video.mp4")
139
 
140
- # Combine Audio & Video
141
- st.write("🎬 Merging audio with the video...")
142
 
143
- def add_audio_to_video(video_path, audio_path, output_video="final_video.mp4"):
144
- video = mp.VideoFileClip(video_path)
145
- audio = mp.AudioFileClip(audio_path)
146
- if audio.duration > video.duration:
147
- audio = audio.subclip(0, video.duration)
148
- final_video = video.set_audio(audio)
149
- final_video.write_videofile(output_video, codec="libx264", audio_codec="aac")
150
 
151
- add_audio_to_video("summary_video.mp4", "summary_audio.mp3")
152
 
153
- st.video("final_video.mp4")
 
8
  from transformers import BlipProcessor, BlipForConditionalGeneration
9
  from openai import OpenAI
10
  import base64
11
+ import moviepy.editor as mp
12
  # Load AI models
13
  caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
14
  caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 
126
  st.audio('summary_audio.mp3')
127
 
128
 
 
 
129
  def create_summary_video(image_folder, output_video):
130
  images = sorted([os.path.join(image_folder, img) for img in os.listdir(image_folder) if img.endswith(".jpg")])
131
  clips = [mp.ImageClip(img).set_duration(2) for img in images] # 2 sec per frame
 
136
  # Example usage
137
  create_summary_video("scenes", "summary_video.mp4")
138
 
139
+ # # Combine Audio & Video
140
+ # st.write("🎬 Merging audio with the video...")
141
 
142
+ # def add_audio_to_video(video_path, audio_path, output_video="final_video.mp4"):
143
+ # video = mp.VideoFileClip(video_path)
144
+ # audio = mp.AudioFileClip(audio_path)
145
+ # if audio.duration > video.duration:
146
+ # audio = audio.subclip(0, video.duration)
147
+ # final_video = video.set_audio(audio)
148
+ # final_video.write_videofile(output_video, codec="libx264", audio_codec="aac")
149
 
150
+ # add_audio_to_video("summary_video.mp4", "summary_audio.mp3")
151
 
152
+ # st.video("final_video.mp4")