Update app.py

app.py

@@ -168,16 +168,25 @@ def generate_tts():
         model = tts_models[language]
         inputs = processor(text_input, return_tensors="pt")
 
+        # Generate speech - using model(**inputs) instead of model.generate()
         with torch.no_grad():
-            output = model.generate(**inputs)
-
-        waveform = output.cpu().numpy().flatten()
-        output_filename = os.path.join(OUTPUT_DIR, f"{language}_tts.wav")
-        sf.write(output_filename, waveform, SAMPLE_RATE)
-
-        return jsonify({"file_url": f"/download/{language}_tts.wav"})
+            output = model(**inputs).waveform
+            waveform = output.squeeze().cpu().numpy()
+
+        # Save to file
+        output_filename = os.path.join(OUTPUT_DIR, f"{language}_output.wav")
+        # Use the model's sampling rate
+        sampling_rate = model.config.sampling_rate
+        sf.write(output_filename, waveform, sampling_rate)
+        print(f"✅ Speech generated! File saved: {output_filename}")
+
+        return jsonify({
+            "message": "TTS audio generated",
+            "file_url": f"/download/{language}_output.wav"
+        })
     except Exception as e:
-        return jsonify({"error": f"TTS failed: {e}"}), 500
+        print(f"❌ Error generating TTS: {e}")
+        return jsonify({"error": f"Internal server error: {str(e)}"}), 500
 
 
 @app.route("/download/<filename>", methods=["GET"])