Update app.py

app.py

@@ -25,18 +25,25 @@ app.config.update({
     'ALLOWED_EXTENSIONS': {'mp4', 'mov', 'webm', 'avi'}
 })
 
-# Create directories if they don't exist
+# Create directories
 os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 os.makedirs(app.config['DOWNLOAD_FOLDER'], exist_ok=True)
 
 # Initialize Gemini AI
 genai.configure(api_key=app.config['GEMINI_API_KEY'])
 
-# Constants
-VOICE_CHOICES = {
-    "Male (Deep Voice)": "deep_male",
-    "Female (Soft Tone)": "soft_female",
-    "Neutral (Professional)": "neutral"
+# Voice Configuration
+VOICE_OPTIONS = {
+    "Male (Charon)": {
+        "id": "charon",
+        "description": "Deep, authoritative male voice",
+        "tone_options": ["neutral", "serious", "cheerful"]
+    },
+    "Female (Zephyr)": {
+        "id": "zephyr",
+        "description": "Soft, clear female voice",
+        "tone_options": ["neutral", "gentle", "energetic"]
+    }
 }
 
 GEMINI_PROMPT = """
@@ -69,7 +76,7 @@ def generate_script(video_path):
         if video_file.state.name != "ACTIVE":
             raise Exception("Gemini processing failed")
         
-        model = genai.GenerativeModel("models/gemini-2.5-flash")
+        model = genai.GenerativeModel("models/gemini-pro-vision")
         response = model.generate_content([GEMINI_PROMPT, video_file])
         genai.delete_file(video_file.name)
         
@@ -78,15 +85,16 @@ def generate_script(video_path):
         print(f"Gemini Error: {str(e)}")
         raise
 
-def generate_audio(script, voice, tone):
-    """Generate audio using TTS API"""
+def generate_audio(script, voice_id, tone):
+    """Generate audio using TTS API with specific voice"""
     try:
         response = requests.post(
             app.config['TTS_API_URL'],
             json={
                 "text": script,
-                "voice": voice,
-                "tone": tone
+                "voice_id": voice_id,
+                "tone": tone,
+                "language": "tamil"
             },
             timeout=300
         )
@@ -131,22 +139,22 @@ def process_video(input_path, audio_data, output_filename):
 
 @app.route('/', methods=['GET'])
 def home():
-    return render_template('index.html', voices=VOICE_CHOICES)
+    return render_template('index.html', voices=VOICE_OPTIONS)
 
 @app.route('/process', methods=['POST'])
 def process():
     if 'video' not in request.files:
         flash('No file selected', 'error')
-        return render_template('index.html', voices=VOICE_CHOICES)
+        return render_template('index.html', voices=VOICE_OPTIONS)
     
     file = request.files['video']
     if file.filename == '':
         flash('No file selected', 'error')
-        return render_template('index.html', voices=VOICE_CHOICES)
+        return render_template('index.html', voices=VOICE_OPTIONS)
     
     if not allowed_file(file.filename):
         flash('Invalid file type. Allowed: MP4, MOV, WEBM, AVI', 'error')
-        return render_template('index.html', voices=VOICE_CHOICES)
+        return render_template('index.html', voices=VOICE_OPTIONS)
     
     try:
         # Save uploaded file
@@ -155,12 +163,13 @@ def process():
         file.save(input_path)
         
         # Get processing options
-        voice = request.form.get('voice', 'neutral')
-        tone = 'cheerful' if request.form.get('tone') == 'on' else 'neutral'
+        voice_name = request.form.get('voice')
+        voice_id = VOICE_OPTIONS[voice_name]['id']
+        tone = request.form.get('tone', 'neutral')
         
         # Generate script and audio
         script = generate_script(input_path)
-        audio_data = generate_audio(script, voice, tone)
+        audio_data = generate_audio(script, voice_id, tone)
         
         # Process video
         output_filename = f"dubbed_{filename}"
@@ -169,11 +178,13 @@ def process():
         flash('Processing completed successfully!', 'success')
         return render_template('result.html', 
                             video_url=url_for('download', filename=output_filename),
-                            script=script)
+                            script=script,
+                            voice_used=voice_name,
+                            tone_used=tone)
     
     except Exception as e:
         flash(f'Processing failed: {str(e)}', 'error')
-        return render_template('index.html', voices=VOICE_CHOICES)
+        return render_template('index.html', voices=VOICE_OPTIONS)
     finally:
         if 'input_path' in locals() and os.path.exists(input_path):
             os.remove(input_path)
@@ -183,4 +194,4 @@ def download(filename):
     return send_from_directory(app.config['DOWNLOAD_FOLDER'], filename)
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host='0.0.0.0', port=5000, debug=True)