Update app.py

app.py

@@ -1,47 +1,49 @@
 import gradio as gr
-from moviepy import VideoFileClip, TextClip, CompositeVideoClip
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 from PIL import Image, ImageDraw, ImageFont
 import whisper
 from keybert import KeyBERT
 import numpy as np
+import warnings
 import os
 
-# Load Whisper and KeyBERT models
+# Suppress Whisper FP16 warning on CPU
+warnings.filterwarnings("ignore", category=UserWarning, module="whisper")
+
+# Load models
 whisper_model = whisper.load_model("base")
 kw_model = KeyBERT()
 
-# Path to Hindi-supporting font (adjust if needed)
-FONT_PATH = "/usr/share/fonts/truetype/noto/NotoSansDevanagari-Regular.ttf"  # Use system font or bundle locally
+# Path to Hindi-supporting TTF font
+FONT_PATH = "/usr/share/fonts/truetype/noto/NotoSansDevanagari-Regular.ttf"
 
 def process_video(video_path, caption="Your Caption"):
-    # Load the video
+    # Load video
     clip = VideoFileClip(video_path)
 
-    # Extract frame at 5 seconds
+    # === 1. Extract Thumbnail ===
     frame = clip.get_frame(5)
     image = Image.fromarray(np.uint8(frame))
     draw = ImageDraw.Draw(image)
 
-    # Load PIL font (for thumbnail)
     try:
-        font = ImageFont.truetype(FONT_PATH, size=40)
+        pil_font = ImageFont.truetype(FONT_PATH, size=40)
     except:
-        font = ImageFont.load_default()
-
-    draw.text((50, image.height - 100), caption, fill="white", font=font)
+        pil_font = ImageFont.load_default()
 
+    draw.text((50, image.height - 100), caption, fill="white", font=pil_font)
     thumbnail_path = "thumbnail.jpg"
     image.save(thumbnail_path)
 
-    # Transcribe audio with Whisper in Hindi
+    # === 2. Transcribe Audio in Hindi ===
     result = whisper_model.transcribe(video_path, language="hi")
     text = result["text"]
 
-    # Extract keywords using KeyBERT
+    # === 3. Extract SEO Keywords ===
     keywords = kw_model.extract_keywords(text, keyphrase_ngram_range=(1, 2), stop_words='english')
     keywords_list = [kw[0] for kw in keywords]
 
-    # Burn caption onto video
+    # === 4. Burn Caption onto Video ===
     try:
         text_clip = TextClip(
             caption,
@@ -50,31 +52,38 @@ def process_video(video_path, caption="Your Caption"):
             font=FONT_PATH,
             method='caption',
             size=(clip.w, None)
-        ).set_position(("center", "bottom")).set_duration(clip.duration)
+        )
     except:
-        # fallback in case font doesn't support Hindi
         text_clip = TextClip(
             caption,
             fontsize=50,
             color='white',
             method='caption',
             size=(clip.w, None)
-        ).set_position(("center", "bottom")).set_duration(clip.duration)
+        )
 
-    final = CompositeVideoClip([clip, text_clip])
+    text_clip = text_clip.set_position(("center", "bottom")).set_duration(clip.duration)
+
+    final_video = CompositeVideoClip([clip, text_clip])
     output_path = "output_with_caption.mp4"
-    final.write_videofile(output_path, codec="libx264", audio_codec="aac")
+    final_video.write_videofile(output_path, codec="libx264", audio_codec="aac", verbose=False, logger=None)
 
     return thumbnail_path, ", ".join(keywords_list), output_path
 
-# Gradio UI
+# === Gradio UI ===
 with gr.Blocks() as demo:
-    gr.Markdown("# 📹 Video Thumbnail & SEO Tool (Hindi Supported)")
-    video_input = gr.File(label="📁 Upload Video", type="filepath")
-    caption_input = gr.Textbox(label="📝 Caption for Thumbnail/Video", value="यह शानदार वीडियो है!")
-    generate_button = gr.Button("🚀 Generate Thumbnail, SEO & Video")
-    thumbnail_output = gr.Image(label="🖼️ Generated Thumbnail")
-    keywords_output = gr.Textbox(label="🔑 Extracted SEO Keywords")
+    gr.Markdown("# 📹 Video Thumbnail & SEO Keyword Generator (Hindi Supported)")
+
+    with gr.Row():
+        video_input = gr.File(label="📁 Upload Video", type="filepath")
+        caption_input = gr.Textbox(label="📝 Caption for Thumbnail & Video", value="यह शानदार वीडियो है!")
+
+    generate_button = gr.Button("🚀 Generate")
+
+    with gr.Row():
+        thumbnail_output = gr.Image(label="🖼️ Generated Thumbnail")
+        keywords_output = gr.Textbox(label="🔑 Extracted SEO Keywords")
+
     video_output = gr.File(label="⬇️ Download Final Video")
 
     generate_button.click(
@@ -83,4 +92,5 @@ with gr.Blocks() as demo:
         outputs=[thumbnail_output, keywords_output, video_output]
     )
 
-demo.launch(share=True)  # share=True creates public link
+demo.launch(share=True)
+