Enhance font management with Google Fonts integration, automatic clea…

.gitignore

@@ -10,12 +10,17 @@ ENV/
 temp_uploads/
 temp_outputs/
 
+# Downloaded Fonts (regenerated on demand)
+assets/fonts/
+
 # Python Cache
 __pycache__/
 *.py[cod]
 *$py.class
 *.so
 *.pyc
+*.pyo
+.pytest_cache/
 
 # Distribution / Packaging
 .Python
@@ -54,6 +59,10 @@ Thumbs.db
 # Video Cache
 *.cache
 
+# MoviePy Temp Files
+TEMP/
+*.tmp
+
 # Test Files
 test_output/
 test_images/

README.md

@@ -1,13 +1,13 @@
 # 🎬 PyMontage - Automatic Video Slideshow Creator
 
-![Version](https://img.shields.io/badge/version-1.0.0-blue.svg)
+![Version](https://img.shields.io/badge/version-1.0.1-blue.svg)
 ![Python](https://img.shields.io/badge/python-3.8+-blue.svg)
 ![Flask](https://img.shields.io/badge/flask-3.0.0-green.svg)
 ![License](https://img.shields.io/badge/license-MIT-orange.svg)
 
 **PyMontage** is a powerful web-based application that automatically creates professional video slideshows from your photos and music. Upload your images, select a soundtrack, and let PyMontage do the rest!
 
-**Version 1.0 Release** - Production ready with full feature set including audio crossfading, smart layouts, and comprehensive customization options.
+**Version 1.0.1 Release** - Enhanced with Google Fonts integration, automatic server cleanup, and improved font management system.
 
 ## ✨ Features
 
@@ -18,7 +18,9 @@
 - 📊 **Real-time Progress Bar**: Track video creation progress with detailed status updates
 - ⚙️ **Full Customization**: Control resolution, timing, transitions, fonts, and quality settings
 - 🎬 **Professional Output**: High-quality video output with smooth transitions and title cards
-- 🔤 **Custom Fonts**: Choose from multiple font options for titles and overlays
+- 🔤 **Custom Fonts**: Choose from built-in Windows fonts or download from Google Fonts
+- 🔎 **Font Search**: Search and instantly download any font from Google Fonts library
+- 🧹 **Auto Cleanup**: Automatic cleanup of temporary files and cache on shutdown
 - 🌐 **Web Interface**: Easy-to-use browser-based interface with drag-and-drop support
 - 📱 **Format Support**: Supports JPG, PNG, HEIC, GIF, BMP, TIFF, WebP, and more
 - 🎥 **Hardware Acceleration**: Automatic GPU detection for faster rendering (NVIDIA NVENC)
@@ -137,19 +139,22 @@
   - Bitrate: 2000k (low) to 15000k (ultra)
   - CRF Quality: 18 (best quality) to 32 (smaller file)
 - **Text Styling**:
-  - Font Family: Choose from 10+ Windows fonts
+  - Font Family: Choose from built-in Windows fonts or search and download from Google Fonts
   - Title Font Size: Size for intro/outro text (30-200)
   - Date Font Size: Size for date overlays (20-150)
+  - Font Search: Live search to find and download fonts from Google Fonts library
 - **Image Processing**:
   - Max Image Width: Downscale images to save memory (1920-4800 pixels)
 
 ## 🛠️ Technical Details
 
 ### Architecture
-- **Backend**: Flask (Python web framework)
+- **Backend**: Flask (Python web framework) with signal handlers for graceful shutdown
 - **Video Processing**: MoviePy, OpenCV, FFmpeg
 - **Audio Analysis**: librosa
-- **Frontend**: HTML5, Bootstrap 5, vanilla JavaScript
+- **Frontend**: HTML5, Bootstrap 5, vanilla JavaScript with async font management
+- **Font Management**: Google Fonts API integration with local caching and automatic downloads
+- **Cleanup System**: Automatic cleanup of temporary files, cache, and bytecode on exit
 
 ### File Structure
 ```

app.py

@@ -3,48 +3,215 @@
 import uuid
 import atexit
 import signal
-from flask import Flask, render_template, request, send_file, after_this_request
-from video_engine import generate_video_from_web  # Import from video_engine.py
+import requests
+import json
+import tempfile
+from flask import Flask, render_template, request, send_file, after_this_request, jsonify
+from video_engine import generate_video_from_web, install_google_font  # Import from video_engine.py
 
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'temp_uploads'
 app.config['OUTPUT_FOLDER'] = 'temp_outputs'
-
-# Create temporary folders if they don't exist
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 os.makedirs(app.config['OUTPUT_FOLDER'], exist_ok=True)
 
+
 def cleanup_temp_folders():
-    """Clean up all temporary folders when server shuts down"""
-    print("\n🧹 Cleaning up temporary folders...")
+    """Clean up all temporary folders and cache when server shuts down"""
+    print("\n🧹 Cleaning up temporary files and caches...")
+
+    folders_to_remove = [
+        app.config['UPLOAD_FOLDER'],
+        app.config['OUTPUT_FOLDER'],
+        '__pycache__',
+        '.pytest_cache',
+    ]
+
     try:
-        if os.path.exists(app.config['UPLOAD_FOLDER']):
-            shutil.rmtree(app.config['UPLOAD_FOLDER'])
-            print(f"  ✓ Removed {app.config['UPLOAD_FOLDER']}")
-        if os.path.exists(app.config['OUTPUT_FOLDER']):
-            shutil.rmtree(app.config['OUTPUT_FOLDER'])
-            print(f"  ✓ Removed {app.config['OUTPUT_FOLDER']}")
-        print("✓ Cleanup complete!")
+        # Remove temp folders
+        for folder in folders_to_remove:
+            if os.path.exists(folder):
+                if os.path.isdir(folder):
+                    shutil.rmtree(folder)
+                    print(f"  ✓ Removed {folder}/")
+                else:
+                    os.remove(folder)
+                    print(f"  ✓ Removed {folder}")
+
+        # Remove .pyc files recursively
+        for root, dirs, files in os.walk('.'):
+            for file in files:
+                if file.endswith('.pyc') or file.endswith('.pyo'):
+                    filepath = os.path.join(root, file)
+                    try:
+                        os.remove(filepath)
+                    except:
+                        pass
+        print("  ✓ Removed .pyc and .pyo files")
+
+        # Remove moviepy temp files
+        temp_dir = tempfile.gettempdir()
+        for file in os.listdir(temp_dir):
+            if 'mpy' in file.lower():
+                try:
+                    filepath = os.path.join(temp_dir, file)
+                    if os.path.isfile(filepath):
+                        os.remove(filepath)
+                except:
+                    pass
+        print("  ✓ Cleaned moviepy temp files")
+
+        print("✓ Full cleanup complete!")
     except Exception as e:
-        print(f"⚠ Warning: Could not clean up temp folders: {e}")
+        print(f"⚠ Warning: Could not complete full cleanup: {e}")
+
 
 def signal_handler(signum, frame):
     """Handle termination signals (Ctrl+C, etc.)"""
     print(f"\n\n🛑 Received signal {signum}, shutting down server...")
     cleanup_temp_folders()
     exit(0)
 
+
 # Register cleanup handlers
 atexit.register(cleanup_temp_folders)
 signal.signal(signal.SIGINT, signal_handler)   # Ctrl+C
 signal.signal(signal.SIGTERM, signal_handler)  # Termination signal
 
+
 @app.route('/')
 def index():
     return render_template('index.html')
 
+
+@app.route('/fonts/search')
+def search_fonts():
+    query = request.args.get('q', '').strip().lower()
+    if not query:
+        return jsonify({'fonts': []})
+
+    try:
+        resp = requests.get('https://fonts.google.com/metadata/fonts', timeout=10)
+        resp.raise_for_status()
+        content = resp.text
+        if content.startswith(")]}'"):
+            content = content[5:]
+        data = json.loads(content)
+    except Exception:
+        return jsonify({'fonts': [], 'error': 'Failed to reach Google Fonts'}), 502
+
+    items = data.get('familyMetadataList', []) if isinstance(data, dict) else []
+    matches = []
+    for item in items:
+        name = item.get('family', '')
+        if query in name.lower():
+            matches.append(name)
+        if len(matches) >= 15:
+            break
+    return jsonify({'fonts': matches})
+
+
+@app.route('/fonts/download', methods=['POST'])
+def download_font():
+    payload = request.get_json(silent=True) or {}
+    name = payload.get('name') or request.form.get('name')
+    if not name:
+        return jsonify({'success': False, 'message': 'Font name required'}), 400
+
+    path = install_google_font(name)
+    if path and os.path.exists(path):
+        return jsonify({'success': True, 'path': path, 'name': name})
+    return jsonify({'success': False, 'message': 'Could not download font'}), 502
+
+@app.route('/preview', methods=['POST'])
+def preview():
+    """Generate a preview video"""
+    session_id = str(uuid.uuid4())
+    session_upload_path = os.path.join(app.config['UPLOAD_FOLDER'], session_id)
+    session_images_path = os.path.join(session_upload_path, 'images')
+    os.makedirs(session_images_path, exist_ok=True)
+
+    try:
+        # Save audio file
+        if 'audio' not in request.files:
+            return jsonify({'success': False, 'message': 'No audio file'}), 400
+        audio_file = request.files['audio']
+        audio_path = os.path.join(session_upload_path, 'music.mp3')
+        audio_file.save(audio_path)
+
+        # Save images
+        if 'images' not in request.files:
+            return jsonify({'success': False, 'message': 'No images'}), 400
+        files = request.files.getlist('images')
+        for file in files:
+            if file.filename:
+                file.save(os.path.join(session_images_path, file.filename))
+
+        # Generate preview
+        success, message = generate_video_from_web(
+            image_folder=session_images_path,
+            audio_paths=[audio_path],
+            output_path=os.path.join(app.config['OUTPUT_FOLDER'], f'preview_{session_id}.mp4'),
+            intro_text=request.form.get('intro_text', 'Preview'),
+            outro_text=request.form.get('outro_text', 'End'),
+            config={}
+        )
+        if success:
+            return jsonify({'success': True, 'filename': f'preview_{session_id}.mp4'})
+        else:
+            return jsonify({'success': False, 'message': message}), 500
+
+    except Exception as e:
+        print(f"Preview error: {e}")
+        return jsonify({'success': False, 'message': str(e)}), 500
+    finally:
+        # Clean up uploads
+        try:
+            if os.path.exists(session_upload_path):
+                shutil.rmtree(session_upload_path)
+        except Exception:
+            pass
+
+@app.route('/get_preview/<filename>')
+def get_preview(filename):
+    """Stream preview video"""
+    # Validate the filename to prevent path traversal
+    if not filename:
+        return "Invalid filename", 400
+
+    # Disallow any path separators or parent directory references
+    if os.path.sep in filename or (os.path.altsep and os.path.altsep in filename) or '..' in filename:
+        return "Invalid filename", 400
+
+    # Enforce expected naming pattern for preview files
+    if not (filename.startswith('preview_') and filename.endswith('.mp4')):
+        return "Invalid filename", 400
+
+    # Build absolute paths and ensure the target stays within OUTPUT_FOLDER
+    output_folder = os.path.abspath(app.config['OUTPUT_FOLDER'])
+    preview_path = os.path.abspath(os.path.join(output_folder, filename))
+
+    if not preview_path.startswith(output_folder + os.path.sep):
+        return "Invalid filename", 400
+    if os.path.exists(preview_path):
+        return send_file(preview_path, mimetype='video/mp4')
+    return "Preview not found", 404
+
 @app.route('/create', methods=['POST'])
 def create_video():
+    # Clean up ALL preview files when creating final video
+    try:
+        for f in os.listdir(app.config['OUTPUT_FOLDER']):
+            if f.startswith('preview_') and f.endswith('.mp4'):
+                file_path = os.path.join(app.config['OUTPUT_FOLDER'], f)
+                try:
+                    os.remove(file_path)
+                    print(f"🗑️ Cleaned up preview: {f}")
+                except:
+                    pass
+    except Exception as e:
+        print(f"⚠️ Could not clean previews: {e}")
+
     # Create unique session ID to avoid conflicts
     session_id = str(uuid.uuid4())
     session_upload_path = os.path.join(app.config['UPLOAD_FOLDER'], session_id)

requirements.txt

@@ -6,4 +6,5 @@ numpy>=1.24.0
 tqdm>=4.66.0
 librosa>=0.10.0
 pillow-heif>=0.13.0
+requests>=2.31.0
 setuptools>=65.0.0