Update generate.py

generate.py

@@ -4,8 +4,8 @@ import re
 import textwrap
 import numpy as np
 import requests
-from PIL import Image, ImageDraw, ImageFont, ImageFilter
-from moviepy.editor import ImageClip, concatenate_videoclips, AudioFileClip, CompositeVideoClip
+from PIL import Image, ImageDraw, ImageFont
+from moviepy.editor import ImageClip, concatenate_videoclips, AudioFileClip
 from io import BytesIO
 import upload
 
@@ -15,7 +15,6 @@ DURATION_PER_LINE = 3
 PEXELS_API_KEY = os.environ.get('PEXELS_API_KEY', '')
 
 def get_background_image(keyword):
-    """Fetch background image from Pexels"""
     try:
         headers = {'Authorization': PEXELS_API_KEY}
         r = requests.get(
@@ -28,7 +27,6 @@ def get_background_image(keyword):
             img_response = requests.get(photo_url, timeout=15)
             img = Image.open(BytesIO(img_response.content)).convert('RGB')
             img = img.resize((WIDTH, HEIGHT))
-            # Darken image for text readability
             overlay = Image.new('RGB', (WIDTH, HEIGHT), (0, 0, 0))
             img = Image.blend(img, overlay, 0.55)
             return img
@@ -37,41 +35,29 @@ def get_background_image(keyword):
     return None
 
 def make_text_frame(text, bg_image=None):
-    """Create frame with background image and text"""
     if bg_image:
         img = bg_image.copy()
     else:
         img = Image.new('RGB', (WIDTH, HEIGHT), color=(15, 15, 25))
-
     draw = ImageDraw.Draw(img)
-
     try:
         font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 65)
     except:
         font = ImageFont.load_default()
-
-    # Draw accent lines
     draw.rectangle([60, 180, WIDTH-60, 193], fill=(255, 80, 80))
     draw.rectangle([60, HEIGHT-193, WIDTH-60, HEIGHT-180], fill=(255, 80, 80))
-
-    # Wrap text
     lines = textwrap.wrap(text, width=22)
     y = HEIGHT // 2 - (len(lines) * 85) // 2
-
     for line in lines:
         bbox = draw.textbbox((0, 0), line, font=font)
         w = bbox[2] - bbox[0]
         x = (WIDTH - w) // 2
-        # Shadow
         draw.text((x+4, y+4), line, font=font, fill=(0, 0, 0))
-        # Text
         draw.text((x, y), line, font=font, fill=(255, 255, 255))
         y += 90
-
     return np.array(img)
 
 def generate_tts(text, output_path):
-    """Generate voiceover using Google TTS"""
     try:
         from gtts import gTTS
         tts = gTTS(text=text, lang='en', slow=False)
@@ -84,37 +70,22 @@ def generate_tts(text, output_path):
 def generate_video(script, title, description):
     print("Starting video generation...")
     print(f"Script received: {script[:100]}")
-
-    # Split script into sentences
     sentences = re.split(r'[.!?\n]', script)
     sentences = [s.strip() for s in sentences if len(s.strip()) > 5]
-
     if not sentences:
         words = script.split()
         sentences = [' '.join(words[i:i+8]) for i in range(0, len(words), 8)]
-
     sentences = sentences[:15]
     print(f"Total clips: {len(sentences)}")
-
-    # Get background image from Pexels
     keyword = title.replace('#shorts', '').replace('Facts', '').strip()
     print(f"Fetching background for: {keyword}")
     bg_image = get_background_image(keyword)
-
     clips = []
-    audio_clips = []
-
     for i, sentence in enumerate(sentences):
         print(f"Creating clip {i+1}/{len(sentences)}")
-
-        # Generate TTS audio
         audio_path = f'/app/audio_{i}.mp3'
         has_audio = generate_tts(sentence, audio_path)
-
-        # Create video frame
         frame = make_text_frame(sentence, bg_image)
-        
-        # Get audio duration
         if has_audio and os.path.exists(audio_path):
             audio = AudioFileClip(audio_path)
             duration = max(audio.duration + 0.5, DURATION_PER_LINE)
@@ -122,17 +93,13 @@ def generate_video(script, title, description):
             clip = clip.set_audio(audio)
         else:
             clip = ImageClip(frame, duration=DURATION_PER_LINE)
-
         clip = clip.fadein(0.3).fadeout(0.3)
         clips.append(clip)
-
     if not clips:
         print("No clips generated!")
         return False
-
     print("Combining clips...")
     final = concatenate_videoclips(clips, method="compose")
-
     output_path = '/app/video.mp4'
     print("Writing video...")
     final.write_videofile(
@@ -143,14 +110,11 @@ def generate_video(script, title, description):
         verbose=False,
         logger=None
     )
-
-    # Cleanup audio files
     for i in range(len(sentences)):
         try:
             os.remove(f'/app/audio_{i}.mp3')
         except:
             pass
-
     print("Video generated successfully!")
     return True
 
@@ -164,20 +128,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!")
-```
-
----
-
-## Step 4 — Update `requirements.txt`:
-```
-moviepy==1.0.3
-Pillow
-numpy
-imageio
-imageio-ffmpeg
-gtts
-requests
-google-api-python-client
-google-auth
-google-auth-httplib2
+        print("Video generation failed!")