Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import random
 import requests
 import gradio as gr
-from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, TextClip, CompositeVideoClip
+from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeVideoClip
 from moviepy.audio.fx.all import audio_loop
 import edge_tts
 import asyncio
@@ -21,7 +21,7 @@ def run_async(coro):
     loop.close()
     return result
 
-# Load local text generation model (adjust based on Hugging Face Spaces resources)
+# Load local text generation model
 try:
     generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", device="cpu")
 except Exception as e:
@@ -53,7 +53,7 @@ def generate_script(prompt, custom_text=None):
         except Exception as e:
             print(f"Error generating script: {e}")
     
-    # Fallback mock response if model fails
+    # Fallback mock response
     if "recetas" in prompt.lower():
         return """
         1. Tacos al pastor: Jugosa carne marinada con piña.
@@ -75,19 +75,6 @@ async def generate_voice(text, output_file="output.mp3"):
     await communicate.save(output_file)
     return output_file
 
-# Generate subtitles
-def generate_subtitles(text, duration, fps=24):
-    words = text.split()
-    avg_duration = duration / len(words)
-    subtitles = []
-    for i, word in enumerate(words):
-        start_time = i * avg_duration
-        end_time = (i + 1) * avg_duration
-        subtitle = TextClip(word, fontsize=30, color='white', bg_color='black', size=(1280, 100), font='Arial')
-        subtitle = subtitle.set_position(('center', 'bottom')).set_duration(avg_duration).set_fps(fps)
-        subtitles.append(subtitle)
-    return subtitles
-
 # Download and trim video
 def download_and_trim_video(url, duration, output_path):
     response = requests.get(url, stream=True)
@@ -138,15 +125,11 @@ def create_video(prompt, custom_text, music_file):
     if music_file:
         music = AudioFileClip(music_file.name)
         music = audio_loop(music, duration=video_duration)
-        final_audio = final_clip.set_audio(music.set_duration(video_duration))
+        final_clip = final_clip.set_audio(music.set_duration(video_duration))
     else:
-        final_audio = final_clip.set_audio(audio)
-
-    # Add subtitles
-    subtitles = generate_subtitles(script, video_duration)
-    final_clip = CompositeVideoClip([final_clip] + subtitles)
+        final_clip = final_clip.set_audio(audio)
 
-    # Write final video
+    # Write final video (no subtitles to avoid ImageMagick error)
     final_clip.write_videofile(output_video, codec="libx264", audio_codec="aac", fps=24)
     
     # Clean up