generative_AI / src /page8.py
alok94's picture
adding project
561acc5
import streamlit as st
# import openai
import replicate
import os
from dotenv import load_dotenv
from streamlit_extras.stylable_container import stylable_container
import streamlit_extras
# load_dotenv()
# REPLICATE_API_TOKEN = os.environ.get("REPLICATE_API_TOKEN")
# replicate = replicate.Client(api_token=REPLICATE_API_TOKEN)
streamlit_style = """
<style>
#MainMenu {visibility: hidden;}
footer {visibility: hidden;}
video{width:200px;}
.css-1wbqy5l {visibility: hidden;}
.css-15zrgzn {visibility: hidden;}
.css-klqnuk {visibility: hidden;}
.en6cib64 {visibility: hidden;}
.css-1u4fkce {visibility: hidden;}
.en6cib62 {visibility: hidden;}
.css-19rxjzo, .ef3psqc11 {
background-color: purple;
text-color: white;
}
div.stButton > button:first-child {
background-color: darkgreen;
text-weight: bold;
}
</style>
"""
def page8():
with stylable_container(
key="title",
css_styles=[
""" span {
text-align: center;
padding-top: 0px;
padding-right: 0px;
padding-bottom: 0px;
padding-left: 0px;
}"""
,
"""
st-emotion-cache-0{
text-align: center;
padding-top: 0px;
padding-right: 0px;
padding-bottom: 0px;
padding-left: 0px;
}""",
"""
.e1f1d6gn0{
text-align: center;
padding-top: 0px;
padding-right: 0px;
padding-bottom: 0px;
padding-left: 0px;
}
""",
],
):
st.markdown("<h3>Text to video</h3>", unsafe_allow_html=True) #This is under a css style
st.markdown(streamlit_style, unsafe_allow_html=True)
with st.form(key='form'):
prompt = st.text_input(label='Enter text prompt for Video generation')
placeholder=st.empty()
col1,col2=placeholder.columns(2)
video_length = col1.number_input("Enter video length (seconds)", step=1, min_value=1, max_value=8, value=4, placeholder="Type a number...")
submit_button = st.form_submit_button(label='Generate Video')
if submit_button:
if prompt:
placeholder=st.empty()
col1,col2=placeholder.columns(2)
with st.spinner("Generating Video from Text. Please wait...."):
output = replicate.run(
"cjwbw/text2video-zero:e671ffe4e976c0ec813f15a9836ebcfd08857ac2669af6917e3c2549307f9fae",
input={
"fps": 4,
"prompt": prompt,
"model_name": "dreamlike-art/dreamlike-photoreal-2.0",
"timestep_t0": 44,
"timestep_t1": 47,
"video_length": video_length,
"negative_prompt": "",
"motion_field_strength_x": 12,
"motion_field_strength_y": 12
}
)
col1.text("Generated Video")
col1.video(output)
st.markdown(
"""
<script>
const video = document.querySelector('video');
video.loop = true;
video.autoplay = true;
</script>
""",
unsafe_allow_html=True,
)