Update app.py

app.py

@@ -1,99 +1,91 @@
 import os
 import requests
-import pyttsx3
+import textwrap
+from moviepy.editor import *
+from gtts import gTTS
+import gradio as gr
 from transformers import pipeline
+from tempfile import NamedTemporaryFile
 from PIL import Image
 from io import BytesIO
-from moviepy.editor import *
-import textwrap
 
-# ========== CONFIG ==========
-HF_API_KEY = os.getenv("Hugging_API_KEY")
-TTS_MODEL = "facebook/fastspeech2-en-ljspeech"  # Use Hugging Face TTS model
+# 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"
 ]
-# ============================
 
-# --- Generate remedy text from Hugging Face ---
+# Hugging Face model
+generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
+
+# Remedy script generator
 def generate_script(topic):
-    print("🔠 Generating remedy script from Hugging Face...")
-    generator = pipeline("text-generation", model="gpt2")  # Use GPT-2 for script generation
-    return generator(f"Give a simple 60-second home remedy for {topic} in natural language.", max_length=150)[0]['generated_text'].strip()
+    prompt = f"Suggest a simple 60-second home remedy for {topic}."
+    response = generator(prompt, max_new_tokens=150)
+    return response[0]["generated_text"].strip()
 
-# --- Generate voiceover using Hugging Face TTS or offline ---
-def generate_voiceover(text, filename="voice.mp3"):
-    print("🔊 Generating voiceover...")
-    
-    # If you want online TTS (Hugging Face)
-    tts = pipeline("text-to-speech", model=TTS_MODEL, tokenizer=TTS_MODEL)
-    tts(text, path=filename)
-    
-    # If using offline TTS (uncomment if preferred)
-    # engine = pyttsx3.init()
-    # engine.setProperty('rate', 150)
-    # engine.save_to_file(text, filename)
-    # engine.runAndWait()
-    
-    return filename
+# Voice generation
+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
 
-# --- Download images (3–5) ---
+# Image downloader
 def download_images():
-    print("🖼️ Downloading images...")
-    image_paths = []
-    for i, url in enumerate(UNSPLASH_IMAGES[:4]):
+    paths = []
+    for i, url in enumerate(UNSPLASH_IMAGES[:3]):
         response = requests.get(url)
         img = Image.open(BytesIO(response.content))
         path = f"image_{i+1}.jpg"
         img.save(path)
-        image_paths.append(path)
-    return image_paths
+        paths.append(path)
+    return paths
 
-# --- Create subtitle text clip ---
-def create_subtitle(text, duration, video_size):
+# Subtitle generator
+def create_subtitle(text, duration, size):
     wrapped = textwrap.fill(text, width=60)
-    txt_clip = TextClip(wrapped, fontsize=30, color='white', bg_color='black', size=(video_size[0], 100))
-    return txt_clip.set_duration(duration).set_position(("center", video_size[1] - 120))
+    txt_clip = TextClip(wrapped, fontsize=28, color='white', bg_color='black', size=(size[0], 100))
+    return txt_clip.set_duration(duration).set_position(("center", size[1] - 100))
 
-# --- Create slideshow video with audio and subtitles ---
+# Video creation
 def create_video(topic, script, audio_path, image_files):
-    print("🎞️ Creating final video...")
     audio_clip = AudioFileClip(audio_path)
     duration = audio_clip.duration
     per_image_duration = duration / len(image_files)
 
-    image_clips = []
-    for path in image_files:
-        img_clip = ImageClip(path).set_duration(per_image_duration).resize(height=720)
-        image_clips.append(img_clip)
-
+    image_clips = [ImageClip(img).set_duration(per_image_duration).resize(height=720) for img in image_files]
     slideshow = concatenate_videoclips(image_clips, method="compose")
-    video_size = slideshow.size
-    subtitle = create_subtitle(script, duration, video_size)
-    final = CompositeVideoClip([slideshow.set_audio(audio_clip), subtitle])
 
-    # Optional background music if 'music.mp3' exists
-    if os.path.exists("music.mp3"):
-        print("🎵 Adding background music...")
-        music = AudioFileClip("music.mp3").volumex(0.2).set_duration(duration)
-        final_audio = CompositeAudioClip([audio_clip.volumex(1.0), music])
-        final = final.set_audio(final_audio)
+    subtitle = create_subtitle(script, duration, slideshow.size)
+    video = CompositeVideoClip([slideshow.set_audio(audio_clip), subtitle])
+    output_path = f"/tmp/{topic.replace(' ', '_')}_video.mp4"
+    video.write_videofile(output_path, fps=24)
+    return output_path
 
-    filename = f"{topic.replace(' ', '_')}_remedy_video.mp4"
-    final.write_videofile(filename, fps=24)
-    print(f"✅ Video saved as: {filename}")
-
-# --- Main function ---
-def generate_remedy_video():
-    topic = input("Enter a home remedy topic (e.g., hair fall): ")
+# Gradio pipeline
+def generate_remedy(topic, language):
     script = generate_script(topic)
-    audio_file = generate_voiceover(script)
+    audio = generate_voice(script, lang="ur" if language == "Urdu" else "en")
     images = download_images()
-    create_video(topic, script, audio_file, images)
+    video = create_video(topic, script, audio, images)
+    return video, script
+
+# Gradio interface
+iface = gr.Interface(
+    fn=generate_remedy,
+    inputs=[
+        gr.Textbox(label="Enter medical condition (e.g., cough, headache)"),
+        gr.Radio(["English", "Urdu"], label="Narration Language")
+    ],
+    outputs=[
+        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 subtitles and Unsplash images."
+)
 
-# --- Entry point ---
-if __name__ == "__main__":
-    generate_remedy_video()
+iface.launch()