Update app.py

app.py

@@ -1,83 +1,84 @@
-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})
-    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('Generate'):
-    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:
-    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)
-
+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})
+    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('Generate'):
+    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)
+