Spaces:
Runtime error
Runtime error
File size: 2,641 Bytes
1bafcb4 85f21f9 c88d358 1bafcb4 862e1e5 1bafcb4 862e1e5 1bafcb4 c984de1 325ebc1 1bafcb4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import streamlit as st
import os
from crewai import Crew, Process
from agents import get_agents_and_tasks
from tools import send_mail
def generate_video(topic, openai_api_key, stabilityai_api_key, user_mail):
os.environ['OPENAI_API_KEY'] = openai_api_key
grow_api_key = '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'
if stabilityai_api_key is not None:
os.environ['STABILITY_AI_API_KEY'] = stabilityai_api_key
model = 'Stability AI'
else:
model = 'Dalle-2'
agents, tasks = get_agents_and_tasks(grow_api_key)
crew = Crew(
agents=agents,
tasks=tasks,
process=Process.sequential,
memory=True,
verbose=2
)
result = crew.kickoff(inputs={'topic': topic, 'model' : model})
if 'tmp' in result:
result = result.split('/')[-1]
send_mail(user_mail, result)
return result
st.markdown("""
<style>
.heading {
font-size: 24px;
font-weight: bold;
text-align: center;
}
.subheading {
font-size: 18px;
font-weight: bold;
text-align: center;
margin-top: 10px;
}
</style>
""", unsafe_allow_html=True)
st.markdown("""
<div class="heading">
YouTube Shorts Creator
</div>
<div class="subheading">
Generate stunning short videos with simple text input
</div>
""", unsafe_allow_html=True)
st.text(" ")
topic = st.text_input('Enter title here')
openai_api_key = st.text_input('Enter your OpenAI API key here')
size = st.selectbox('Size',
('512 X 512', '1024 X 1024', '9:16'))
stabilityai_api_key = None
if size=='9:16':
stabilityai_api_key = st.text_input('Enter your stability ai API key here')
mail = st.text_input('Enter you email address')
if st.button('submit'):
st.text(f"Video will be sent to {mail}")
result = generate_video(topic, openai_api_key, stabilityai_api_key, mail)
with open(result, 'rb') as video_file:
video_data = video_file.read()
st.video(video_data)
# Sidebar for Example Videos
st.sidebar.markdown("### Example Videos")
example_paths = os.listdir('results')
examples = [os.path.join('results', i) for i in example_paths]
# Display videos in a row
st.sidebar.markdown('<div class="example-video-container">', unsafe_allow_html=True)
for video_url in examples:
# print(video_url)
title = video_url.split('/')[1].split('.')[0]
st.sidebar.text(f"input: {title}")
st.sidebar.video(video_url, format="video/mp4", start_time=0)
st.sidebar.markdown("</div>", unsafe_allow_html=True)
|