Update app.py

app.py

@@ -4,64 +4,25 @@ import gradio as gr
 from groq import Groq
 from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 import subprocess
 
 # Ensure ImageMagick is installed
 def install_imagemagick():
     if not os.path.exists('/usr/bin/convert'):
-        subprocess.run(['sudo', 'apt-get', 'update'])
-        subprocess.run(['sudo', 'apt-get', 'install', '-y', 'imagemagick'])
-
-install_imagemagick
-
-# Initialize client with API key
-client = Groq(
-    api_key=os.getenv("Groq_Api_Key")
-)
-
-if client.api_key is None:
-    raise EnvironmentError("Groq_Api_Key environment variable is not set.")
-
-# Helper to create messages from history
-def create_history_messages(history):
-    history_messages = [{"role": "user", "content": m[0]} for m in history]
-    history_messages.extend([{"role": "assistant", "content": m[1]} for m in history])
-    return history_messages
-
-# Generate response function
-def generate_response(prompt, history, model, temperature, max_tokens, top_p, seed):
-    messages = create_history_messages(history)
-    messages.append({"role": "user", "content": prompt})
-
-    if seed == 0:
-        seed = random.randint(1, 100000)
-
-    stream = client.chat.completions.create(
-        messages=messages,
-        model=model,
-        temperature=temperature,
-        max_tokens=max_tokens,
-        top_p=top_p,
-        seed=seed,
-        stop=None,
-        stream=True,
-    )
-
-    response = ""
-    for chunk in stream:
-        delta_content = chunk.choices[0].delta.content
-        if delta_content is not None:
-            response += delta_content
-            yield response
+        subprocess.run(['apt-get', 'update'])
+        subprocess.run(['apt-get', 'install', '-y', 'imagemagick'])
 
-    return response
+install_imagemagick()
 
-# Process video function
-from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
-from PIL import Image
+def create_text_clip(text, fontsize, color, size):
+    img = Image.new('RGB', size, color='black')
+    draw = ImageDraw.Draw(img)
+    font = ImageFont.truetype("arial.ttf", fontsize)
+    w, h = draw.textsize(text, font=font)
+    draw.text(((size[0] - w) / 2, (size[1] - h) / 2), text, font=font, fill=color)
+    return np.array(img)
 
-# Adjusting MoviePy's resize function to use Image.LANCZOS directly
 def process_video(text):
     video_folder = "videos"
     video_files = [os.path.join(video_folder, f) for f in os.listdir(video_folder) if f.endswith(('mp4', 'mov', 'avi', 'mkv'))]
@@ -72,22 +33,21 @@ def process_video(text):
     video = VideoFileClip(selected_video)
     start_time = random.uniform(0, max(0, video.duration - 60))
     video = video.subclip(start_time, min(start_time + 60, video.duration))
-    
-    # Manually resize using PIL to avoid the issue
+
     def resize_image(image, new_size):
         pil_image = Image.fromarray(image)
         resized_pil = pil_image.resize(new_size[::-1], Image.LANCZOS)
         return np.array(resized_pil)
-    
+
     new_size = (1080, int(video.h * (1080 / video.w)))
     video = video.fl_image(lambda image: resize_image(image, new_size))
     video = video.crop(x1=video.w // 2 - 540, x2=video.w // 2 + 540)
 
     text_lines = text.split()
     text = "\n".join([" ".join(text_lines[i:i+8]) for i in range(0, len(text_lines), 8)])
-    
-    text_clip = TextClip(text, fontsize=70, color='white', size=video.size, method='caption')
-    text_clip = text_clip.set_position('center').set_duration(video.duration)
+
+    text_img = create_text_clip(text, fontsize=70, color='white', size=video.size)
+    text_clip = TextClip(img=text_img, size=video.size, duration=video.duration)
 
     final = CompositeVideoClip([video, text_clip])