Update app.py

app.py

@@ -24,26 +24,31 @@ from io import BytesIO
 
 class ImageScraper:
     def __init__(self):
-        # Get your free API key from https://pixabay.com/api/docs/
-        self.PIXABAY_API_KEY = "48069976-37e20099248207cee12385560"  # Replace with your key
-        
+        self.headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+            'Accept-Language': 'en-US,en;q=0.5',
+            'Connection': 'keep-alive',
+        }
+
+    def get_stock_images(self) -> List[str]:
+        """Return preset stock images relevant to digital security and technology"""
+        return [
+            "https://images.pexels.com/photos/60504/security-protection-anti-virus-software-60504.jpeg",
+            "https://images.pexels.com/photos/5380642/pexels-photo-5380642.jpeg",
+            "https://images.pexels.com/photos/2582937/pexels-photo-2582937.jpeg",
+            "https://images.pexels.com/photos/7319074/pexels-photo-7319074.jpeg",
+            "https://images.pexels.com/photos/4164418/pexels-photo-4164418.jpeg",
+            "https://images.pexels.com/photos/3861969/pexels-photo-3861969.jpeg",
+            "https://images.pexels.com/photos/5473298/pexels-photo-5473298.jpeg",
+            "https://images.pexels.com/photos/4348401/pexels-photo-4348401.jpeg",
+            "https://images.pexels.com/photos/8386440/pexels-photo-8386440.jpeg",
+            "https://images.pexels.com/photos/5473950/pexels-photo-5473950.jpeg"
+        ]
+
     def get_images(self, query: str, num_images: int = 15) -> List[str]:
-        try:
-            # Add relevant keywords for tech/security images
-            search_query = f"{query} digital technology security"
-            
-            url = f"https://pixabay.com/api/?key={self.PIXABAY_API_KEY}&q={search_query.replace(' ', '+')}&image_type=photo&per_page={num_images}"
-            
-            response = requests.get(url)
-            data = response.json()
-            
-            if 'hits' in data:
-                return [img['largeImageURL'] for img in data['hits']]
-            return []
-            
-        except Exception as e:
-            print(f"Pixabay API error: {e}")
-            return []
+        # Return preset stock images regardless of query
+        return self.get_stock_images()
             
     def scrape_pexels(self, query: str) -> List[str]:
         urls = []
@@ -557,60 +562,98 @@ class VideoGeneratorUI:
         self.setup_ui()
 
     def setup_ui(self):
-        st.title("Enhanced Video Generator")
+        st.set_page_config(layout="wide")
         
-        # Step 1: Input prompt
-        prompt = st.text_input("Enter your video topic/prompt")
+        # Custom CSS
+        st.markdown("""
+            <style>
+            .stApp {
+                max-width: 1200px;
+                margin: 0 auto;
+            }
+            .image-grid {
+                display: grid;
+                grid-template-columns: repeat(auto-fill, minmax(250px, 1fr));
+                gap: 1rem;
+            }
+            .stButton>button {
+                width: 100%;
+            }
+            </style>
+        """, unsafe_allow_html=True)
+
+        # Header
+        st.title("VaultGenix Video Generator")
+        st.markdown("Create professional videos for your digital legacy management platform")
         
-        if prompt:
-            # Step 2: Image Selection
-            st.subheader("Select Images for Your Video")
-            images = self.generator.image_scraper.get_images(prompt)
+        # Input Section
+        with st.container():
+            prompt = st.text_area("Enter your video script", height=200)
             
-            if not images:
-                st.warning("No images found. Try a different search term.")
-                return
+            if prompt:
+                st.subheader("Select Images for Your Video")
+                images = self.generator.image_scraper.get_images(prompt)
                 
-            selected_images = []
-            cols = st.columns(3)
-            for idx, img_url in enumerate(images):
-                with cols[idx % 3]:
-                    try:
-                        response = requests.get(img_url)
-                        img = Image.open(BytesIO(response.content))
-                        st.image(img, use_column_width=True)
-                        if st.checkbox(f"Select Image {idx + 1}", key=f"img_{idx}"):
-                            selected_images.append(img_url)
-                    except:
-                        continue
-            
-            # Step 3: Video Generation (only show if images are selected)
-            if selected_images:
-                st.subheader("Video Generation Settings")
-                col1, col2 = st.columns(2)
-                with col1:
-                    style = st.selectbox("Choose style", options=list(self.generator.themes.keys()))
-                with col2:
-                    duration = st.slider("Video duration (seconds)", 10, 300, 60, 10)
-
-                if st.button("Generate Video"):
-                    try:
-                        output_path = os.path.join(os.getcwd(), f"generated_video_{int(time.time())}.mp4")
-                        video_path = self.generator.create_video(prompt, style, duration, output_path, selected_images)
-                        
-                        if os.path.exists(video_path):
-                            st.success("Video generated successfully!")
-                            st.video(video_path)
-                            
-                            with open(video_path, 'rb') as video_file:
-                                st.download_button(
-                                    "Download Video",
-                                    video_file.read(),
-                                    file_name=os.path.basename(video_path),
-                                    mime="video/mp4"
+                if not images:
+                    st.warning("Loading default technology images...")
+                    images = self.generator.image_scraper.get_stock_images()
+                
+                # Image Selection Grid
+                selected_images = []
+                cols = st.columns(3)
+                for idx, img_url in enumerate(images):
+                    with cols[idx % 3]:
+                        try:
+                            st.image(img_url, use_column_width=True)
+                            if st.checkbox(f"Select Image {idx + 1}", key=f"img_{idx}"):
+                                selected_images.append(img_url)
+                        except:
+                            continue
+                
+                # Video Generation Settings
+                if selected_images:
+                    st.subheader("Video Settings")
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        style = st.selectbox(
+                            "Choose style",
+                            options=["Professional", "Creative", "Educational"],
+                            index=0
+                        )
+                    with col2:
+                        duration = st.slider(
+                            "Video duration (seconds)",
+                            min_value=30,
+                            max_value=180,
+                            value=60,
+                            step=30
+                        )
+
+                    if st.button("Generate Video", type="primary"):
+                        with st.spinner("Generating your video..."):
+                            try:
+                                output_path = f"vaultgenix_video_{int(time.time())}.mp4"
+                                video_path = self.generator.create_video(
+                                    prompt,
+                                    style,
+                                    duration,
+                                    output_path,
+                                    selected_images
                                 )
-                    except Exception as e:
-                        st.error(f"Failed to generate video: {str(e)}")
+                                
+                                if os.path.exists(video_path):
+                                    st.success("✨ Video generated successfully!")
+                                    st.video(video_path)
+                                    
+                                    with open(video_path, 'rb') as video_file:
+                                        st.download_button(
+                                            "⬇️ Download Video",
+                                            video_file.read(),
+                                            file_name=output_path,
+                                            mime="video/mp4"
+                                        )
+                            except Exception as e:
+                                st.error(f"Failed to generate video: {str(e)}")
 
 if __name__ == "__main__":
     ui = VideoGeneratorUI()