Update app.py

app.py

@@ -1,37 +1,33 @@
-from flask import Flask, request, jsonify, send_file from flask_cors import CORS from audio_generation import generate_audio import os import tempfile from werkzeug.utils import secure_filename import pytesseract from PIL import Image
-
-app = Flask(__name__) CORS(app)
-
-UPLOAD_FOLDER = tempfile.gettempdir() app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
-@app.route("/upload", methods=["POST"]) def upload_file(): if 'file' not in request.files: return jsonify({"error": "No file part"}), 400
-
-file = request.files['file']
-if file.filename == '':
-    return jsonify({"error": "No selected file"}), 400
-
-filename = secure_filename(file.filename)
-filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-file.save(filepath)
-
-ext = os.path.splitext(filename)[1].lower()
-if ext in ['.png', '.jpg', '.jpeg', '.bmp']:
-    text = pytesseract.image_to_string(Image.open(filepath))
-elif ext in ['.txt']:
-    with open(filepath, 'r', encoding='utf-8') as f:
-        text = f.read()
-else:
-    text = "[Unsupported file type for text extraction]"
-
-return jsonify({"text": text})
-
-@app.route("/speak", methods=["POST"]) def speak(): data = request.get_json() text = data.get("text", "")
-
-if not text:
-    return jsonify({"error": "No text provided"}), 400
-
-audio_path = generate_audio(text)
-return send_file(audio_path, mimetype='audio/wav', as_attachment=False)
-
-if __name__ == "__main__": 
+from flask import Flask, request, jsonify
+from AudioGeneration import generate_audio
+import os
+import uuid
+
+app = Flask(__name__)
+
+@app.route("/generate-audio", methods=["POST"])
+def generate_audio_endpoint():
+    try:
+        data = request.get_json()
+        text = data.get("text", "").strip()
+        if not text:
+            return jsonify({"error": "Text input is empty"}), 400
+
+        filename = f"{uuid.uuid4().hex}.wav"
+        audio_path = generate_audio(text, filename)
+        audio_url = f"/audio/{filename}"
+        return jsonify({"audio_url": audio_url})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/audio/<filename>")
+def serve_audio(filename):
+    return app.send_static_file(f"audio/{filename}")
+
+@app.route("/")
+def root():
+    return jsonify({"message": "TTS API is running!"})
+
+if __name__ == "__main__":
+    os.makedirs("static/audio", exist_ok=True)
     app.run(host="0.0.0.0", port=7860)