Update generate.py

generate.py

@@ -3,11 +3,10 @@ import os
 import re
 import time
 import textwrap
-import urllib.parse
 import numpy as np
 import requests
 from PIL import Image, ImageDraw, ImageFont
-from moviepy.editor import ImageClip, concatenate_videoclips, AudioFileClip, VideoFileClip
+from moviepy.editor import ImageClip, concatenate_videoclips, AudioFileClip
 from io import BytesIO
 import asyncio
 import edge_tts
@@ -18,6 +17,13 @@ FPS = 30
 DURATION_PER_LINE = 4
 PEXELS_API_KEY = os.environ.get('PEXELS_API_KEY', '')
 
+KEYWORDS = [
+    "india city", "india technology", "india traffic",
+    "india government", "india future", "india electric car",
+    "india startup", "india innovation", "india growth",
+    "india 2026", "india street", "india economy"
+]
+
 def clean_script(text):
     text = re.sub(r'\*\*.*?\*\*', '', text)
     text = re.sub(r'\(.*?\)', '', text)
@@ -29,23 +35,24 @@ def clean_script(text):
 def get_pexels_image(keyword):
     try:
         headers = {'Authorization': PEXELS_API_KEY}
-        clean_keyword = re.sub(r'#\w+', '', keyword).strip()[:50]
         r = requests.get(
-            f'https://api.pexels.com/v1/search?query={clean_keyword}&per_page=5&orientation=portrait',
+            f'https://api.pexels.com/v1/search?query={keyword}&per_page=10&orientation=portrait',
             headers=headers, timeout=15
         )
         data = r.json()
         if data.get('photos'):
-            photo_url = data['photos'][0]['src']['portrait']
+            import random
+            photo = random.choice(data['photos'])
+            photo_url = photo['src']['portrait']
             img_response = requests.get(photo_url, timeout=15)
             img = Image.open(BytesIO(img_response.content)).convert('RGB')
             img = img.resize((WIDTH, HEIGHT))
             overlay = Image.new('RGB', (WIDTH, HEIGHT), (0, 0, 0))
             img = Image.blend(img, overlay, 0.5)
-            print(f"Pexels image fetched! ✅")
+            print(f"Pexels image fetched for '{keyword}' ✅")
             return img
         else:
-            print("No Pexels photos found")
+            print(f"No photos for: {keyword}")
             return None
     except Exception as e:
         print(f"Pexels error: {e}")
@@ -105,21 +112,19 @@ def generate_video(script, title, description):
     sentences = sentences[:10]
     print(f"Total clips: {len(sentences)}")
 
-    # Get Pexels background image
-    print("Fetching Pexels background...")
-    keyword = title.replace('#shorts', '').replace('Facts', '').replace('Trending India', 'India').strip()
-    bg_image = get_pexels_image(keyword)
-    if not bg_image:
-        # Fallback keyword
-        bg_image = get_pexels_image("India")
-
     clips = []
     for i, sentence in enumerate(sentences):
         print(f"Creating clip {i+1}/{len(sentences)}: {sentence[:40]}")
 
+        # Different Pexels image for each clip
+        keyword = KEYWORDS[i % len(KEYWORDS)]
+        bg_image = get_pexels_image(keyword)
+
+        # TTS audio
         audio_path = f'/app/audio_{i}.mp3'
         has_audio = generate_tts(sentence, audio_path)
 
+        # Create frame
         frame = make_text_frame(sentence, bg_image)
 
         if has_audio and os.path.exists(audio_path):
@@ -169,4 +174,4 @@ if __name__ == '__main__':
         video_id = upload.upload_video(title, description)
         print(f"SUCCESS: https://youtube.com/shorts/{video_id}")
     else:
-        print("Video generation failed!")
+        print("Video generation failed!")