Update app.py

app.py

@@ -133,7 +133,7 @@ def main():
     st.title("GPT4V AI Voiceover 🎥🔮")
     st.text("Explore how GPT4V changes the way we voiceover videos.")
 
- # Check password before proceeding
+    # Check password before proceeding
     if not check_password():
         return  # Exit the function if the password check fails
     
@@ -144,7 +144,7 @@ def main():
         return  # or handle the error as you see fit
         
     uploaded_file = st.file_uploader("Select a video file", type=["mp4", "avi"])
-
+    
     option = st.selectbox(
         'Choose the voice you want',
         ('Female Voice', 'Male Voice'))
@@ -157,32 +157,35 @@ def main():
     if uploaded_file is not None:
         st.video(uploaded_file)
         p = 'Generate a short voiceover script for the video, matching the content with the video scenes. The style should be...'
-        # # Ignore and don't generate anything else than the script that you'll voice over the video.
-        prompt = st.text_area(
-            "Prompt", value=p
-        )
-    
-    if st.button("START PROCESSING", type="primary") and uploaded_file is not None:
-        with st.spinner("Video is being processed..."):
-            base64Frame, video_filename, video_duration = video_to_frames(uploaded_file)
-            est_word_count = video_duration * 4
-            final_prompt = prompt + f"(This video is ONLY {video_duration} seconds long. So make sure the voiceover MUST be able to be explained in less than {est_word_count} words. Ignore and don't generate anything else than the script that you'll use to voice over the video.)"
-            text = frames_to_story(base64Frame, final_prompt, openai_key)
-            st.write(text)
-            # Generate audio from text
-            audio_filename, audio_bytes_io = text_to_audio(text, openai_key, classify)
-            # Merge audio and video
-            output_video_filename = os.path.splitext(video_filename)[0] + "_output.mp4"
-            
-            final_video_filename = merge_audio_video(video_filename, audio_filename, output_video_filename)
-            
-            # Display the result
-            st.video(final_video_filename)
-            
-            # Clean up the temporary files
-            os.unlink(video_filename)
-            os.unlink(audio_filename)
-            os.unlink(final_video_filename)
+        prompt = st.text_area("Prompt", value=p)
+
+        if st.button("START PROCESSING", type="primary"):
+            with st.spinner("Video is being processed..."):
+                base64Frame, video_filename, video_duration = video_to_frames(uploaded_file)
+                
+                # Check if the video duration exceeds 30 seconds
+                if video_duration > 30:
+                    st.error("The video exceeds the maximum allowed duration of 30 seconds.")
+                    return  # Stop processing further
+                
+                est_word_count = video_duration * 4
+                final_prompt = prompt + f"(This video is ONLY {video_duration} seconds long. So make sure the voiceover MUST be able to be explained in less than {est_word_count} words. Ignore and don't generate anything else than the script that you'll use to voice over the video.)"
+                text = frames_to_story(base64Frame, final_prompt, openai_key)
+                st.write(text)
+                # Generate audio from text
+                audio_filename, audio_bytes_io = text_to_audio(text, openai_key, classify)
+                # Merge audio and video
+                output_video_filename = os.path.splitext(video_filename)[0] + "_output.mp4"
+                
+                final_video_filename = merge_audio_video(video_filename, audio_filename, output_video_filename)
+                
+                # Display the result
+                st.video(final_video_filename)
+                
+                # Clean up the temporary files
+                os.unlink(video_filename)
+                os.unlink(audio_filename)
+                os.unlink(final_video_filename)
 
 if __name__ == "__main__":
-    main()
+    main()