Create app.py

app.py

@@ -0,0 +1,151 @@
+
+print("✅ Dependencies installed.")
+
+# @title Import Libraries and Define Globals
+import gradio as gr
+import os
+import shutil
+import requests
+import io
+import time
+import re
+import random
+import math
+import tempfile
+import traceback
+import numpy as np
+import soundfile as sf
+import pysrt
+import cv2
+from kokoro import KPipeline
+from moviepy.editor import (
+    VideoFileClip, AudioFileClip, ImageClip, concatenate_videoclips,
+    CompositeVideoClip, TextClip, CompositeAudioClip
+)
+import moviepy.video.fx.all as vfx
+import moviepy.config as mpy_config
+from pydub import AudioSegment
+from PIL import Image, ImageDraw, ImageFont
+from bs4 import BeautifulSoup
+from urllib.parse import quote
+from gtts import gTTS
+
+# --- Configuration ---
+PEXELS_API_KEY = 'BhJqbcdm9Vi90KqzXKAhnEHGsuFNv4irXuOjWtT761U49lRzo03qBGna'
+OPENROUTER_API_KEY = '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'
+OPENROUTER_MODEL = "mistralai/mistral-small-3.1-24b-instruct:free"
+TEMP_FOLDER = "/tmp/temp_video_processing"
+OUTPUT_VIDEO_FILENAME = "final_video.mp4"
+USER_AGENT = "Mozilla/5.0"
+
+# Initialize Kokoro TTS
+try:
+    pipeline = KPipeline(lang_code='a', device='cpu')
+    print("✅ Kokoro TTS Initialized.")
+except Exception as e:
+    print(f"⚠️ Error initializing Kokoro TTS: {e}. Using gTTS fallback.")
+    pipeline = None
+
+# --- Helper Functions ---
+def fix_imagemagick_policy():
+    try:
+        policy_paths = [
+            "/etc/ImageMagick-6/policy.xml",
+            "/etc/ImageMagick-7/policy.xml",
+            "/etc/ImageMagick/policy.xml",
+            "/usr/local/etc/ImageMagick-7/policy.xml"
+        ]
+        for path in policy_paths:
+            if os.path.exists(path):
+                os.system(f"sudo sed -i 's/rights=\"none\" pattern=\"PS\"/rights=\"read|write\" pattern=\"PS\"/' {path}")
+                os.system(f"sudo sed -i 's/rights=\"none\" pattern=\"LABEL\"/rights=\"read|write\" pattern=\"LABEL\"/' {path}")
+        return True
+    except:
+        return False
+
+fix_imagemagick_policy()
+
+# [Include all other helper functions from the original code here:
+# generate_script, parse_script, search_pexels_videos, search_pexels_images,
+# search_google_images, download_image, download_video, generate_media,
+# generate_tts, apply_kenburns_effect, resize_to_fill, add_background_music,
+# create_clip]
+
+# --- Gradio Interface ---
+def generate_video(topic, resolution, captions, bg_music):
+    status = []
+    temp_folder = f"{TEMP_FOLDER}_{int(time.time())}"
+    os.makedirs(temp_folder, exist_ok=True)
+    
+    try:
+        status.append("Generating script...")
+        script = generate_script(topic)
+        if not script:
+            return "❌ Script generation failed", None
+            
+        status.append("Parsing script...")
+        elements = parse_script(script)
+        if not elements:
+            return "❌ Script parsing failed", None
+            
+        clips = []
+        for i in range(0, len(elements), 2):
+            media_elem = elements[i]
+            tts_elem = elements[i+1]
+            
+            status.append(f"Processing: {media_elem['prompt']}")
+            media = generate_media(media_elem['prompt'])
+            if not media:
+                status.append(f"⚠️ Skipping {media_elem['prompt']} - No media found")
+                continue
+                
+            tts_path = generate_tts(tts_elem['text'], 'en')
+            if not tts_path:
+                status.append(f"⚠️ Skipping {tts_elem['text']} - TTS failed")
+                continue
+                
+            clip = create_clip(
+                media['path'],
+                media['asset_type'],
+                tts_path,
+                (1080, 1920) if resolution == "Short" else (1920, 1080),
+                duration=tts_elem['duration'],
+                narration_text=tts_elem['text'],
+                segment_index=i//2
+            )
+            if clip:
+                clips.append(clip)
+                
+        if not clips:
+            return "❌ No valid clips created", None
+            
+        final = concatenate_videoclips(clips)
+        final = add_background_music(final)
+        
+        output_path = os.path.join(temp_folder, OUTPUT_VIDEO_FILENAME)
+        final.write_videofile(output_path, codec='libx264', fps=24, preset='ultrafast')
+        
+        return "\n".join(status), output_path
+    except Exception as e:
+        return f"❌ Error: {str(e)}", None
+    finally:
+        shutil.rmtree(temp_folder, ignore_errors=True)
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=generate_video,
+    inputs=[
+        gr.Textbox(label="Video Topic", placeholder="Enter your documentary topic here..."),
+        gr.Radio(label="Resolution", choices=["Short", "Full HD"], value="Short"),
+        gr.Checkbox(label="Add Captions", value=True),
+
+    ],
+    outputs=[
+        gr.Textbox(label="Status Log"),
+        gr.Video(label="Generated Video")
+    ],
+    title="AI Documentary Generator",
+    description="Create short documentaries with AI-generated scripts and media"
+)
+
+iface.launch(debug=True)