Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,11 +2,11 @@ import streamlit as st
|
|
2 |
import os
|
3 |
import cv2
|
4 |
import torch
|
5 |
-
|
6 |
from PIL import Image
|
7 |
from gtts import gTTS
|
8 |
from scenedetect import open_video, SceneManager, ContentDetector
|
9 |
-
from transformers import BlipProcessor, BlipForConditionalGeneration
|
10 |
from openai import OpenAI
|
11 |
# Load AI models
|
12 |
caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
@@ -97,18 +97,19 @@ if uploaded_file:
|
|
97 |
tts.save(output_audio)
|
98 |
|
99 |
text_to_speech(summary)
|
|
|
100 |
|
101 |
-
# Combine Audio & Video
|
102 |
-
st.write("🎬 Merging audio with the video...")
|
103 |
|
104 |
-
def add_audio_to_video(video_path, audio_path, output_video="final_video.mp4"):
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
|
112 |
-
add_audio_to_video(video_path, "summary_audio.mp3")
|
113 |
|
114 |
-
st.video("final_video.mp4")
|
|
|
2 |
import os
|
3 |
import cv2
|
4 |
import torch
|
5 |
+
|
6 |
from PIL import Image
|
7 |
from gtts import gTTS
|
8 |
from scenedetect import open_video, SceneManager, ContentDetector
|
9 |
+
from transformers import BlipProcessor, BlipForConditionalGeneration
|
10 |
from openai import OpenAI
|
11 |
# Load AI models
|
12 |
caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
|
|
97 |
tts.save(output_audio)
|
98 |
|
99 |
text_to_speech(summary)
|
100 |
+
st.audio('summary_audio.mp3')
|
101 |
|
102 |
+
# # Combine Audio & Video
|
103 |
+
# st.write("🎬 Merging audio with the video...")
|
104 |
|
105 |
+
# def add_audio_to_video(video_path, audio_path, output_video="final_video.mp4"):
|
106 |
+
# video = mp.VideoFileClip(video_path)
|
107 |
+
# audio = mp.AudioFileClip(audio_path)
|
108 |
+
# if audio.duration > video.duration:
|
109 |
+
# audio = audio.subclip(0, video.duration)
|
110 |
+
# final_video = video.set_audio(audio)
|
111 |
+
# final_video.write_videofile(output_video, codec="libx264", audio_codec="aac")
|
112 |
|
113 |
+
# add_audio_to_video(video_path, "summary_audio.mp3")
|
114 |
|
115 |
+
# st.video("final_video.mp4")
|