Update app.py

app.py

@@ -6,59 +6,76 @@ from keybert import KeyBERT
 import numpy as np
 import os
 
-# Load models
+# Load Whisper and KeyBERT 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
+
 def process_video(video_path, caption="Your Caption"):
-    # Load video
+    # Load the video
     clip = VideoFileClip(video_path)
 
-    # Extract a frame at 5 seconds and save thumbnail
+    # Extract frame at 5 seconds
     frame = clip.get_frame(5)
     image = Image.fromarray(np.uint8(frame))
     draw = ImageDraw.Draw(image)
-    font = ImageFont.load_default()
-    text_position = (50, image.height - 100)
-    draw.text(text_position, caption, fill="white", font=font)
+
+    # Load PIL font (for thumbnail)
+    try:
+        font = ImageFont.truetype(FONT_PATH, size=40)
+    except:
+        font = ImageFont.load_default()
+
+    draw.text((50, image.height - 100), caption, fill="white", font=font)
+
     thumbnail_path = "thumbnail.jpg"
     image.save(thumbnail_path)
 
-    # Transcribe Hindi audio
+    # Transcribe audio with Whisper in Hindi
     result = whisper_model.transcribe(video_path, language="hi")
     text = result["text"]
 
-    # Extract keywords from Hindi text
+    # Extract keywords using KeyBERT
     keywords = kw_model.extract_keywords(text, keyphrase_ngram_range=(1, 2), stop_words='english')
     keywords_list = [kw[0] for kw in keywords]
 
-    # Burn caption into video
-    txt_clip = TextClip(
-    txt=caption,
-    fontsize=50,
-    color='white',
-    font='Arial',
-    method='caption',
-    size=(clip.w, None)
-    ).set_position(("center", "bottom")).set_duration(clip.duration)
-
-    final = CompositeVideoClip([clip, txt_clip])
+    # Burn caption onto video
+    try:
+        text_clip = TextClip(
+            caption,
+            fontsize=50,
+            color='white',
+            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)
 
-    # Save processed video
+    final = CompositeVideoClip([clip, text_clip])
     output_path = "output_with_caption.mp4"
-    final.write_videofile(output_path, codec='libx264', audio_codec='aac')
+    final.write_videofile(output_path, codec="libx264", audio_codec="aac")
 
     return thumbnail_path, ", ".join(keywords_list), output_path
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# Video Thumbnail Generator with SEO Keywords")
-    video_input = gr.File(label="Upload Video", type="filepath")
-    caption_input = gr.Textbox(label="Enter Caption for Thumbnail", value="Awesome Video!")
-    generate_button = gr.Button("Generate Thumbnail & Keywords")
-    thumbnail_output = gr.Image(label="Generated Thumbnail")
-    keywords_output = gr.Textbox(label="SEO Keywords")
-    video_output = gr.File(label="Download Processed Video")
+    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")
+    video_output = gr.File(label="⬇️ Download Final Video")
 
     generate_button.click(
         fn=process_video,
@@ -66,4 +83,4 @@ with gr.Blocks() as demo:
         outputs=[thumbnail_output, keywords_output, video_output]
     )
 
-demo.launch()
+demo.launch(share=True)  # share=True creates public link