Update app.py

app.py

@@ -1,3 +1,5 @@
+
+
 import gradio as gr
 from deep_translator import GoogleTranslator
 import os
@@ -6,11 +8,8 @@ import subprocess
 from PIL import Image, ImageDraw, ImageFont
 import re
 import uuid
-# Path to the ZIP file
 zip_file = "./fonts.zip"
-# Destination folder
 extract_to = "./fonts/"
-# Extract the ZIP file
 shutil.unpack_archive(zip_file, extract_to, 'zip')
 
 # Generate unique filename for the video
@@ -138,11 +137,12 @@ def get_video_duration(video_file):
     else:
         raise ValueError("Could not extract video duration.")
 
+import json
 
 
 
 # Generate video function
-def generate_video(input_text, language_set, font_size, theme, canvas_size):
+def generate_video(input_text, language_set, font_size, theme, canvas_size,llm_translation=None):
     width, height = map(int, canvas_size.split('x'))
     theme_colors = {"Black Background": ("#FFFFFF", "#000000"), "White Background": ("#000000", "#FFFFFF")}
     text_color, bg_color = theme_colors[theme]
@@ -165,19 +165,31 @@ def generate_video(input_text, language_set, font_size, theme, canvas_size):
     FRAMES_PER_IMAGE = round(FPS * DURATION_PER_IMAGE)  # Frames needed per image
     extra_frames = FRAMES_PER_IMAGE+10  # Extra frames for the last image to allow fade-out effect
     frame_index = 0  # Start numbering frames
+    if llm_translation:
+        data = json.loads(llm_translation)
+        language_list = list(data.keys())
+        for i, lang in enumerate(language_list):
+            translated_text = data[lang]
+            img_path = create_image(translated_text, lang, font_size, text_color, bg_color, width, height, output_folder)
+            frame_count = FRAMES_PER_IMAGE + extra_frames if i == len(language_list) - 1 else FRAMES_PER_IMAGE
+            # Duplicate frames for smooth video
+            for _ in range(frame_count):
+                frame_filename = os.path.join(frames, f"{frame_index:05d}.png")
+                shutil.copy(img_path, frame_filename)
+                frame_index += 1
+    else:
+        for i, lang in enumerate(language_list):
+            translated_text = translate_text(input_text, lang) if lang != 'en' else input_text
+            img_path = create_image(translated_text, lang, font_size, text_color, bg_color, width, height, output_folder)
 
-    for i, lang in enumerate(language_list):
-        translated_text = translate_text(input_text, lang) if lang != 'en' else input_text
-        img_path = create_image(translated_text, lang, font_size, text_color, bg_color, width, height, output_folder)
-
-        # Check if it's the last image
-        frame_count = FRAMES_PER_IMAGE + extra_frames if i == len(language_list) - 1 else FRAMES_PER_IMAGE
+            # Check if it's the last image
+            frame_count = FRAMES_PER_IMAGE + extra_frames if i == len(language_list) - 1 else FRAMES_PER_IMAGE
 
-        # Duplicate frames for smooth video
-        for _ in range(frame_count):
-            frame_filename = os.path.join(frames, f"{frame_index:05d}.png")
-            shutil.copy(img_path, frame_filename)
-            frame_index += 1
+            # Duplicate frames for smooth video
+            for _ in range(frame_count):
+                frame_filename = os.path.join(frames, f"{frame_index:05d}.png")
+                shutil.copy(img_path, frame_filename)
+                frame_index += 1
 
     # Generate video using FFmpeg
     output_video = "multi_language_video.mp4"
@@ -240,13 +252,14 @@ def ui():
                 with gr.Accordion('🖌️ Text Style', open=False):
                     font_size = gr.Slider(20, 200, value=100, step=1, label="🔠 Font Size")  # FIXED (Removed comma)
                     theme = gr.Radio(["Black Background", "White Background"], label="🎨 Theme", value="Black Background")
+                    llm_translation = gr.Textbox(label='📝 LLM Translation', lines=5, placeholder="Enter a word...",value=None)
 
             with gr.Column():
                 download_video = gr.File(label="📥 Download Video")
                 play_video = gr.Video(label="🎬 Generated Video")
 
         # Define Inputs and Outputs
-        input_list = [input_text, language_set, font_size, theme, canvas_size]
+        input_list = [input_text, language_set, font_size, theme, canvas_size,llm_translation]
         output_list = [download_video, play_video]
 
         # Bind Functions to UI Elements
@@ -258,9 +271,59 @@ def ui():
 
     return demo
 
+
+def prompt_maker(text, language_set="Foreign Languages"):
+    languages = {
+        "Foreign Languages": {
+    "en": "English", "zh-CN": "Mandarin Chinese", "hi": "Hindi", "es": "Spanish",
+    "fr": "French", "ar": "Standard Arabic", "bn": "Bengali", "pt": "Portuguese",
+    "ru": "Russian", "ur": "Urdu", "id": "Indonesian", "de": "German", "ja": "Japanese",
+    "pa": "Punjabi", "te": "Telugu", "tr": "Turkish", "ta": "Tamil", "vi": "Vietnamese", "ko": "Korean"
+    },
+        "Local Indian Languages": {
+        "en": "English","hi": "Hindi", "bn": "Bengali", "mr": "Marathi", "te": "Telugu", "ta": "Tamil",
+        "gu": "Gujarati", "ur": "Urdu", "kn": "Kannada", "or": "Odia", "pa": "Punjabi", "ml": "Malayalam",
+        "mai": "Maithili","ne": "Nepali","sa": "Sanskrit","doi": "Dogri","sd": "Sindhi"
+
+    }
+    }
+
+    selected_languages = languages.get(language_set, languages["Foreign Languages"])
+    
+    prompt = f'Translate this English word "{text}" to different languages in JSON format. Only write the JSON text:\n'
+    prompt += "{\n"
+    prompt += ",\n".join([f' "{code}": ""' for code in selected_languages])
+    prompt += "\n}"
+
+    return prompt
+
+def ui2():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 🌍 LLM Translation")
+        
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label='📝 Enter a Word', lines=1, placeholder="Enter a word...")
+                language_set = gr.Radio(
+                    ["Foreign Languages", "Local Indian Languages"], 
+                    label="🌍 Language Set", 
+                    value="Foreign Languages"
+                )
+                generate_btn = gr.Button('🚀 Generate', variant='primary')
+            
+            with gr.Column():
+                output_text = gr.Textbox(label='📝 Generated Prompt', lines=5)
+
+        generate_btn.click(prompt_maker, inputs=[input_text, language_set], outputs=output_text)
+
+    return demo
+
+
 # Launch the app
-def main(share=False, debug=False):
-    demo = ui()
+def main(share=False, debug=True):
+    demo1 = ui()
+    demo2 = ui2()
+    demo=gr.TabbedInterface([demo1, demo2], ["Video Generation", "LLM Prompt"])
     demo.queue().launch(debug=debug, share=share)
 
 main()