Update app.py

app.py

@@ -9,65 +9,59 @@ from tempfile import NamedTemporaryFile
 from PIL import Image
 from io import BytesIO
 
-# Constants
-UNSPLASH_IMAGES = [
-    "https://images.unsplash.com/photo-1603398938378-7c6a282eb53b",
-    "https://images.unsplash.com/photo-1588776814546-ec7e23689fcc",
-    "https://images.unsplash.com/photo-1599566150163-29194dcaad36",
-    "https://images.unsplash.com/photo-1621072157481-1a4c4e4ff57e"
-]
+# Single Unsplash image for speed
+UNSPLASH_IMAGE = "https://images.unsplash.com/photo-1603398938378-7c6a282eb53b"
 
-# Lightweight text generator
+# Lightweight text generation model
 generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
-# Remedy script generator
+# Generate remedy script
 def generate_script(topic):
-    try:
-        prompt = f"Suggest a simple 60-second home remedy for {topic}."
-        response = generator(prompt, max_new_tokens=150)
-        return response[0]["generated_text"].strip()
-    except:
-        return f"A simple home remedy for {topic} is to rest, stay hydrated, and eat a balanced diet."
+    prompt = f"Suggest a short 30-second natural home remedy for {topic}."
+    response = generator(prompt, max_new_tokens=60)
+    return response[0]["generated_text"].strip()
 
-# Voice generation
+# Generate voice using gTTS
 def generate_voice(text, lang="en"):
     tts = gTTS(text=text, lang=lang)
     temp_audio = NamedTemporaryFile(delete=False, suffix=".mp3")
     tts.save(temp_audio.name)
     return temp_audio.name
 
-# Image downloader (resized to save RAM)
-def download_images():
-    paths = []
-    for i, url in enumerate(UNSPLASH_IMAGES[:3]):
-        response = requests.get(url)
-        img = Image.open(BytesIO(response.content))
-        img = img.resize((854, 480))  # Resize to 480p width
-        path = f"image_{i+1}.jpg"
-        img.save(path)
-        paths.append(path)
-    return paths
+# Download one image
+def download_image():
+    response = requests.get(UNSPLASH_IMAGE)
+    img = Image.open(BytesIO(response.content))
+    path = "image.jpg"
+    img.save(path)
+    return path
 
-# Video creation (subtitles skipped to avoid RAM issues)
-def create_video(topic, script, audio_path, image_files):
+# Generate subtitle
+def create_subtitle(text, duration, size):
+    wrapped = textwrap.fill(text, width=50)
+    txt_clip = TextClip(wrapped, fontsize=24, color='white', bg_color='black', size=(size[0], 80))
+    return txt_clip.set_duration(duration).set_position(("center", size[1] - 100))
+
+# Generate video
+def create_video(topic, script, audio_path, image_file):
     audio_clip = AudioFileClip(audio_path)
     duration = audio_clip.duration
-    per_image_duration = duration / len(image_files)
-
-    image_clips = [ImageClip(img).set_duration(per_image_duration).resize(height=480) for img in image_files]
-    slideshow = concatenate_videoclips(image_clips, method="compose")
 
-    video = slideshow.set_audio(audio_clip)
+    img_clip = ImageClip(image_file).set_duration(duration).resize(height=480)
+    subtitle = create_subtitle(script, duration, img_clip.size)
+    
+    video = CompositeVideoClip([img_clip.set_audio(audio_clip), subtitle])
     output_path = f"/tmp/{topic.replace(' ', '_')}_video.mp4"
-    video.write_videofile(output_path, fps=24, codec="libx264", audio_codec="aac")
+    video.write_videofile(output_path, fps=15, codec="libx264", audio_codec="aac")
     return output_path
 
-# Gradio pipeline
+# Main function for Gradio
 def generate_remedy(topic, language):
     script = generate_script(topic)
-    audio = generate_voice(script, lang="ur" if language == "Urdu" else "en")
-    images = download_images()
-    video = create_video(topic, script, audio, images)
+    lang_code = "ur" if language == "Urdu" else "en"
+    audio = generate_voice(script, lang=lang_code)
+    image = download_image()
+    video = create_video(topic, script, audio, image)
     return video, script
 
 # Gradio interface
@@ -81,8 +75,8 @@ iface = gr.Interface(
         gr.Video(label="Generated Remedy Video"),
         gr.Textbox(label="Remedy Script")
     ],
-    title="RemedyReels: AI-Powered Home Remedy Video Generator",
-    description="Type a medical topic and get a narrated video remedy with Unsplash images."
+    title="RemedyReels (Fast Mode)",
+    description="Quick AI home remedy video generator with voice-over, subtitles, and Unsplash image."
 )
 
 iface.launch()