Update app.py

app.py

@@ -62,8 +62,11 @@ def download_youtube_video(url, output_folder):
     output_path_template = os.path.join(output_folder, os.path.splitext(unique_filename)[0])
     ydl_opts = {
         'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
-        'outtmpl': output_path_template, 'merge_output_format': 'mp4',
-        'quiet': False, 'noplaylist': True, 'progress': True,
+        'outtmpl': output_path_template, 
+        'merge_output_format': 'mp4',
+        'quiet': False, 
+        'noplaylist': True, 
+        'progress': True,
     }
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
@@ -124,13 +127,14 @@ def replace_video_audio(video_path, new_audio_path, output_path):
 @app.route('/', methods=['GET'])
 def index():
     """Renders the main upload page."""
-    return render_template('index.html')
+    return render_template('index.html', voice_choices=VOICE_CHOICES)
 
 @app.route('/process', methods=['POST'])
 def process_video():
     """Handles video input (upload or URL), processing, and renders the result."""
     input_video_path, temp_audio_path = None, None
     try:
+        # Handle YouTube URL or file upload
         youtube_url = request.form.get('youtube_url', '').strip()
         if youtube_url:
             input_video_path = download_youtube_video(youtube_url, app.config['UPLOAD_FOLDER'])
@@ -142,30 +146,38 @@ def process_video():
             file.save(input_video_path)
         else:
             flash("Please either upload a video file or provide a YouTube URL.")
-            return render_template('index.html')
+            return render_template('index.html', voice_choices=VOICE_CHOICES)
 
+        # Get user preferences
         voice_choice = request.form['voice_choice']
         is_cheerful = 'cheerful' in request.form
         voice_name = VOICE_CHOICES[voice_choice]
+        
+        # Generate Tamil script
         script = generate_tamil_script(input_video_path)
         
+        # Generate audio track
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_audio:
             temp_audio_path = temp_audio.name
         
         generate_single_audio_track(script, voice_name, is_cheerful, temp_audio_path)
+        
+        # Create final 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)
         
         flash("Video processing complete!", "success")
         return render_template('index.html', 
-                               result_video=url_for('serve_video', filename=final_video_name), 
-                               script=script)
+                             voice_choices=VOICE_CHOICES,
+                             result_video=url_for('serve_video', filename=final_video_name), 
+                             script=script)
     except Exception as e:
         print(f"An error occurred during processing: {e}")
         flash(f"An unexpected error occurred: {e}. Please check the console and try again.", "error")
-        return render_template('index.html')
+        return render_template('index.html', voice_choices=VOICE_CHOICES)
     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):
@@ -178,4 +190,4 @@ def serve_video(filename):
 
 # --- 5. APPLICATION ENTRY POINT ---
 if __name__ == '__main__':
-  app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)