Update app.py

app.py

@@ -5,7 +5,7 @@ from pathlib import Path
 from typing import List
 import re
 import tempfile
-from flask import Flask, request, render_template, send_file, url_for
+from flask import Flask, request, render_template, send_file
 
 app = Flask(__name__)
 
@@ -76,38 +76,29 @@ class ElevenlabsTTS:
             f.write(combined_audio.getvalue())
         return filename.as_posix()
 
-# Serve static audio files
-app.config['AUDIO_FOLDER'] = tempfile.gettempdir()
-
-@app.route('/audio/<filename>')
-def serve_audio(filename):
-    return send_file(Path(app.config['AUDIO_FOLDER']) / filename, mimetype='audio/mpeg')
-
 # Web Interface
 tts_provider = ElevenlabsTTS()
 
 @app.route('/', methods=['GET', 'POST'])
 def index():
-    audio_url = None
-    download_filename = None
-    error = None
-    
     if request.method == 'POST':
-        text = request.form.get('text')
-        voice = request.form.get('voice', 'Brian')
-        try:
-            audio_file = tts_provider.tts(text, voice)
-            filename = Path(audio_file).name
-            audio_url = url_for('serve_audio', filename=filename)
-            download_filename = f"{voice}_output.mp3"
-        except Exception as e:
-            error = str(e)
-    
-    return render_template('index.html', 
-                         voices=tts_provider.all_voices.keys(),
-                         audio_url=audio_url,
-                         download_filename=download_filename,
-                         error=error)
+        if 'generate' in request.form:
+            text = request.form.get('text')
+            voice = request.form.get('voice', 'Brian')
+            try:
+                audio_file = tts_provider.tts(text, voice)
+                return send_file(audio_file, mimetype='audio/mpeg', as_attachment=True, download_name=f"{voice}_output.mp3")
+            except Exception as e:
+                return render_template('index.html', error=str(e), voices=tts_provider.all_voices.keys())
+        elif 'preview' in request.form:
+            voice = request.form.get('preview_voice')
+            try:
+                preview_text = f"Hello, this is {voice}'s voice sample."
+                audio_file = tts_provider.tts(preview_text, voice)
+                return send_file(audio_file, mimetype='audio/mpeg', as_attachment=False)
+            except Exception as e:
+                return render_template('index.html', error=str(e), voices=tts_provider.all_voices.keys())
+    return render_template('index.html', voices=tts_provider.all_voices.keys())
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0', port=5000)