Update app.py

app.py

@@ -8,7 +8,8 @@ from bs4 import BeautifulSoup
 import tempfile
 import os
 import requests
-from moviepy.editor import *
+from PIL import Image, ImageDraw, ImageFont
+import subprocess
 
 # CPU-friendly summarization model
 summary_pipe = pipeline("text2text-generation", model="google/flan-t5-base", device=-1)
@@ -36,6 +37,30 @@ def extract_main_content(url):
     except Exception as e:
         return f"Error extracting article content: {str(e)}"
 
+def create_text_image(summary_text, image_path):
+    img = Image.new("RGB", (1280, 720), color=(0, 0, 0))
+    draw = ImageDraw.Draw(img)
+    font = ImageFont.load_default()
+    wrapped = summary_text[:1024] + ('...' if len(summary_text) > 1024 else '')
+    draw.text((50, 50), wrapped, fill=(255, 255, 255), font=font)
+    img.save(image_path)
+
+def generate_video(image_path, audio_path, output_path):
+    cmd = [
+        "ffmpeg", "-y",
+        "-loop", "1",
+        "-i", image_path,
+        "-i", audio_path,
+        "-c:v", "libx264",
+        "-tune", "stillimage",
+        "-c:a", "aac",
+        "-b:a", "192k",
+        "-pix_fmt", "yuv420p",
+        "-shortest",
+        output_path
+    ]
+    subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
 def url_to_av_summary(url):
     try:
         article_text = extract_main_content(url)
@@ -44,18 +69,15 @@ def url_to_av_summary(url):
 
         summary = summary_chain.run(text=article_text)
 
-        # Generate speech using gTTS
         tts = gTTS(text=summary)
         audio_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3").name
         tts.save(audio_path)
 
-        # Create video clip (text overlay + audio)
-        video_clip = TextClip(summary, fontsize=32, color='white', bg_color='black', size=(1280, 720), method='caption')
-        video_clip = video_clip.set_duration(AudioFileClip(audio_path).duration)
-        video_clip = video_clip.set_audio(AudioFileClip(audio_path))
-
+        image_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
         video_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
-        video_clip.write_videofile(video_path, fps=24, codec='libx264')
+
+        create_text_image(summary, image_path)
+        generate_video(image_path, audio_path, video_path)
 
         return summary, video_path
 
@@ -69,8 +91,8 @@ iface = gr.Interface(
         gr.Textbox(label="Summary"),
         gr.Video(label="Video Summary")
     ],
-    title="URL to AV Summary Agent",
-    description="Summarizes only article content from a URL and creates a narrated video. CPU-only."
+    title="URL to AV Summary Agent (No MoviePy)",
+    description="Summarizes only article content from a URL and creates a narrated video using ffmpeg + PIL."
 )
 
 if __name__ == "__main__":