Create app.py

app.py

@@ -0,0 +1,198 @@
+import streamlit as st
+from pathlib import Path
+import requests
+import json
+from typing import List, Dict, Optional
+from dataclasses import dataclass
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip, AudioFileClip, ColorClip
+import textwrap
+from PIL import Image
+import io
+import random
+from transformers import pipeline
+import os
+
+@dataclass
+class ContentTheme:
+    name: str
+    keywords: List[str]
+    prompts: List[str]
+    
+THEMES = [
+    ContentTheme(
+        name="Discipline & Success",
+        keywords=["discipline", "success", "achievement", "goals"],
+        prompts=[
+            "Share a powerful insight about the role of discipline in achieving success",
+            "Explain how successful people maintain their daily routines",
+            "Describe the connection between self-discipline and reaching goals"
+        ]
+    ),
+    ContentTheme(
+        name="Psychology & Mindset",
+        keywords=["psychology", "mindset", "growth", "mental-strength"],
+        prompts=[
+            "Reveal a psychological principle that can transform someone's life",
+            "Explain a cognitive bias that holds people back",
+            "Share a mindset shift that leads to breakthrough success"
+        ]
+    ),
+    ContentTheme(
+        name="Motivation & Drive",
+        keywords=["motivation", "inspiration", "drive", "ambition"],
+        prompts=[
+            "Generate a powerful quote about never giving up",
+            "Share an insight about maintaining motivation during tough times",
+            "Explain what separates winners from losers in life"
+        ]
+    )
+]
+
+class ContentGenerator:
+    def __init__(self, pixabay_api_key: str, elevenlabs_api_key: str):
+        self.pixabay_api_key = pixabay_api_key
+        self.elevenlabs_api_key = elevenlabs_api_key
+        self.generator = pipeline('text-generation', model='gpt2')
+        
+    def generate_text_content(self, theme: ContentTheme) -> Dict[str, str]:
+        """Generate AI text content based on theme"""
+        prompt = random.choice(theme.prompts)
+        main_text = self.generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
+        
+        # Generate an accompanying quote
+        quote_prompt = f"Generate an inspirational quote about {random.choice(theme.keywords)}"
+        quote = self.generator(quote_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+        
+        return {
+            "main_text": main_text,
+            "quote": quote
+        }
+    
+    def fetch_background_media(self, theme: ContentTheme) -> str:
+        """Fetch background image/video from Pixabay"""
+        keyword = random.choice(theme.keywords)
+        url = f"https://pixabay.com/api/videos/?key={self.pixabay_api_key}&q={keyword}"
+        response = requests.get(url)
+        data = response.json()
+        
+        if data["hits"]:
+            video_url = data["hits"][0]["videos"]["medium"]["url"]
+            # Download video
+            video_response = requests.get(video_url)
+            with open("temp_background.mp4", "wb") as f:
+                f.write(video_response.content)
+            return "temp_background.mp4"
+        return None
+    
+    def generate_voiceover(self, text: str, output_path: str = "voiceover.mp3"):
+        """Generate voiceover using ElevenLabs"""
+        url = "https://api.elevenlabs.io/v1/text-to-speech"
+        headers = {
+            "Authorization": f"Bearer {self.elevenlabs_api_key}",
+            "Content-Type": "application/json"
+        }
+        data = {
+            "text": text,
+            "voice_settings": {
+                "stability": 0.5,
+                "similarity_boost": 0.5
+            }
+        }
+        
+        response = requests.post(url, json=data, headers=headers)
+        if response.status_code == 200:
+            with open(output_path, "wb") as f:
+                f.write(response.content)
+            return output_path
+        return None
+
+    def create_video(self, 
+                    background_path: str, 
+                    text_content: Dict[str, str],
+                    audio_path: str,
+                    output_path: str = "output.mp4"):
+        """Create final video with all elements"""
+        # Load background
+        background = VideoFileClip(background_path).resize(width=1080)
+        duration = 15  # Standard Instagram Reels duration
+        
+        # Create text clips
+        main_text = TextClip(
+            txt=textwrap.fill(text_content["main_text"], width=30),
+            fontsize=40,
+            color='white',
+            font='Arial-Bold',
+            stroke_color='black',
+            stroke_width=2
+        ).set_position(('center', 'center')).set_duration(duration)
+        
+        quote = TextClip(
+            txt=textwrap.fill(text_content["quote"], width=35),
+            fontsize=30,
+            color='white',
+            font='Arial',
+            stroke_color='black',
+            stroke_width=1
+        ).set_position(('center', 0.8), relative=True).set_duration(duration)
+        
+        # Compose video
+        video = CompositeVideoClip([
+            background,
+            main_text,
+            quote
+        ])
+        
+        # Add audio
+        if audio_path:
+            audio = AudioFileClip(audio_path)
+            video = video.set_audio(audio)
+        
+        # Write final video
+        video.write_videofile(output_path, fps=30)
+        return output_path
+
+def main():
+    st.title("Advanced Instagram Content Generator")
+    
+    # API Keys (in practice, these should be in environment variables)
+    pixabay_api_key = st.sidebar.text_input("Pixabay API Key", type="password")
+    elevenlabs_api_key = st.sidebar.text_input("ElevenLabs API Key", type="password")
+    
+    theme_names = [theme.name for theme in THEMES]
+    selected_theme_name = st.selectbox("Select Content Theme", theme_names)
+    selected_theme = next(theme for theme in THEMES if theme.name == selected_theme_name)
+    
+    if st.button("Generate Content"):
+        if not pixabay_api_key or not elevenlabs_api_key:
+            st.error("Please provide both API keys")
+            return
+            
+        with st.spinner("Generating your content..."):
+            generator = ContentGenerator(pixabay_api_key, elevenlabs_api_key)
+            
+            # Generate all content components
+            text_content = generator.generate_text_content(selected_theme)
+            background_path = generator.fetch_background_media(selected_theme)
+            voiceover_path = generator.generate_voiceover(text_content["main_text"])
+            
+            if all([background_path, voiceover_path]):
+                output_path = generator.create_video(
+                    background_path,
+                    text_content,
+                    voiceover_path
+                )
+                
+                # Show preview and download button
+                st.video(output_path)
+                with open(output_path, "rb") as file:
+                    st.download_button(
+                        label="Download Video",
+                        data=file,
+                        file_name="instagram_content.mp4",
+                        mime="video/mp4"
+                    )
+            else:
+                st.error("Error generating content. Please try again.")
+
+if __name__ == "__main__":
+    main()