Update app.py

app.py

@@ -4,40 +4,36 @@ import tempfile
 import uuid
 import google.generativeai as genai
 import requests
-from flask import Flask, request, render_template, send_from_directory, url_for, flash
+from flask import Flask, request, render_template, send_from_directory, url_for, flash, jsonify
 from moviepy.video.io.VideoFileClip import VideoFileClip
 from moviepy.audio.io.AudioFileClip import AudioFileClip
 from werkzeug.utils import secure_filename
 from dotenv import load_dotenv
 
-# --- 1. INITIALIZE FLASK APP AND LOAD SECRETS ---
+# Initialize Flask app
 load_dotenv()
-app = Flask(__name__)
+app = Flask(__name__, template_folder='templates', static_folder='static')
 
-# Load secrets from environment variables
+# Configuration
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 TTS_API_URL = os.getenv("TTS_API_URL")
 
-# Validate required configurations
-if not GEMINI_API_KEY:
-    raise ValueError("SECURITY ERROR: GEMINI_API_KEY not found in .env file!")
-if not TTS_API_URL:
-    raise ValueError("CONFIGURATION ERROR: TTS_API_URL not found in .env file!")
+if not GEMINI_API_KEY or not TTS_API_URL:
+    raise ValueError("Missing required environment variables")
 
-# Configure Gemini AI
 genai.configure(api_key=GEMINI_API_KEY)
 
-# Configure directories
+# File storage setup
 UPLOAD_FOLDER = 'uploads'
 DOWNLOAD_FOLDER = 'downloads'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(DOWNLOAD_FOLDER, exist_ok=True)
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 app.config['DOWNLOAD_FOLDER'] = DOWNLOAD_FOLDER
-app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024  # 100 MB upload limit
-app.secret_key = os.urandom(24)  # Secure key for flash messages
+app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024  # 100MB
+app.secret_key = os.urandom(24)
 
-# --- 2. APPLICATION CONFIGURATION ---
+# Voice options
 VOICE_CHOICES = {
     "Male (Charon)": "Charon",
     "Female (Zephyr)": "Zephyr"
@@ -55,136 +51,143 @@ You are an expert AI scriptwriter. Your task is to watch the provided video and
 Say happily: வணக்கம்! [laugh] எப்படி இருக்கீங்க? Whisper mysteriously: அந்த ரகசியம் எனக்கு மட்டும் தான் தெரியும்.
 """
 
-# --- 3. CORE APPLICATION FUNCTIONS ---
+def generate_tamil_script(video_path):
+    """Generate Tamil script using Gemini"""
+    try:
+        print("Uploading video to Gemini...")
+        video_file = genai.upload_file(video_path, mime_type="video/mp4")
+        
+        # Wait for processing
+        while video_file.state.name == "PROCESSING":
+            time.sleep(5)
+            video_file = genai.get_file(video_file.name)
+        
+        if video_file.state.name != "ACTIVE":
+            raise Exception(f"Gemini processing failed: {video_file.state.name}")
+        
+        print("Generating script...")
+        model = genai.GenerativeModel(model_name="models/gemini-2.5-flash")
+        response = model.generate_content([GEMINI_PROMPT, video_file])
+        genai.delete_file(video_file.name)
+        
+        if hasattr(response, 'text') and response.text:
+            return " ".join(response.text.strip().splitlines())
+        raise Exception("No valid script generated")
+    except Exception as e:
+        print(f"Script generation failed: {str(e)}")
+        raise
 
-def generate_tamil_script(video_file_path):
-    """Generates a Tamil script from the video using Gemini AI."""
-    print("Uploading video to Gemini for transcription...")
-    video_file = genai.upload_file(video_file_path, mime_type="video/mp4")
-    
-    # Wait for file processing
-    while video_file.state.name == "PROCESSING":
-        time.sleep(5)
-        video_file = genai.get_file(video_file.name)
-    
-    if video_file.state.name != "ACTIVE":
-        raise Exception(f"Gemini file processing failed: {video_file.state.name}")
-    
-    print("Generating script...")
-    model = genai.GenerativeModel(model_name="models/gemini-2.5-flash")
-    response = model.generate_content([GEMINI_PROMPT, video_file])
-    genai.delete_file(video_file.name)
-    
-    if hasattr(response, 'text') and response.text:
-        return " ".join(response.text.strip().splitlines())
-    raise Exception("No valid script was generated by Gemini.")
-
-def generate_audio_track(script_text, voice_name, is_cheerful, output_path):
-    """Generates audio from script using TTS API."""
-    print(f"Generating audio (Voice: {voice_name}, Cheerful: {is_cheerful})")
-    payload = {
-        "text": script_text,
-        "voice_name": voice_name,
-        "cheerful": is_cheerful
-    }
-    
-    response = requests.post(TTS_API_URL, json=payload, timeout=300)
-    if response.status_code == 200:
+def generate_audio_track(text, voice, cheerful, output_path):
+    """Generate audio using TTS API"""
+    try:
+        print("Generating audio track...")
+        payload = {
+            "text": text,
+            "voice_name": voice,
+            "cheerful": cheerful
+        }
+        
+        response = requests.post(TTS_API_URL, json=payload, timeout=300)
+        if response.status_code != 200:
+            raise Exception(f"TTS API error: {response.status_code}")
+        
         with open(output_path, "wb") as f:
             f.write(response.content)
-        return True
-    raise Exception(f"TTS API Error: {response.status_code} - {response.text}")
-
-def replace_video_audio(video_path, new_audio_path, output_path):
-    """Replaces the audio track of a video file."""
-    print("Replacing video audio...")
-    video_clip = None
-    audio_clip = None
-    
+        print("Audio track generated successfully")
+    except Exception as e:
+        print(f"Audio generation failed: {str(e)}")
+        raise
+
+def replace_video_audio(video_path, audio_path, output_path):
+    """Replace video audio track"""
+    video = None
+    audio = None
     try:
-        video_clip = VideoFileClip(video_path)
-        audio_clip = AudioFileClip(new_audio_path)
-        video_clip.audio = audio_clip
-        video_clip.write_videofile(
+        print("Replacing audio track...")
+        video = VideoFileClip(video_path)
+        audio = AudioFileClip(audio_path)
+        video.audio = audio
+        video.write_videofile(
             output_path,
             codec="libx264",
             audio_codec="aac",
-            logger='bar'
+            threads=4,
+            verbose=False
         )
+        print("Video processing complete!")
+    except Exception as e:
+        print(f"Video processing failed: {str(e)}")
+        raise
     finally:
-        if audio_clip:
-            audio_clip.close()
-        if video_clip:
-            video_clip.close()
+        if video:
+            video.close()
+        if audio:
+            audio.close()
 
-# --- 4. FLASK ROUTES ---
-
-@app.route('/', methods=['GET'])
+@app.route('/')
 def index():
-    """Render the main upload page."""
+    """Main page"""
     return render_template('index.html', voices=VOICE_CHOICES)
 
 @app.route('/process', methods=['POST'])
 def process_video():
-    """Handle video upload and processing."""
-    input_video_path = None
-    temp_audio_path = None
+    """Handle video processing"""
+    if 'video' not in request.files:
+        flash("No video file uploaded", "error")
+        return render_template('index.html', voices=VOICE_CHOICES)
+    
+    file = request.files['video']
+    if file.filename == '':
+        flash("No file selected", "error")
+        return render_template('index.html', voices=VOICE_CHOICES)
+    
+    # Save uploaded file
+    filename = secure_filename(f"{uuid.uuid4()}_{file.filename}")
+    video_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+    file.save(video_path)
+    
+    # Get processing options
+    voice = request.form.get('voice', 'Charon')
+    cheerful = request.form.get('cheerful') == 'on'
     
     try:
-        # Validate file upload
-        if 'video' not in request.files or request.files['video'].filename == '':
-            flash("Please upload a video file.", "error")
-            return render_template('index.html', voices=VOICE_CHOICES)
+        # Generate script
+        script = generate_tamil_script(video_path)
         
-        # Save uploaded file
-        file = request.files['video']
-        filename = secure_filename(file.filename)
-        input_video_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        file.save(input_video_path)
-        
-        # Get processing options
-        voice_choice = request.form.get('voice', 'Charon')
-        is_cheerful = request.form.get('tone') == 'on'
-        
-        # Generate script and audio
-        script = generate_tamil_script(input_video_path)
-        
-        # Create temporary audio file
+        # Generate audio
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_audio:
-            temp_audio_path = temp_audio.name
+            audio_path = temp_audio.name
         
-        generate_audio_track(script, voice_choice, is_cheerful, temp_audio_path)
+        generate_audio_track(script, voice, cheerful, audio_path)
         
         # Create dubbed video
-        final_video_name = f"dubbed_{filename}"
-        final_video_path = os.path.join(app.config['DOWNLOAD_FOLDER'], final_video_name)
-        replace_video_audio(input_video_path, temp_audio_path, final_video_path)
+        final_filename = f"dubbed_{filename}"
+        final_path = os.path.join(app.config['DOWNLOAD_FOLDER'], final_filename)
+        replace_video_audio(video_path, audio_path, final_path)
         
-        flash("Video processing complete!", "success")
-        return render_template(
-            'index.html',
-            voices=VOICE_CHOICES,
-            result_video=url_for('serve_video', filename=final_video_name),
-            script=script
-        )
+        # Cleanup
+        os.unlink(audio_path)
+        os.unlink(video_path)
         
-    except Exception as e:
-        print(f"Processing error: {str(e)}")
-        flash(f"An error occurred: {str(e)}", "error")
-        return render_template('index.html', voices=VOICE_CHOICES)
+        return jsonify({
+            'status': 'success',
+            'video_url': url_for('download', filename=final_filename),
+            'script': script
+        })
         
-    finally:
-        # Clean up temporary files
-        if input_video_path and os.path.exists(input_video_path):
-            os.remove(input_video_path)
-        if temp_audio_path and os.path.exists(temp_audio_path):
-            os.remove(temp_audio_path)
-
-@app.route('/downloads/<filename>')
-def serve_video(filename):
-    """Serve the processed video file."""
+    except Exception as e:
+        # Cleanup on error
+        if os.path.exists(video_path):
+            os.unlink(video_path)
+        return jsonify({
+            'status': 'error',
+            'message': str(e)
+        }), 500
+
+@app.route('/download/<filename>')
+def download(filename):
+    """Serve processed video"""
     return send_from_directory(app.config['DOWNLOAD_FOLDER'], filename)
 
-# --- 5. APPLICATION ENTRY POINT ---
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)