hina19 commited on
Commit
89ebf9d
·
verified ·
1 Parent(s): 7094f1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -13
app.py CHANGED
@@ -2,11 +2,11 @@ import streamlit as st
2
  import os
3
  import cv2
4
  import torch
5
- import moviepy.editor as mp
6
  from PIL import Image
7
  from gtts import gTTS
8
  from scenedetect import open_video, SceneManager, ContentDetector
9
- from transformers import BlipProcessor, BlipForConditionalGeneration, AutoModelForCausalLM, AutoTokenizer
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
- video = mp.VideoFileClip(video_path)
106
- audio = mp.AudioFileClip(audio_path)
107
- if audio.duration > video.duration:
108
- audio = audio.subclip(0, video.duration)
109
- final_video = video.set_audio(audio)
110
- final_video.write_videofile(output_video, codec="libx264", audio_codec="aac")
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")