Update app.py

app.py

@@ -10,14 +10,15 @@ import os
 import requests
 from PIL import Image, ImageDraw, ImageFont
 import subprocess
+import concurrent.futures
 
 # CPU-friendly summarization model
 summary_pipe = pipeline("text2text-generation", model="google/flan-t5-base", device=-1)
 llm = HuggingFacePipeline(pipeline=summary_pipe)
 
-# LangChain summarization prompt
+# LangChain summarization prompt (short summary)
 summary_prompt = PromptTemplate.from_template("""
-Summarize the following article content in a clear, concise way:
+Give a crisp and short summary of the following content (under 50 words):
 
 {text}
 
@@ -41,7 +42,7 @@ 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 '')
+    wrapped = summary_text[:512] + ('...' if len(summary_text) > 512 else '')
     draw.text((50, 50), wrapped, fill=(255, 255, 255), font=font)
     img.save(image_path)
 
@@ -51,6 +52,7 @@ def generate_video(image_path, audio_path, output_path):
         "-loop", "1",
         "-i", image_path,
         "-i", audio_path,
+        "-t", "15",
         "-c:v", "libx264",
         "-tune", "stillimage",
         "-c:a", "aac",
@@ -67,6 +69,7 @@ def url_to_av_summary(url):
         if article_text.startswith("Error"):
             return article_text, None
 
+        article_text = article_text[:3000]  # Further truncated
         summary = summary_chain.run(text=article_text)
 
         tts = gTTS(text=summary)
@@ -84,15 +87,23 @@ def url_to_av_summary(url):
     except Exception as e:
         return f"Error: {str(e)}", None
 
+def safe_summary_with_timeout(url, timeout_secs=60):
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        future = executor.submit(url_to_av_summary, url)
+        try:
+            return future.result(timeout=timeout_secs)
+        except concurrent.futures.TimeoutError:
+            return "⏱️ Processing took too long. Try a shorter article.", None
+
 iface = gr.Interface(
-    fn=url_to_av_summary,
+    fn=safe_summary_with_timeout,
     inputs=gr.Textbox(label="Article URL", placeholder="Paste a news/blog URL here..."),
     outputs=[
         gr.Textbox(label="Summary"),
         gr.Video(label="Video Summary")
     ],
-    title="URL to AV Summary Agent (No MoviePy)",
-    description="Summarizes only article content from a URL and creates a narrated video using ffmpeg + PIL."
+    title="🧠 Short AV Summary from URL",
+    description="Extracts clean article content and creates a <15 second narrated video with a short crisp summary. 100% CPU-compatible."
 )
 
 if __name__ == "__main__":