Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
145 |
-
|
146 |
-
|
147 |
-
|
148 |
-
|
149 |
-
|
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")
|