diff --git a/senju/main.py b/senju/main.py
index ef2d676..b1641bb 100644
--- a/senju/main.py
+++ b/senju/main.py
@@ -55,5 +55,6 @@ def generate_haiku():
if request.method == 'POST':
json_data = request.get_json()
prompt = json_data["prompt"]
- haiku = request_haiku(prompt)
- return haiku.get_json()
+ haiku = request_haiku(prompt)
+ id = store.save_haiku(haiku)
+ return str(id)
diff --git a/senju/templates/prompt.html b/senju/templates/prompt.html
index e0387a4..ec0cf93 100644
--- a/senju/templates/prompt.html
+++ b/senju/templates/prompt.html
@@ -57,10 +57,12 @@ document.getElementById("submit-btn").addEventListener("click", function() {
},
body: JSON.stringify({'prompt': userInput})
})
- .then(response => response.json())
+ .then(response => response.text())
.then(data => {
console.log(data);
- responseText.innerHTML = data[0] + "
" + data[1] + "
" + data[2];
+ //responseText.innerHTML = data[0] + "
" + data[1] + "
" + data[2];
+ let id = parseInt(data, 10);
+ window.location.replace("/haiku/"+id);
})
.catch(error => {
document.getElementById('result').innerHTML = 'Error: ' + error.message;