diff --git a/main.py b/main.py
index 607aa59..8037dee 100644
--- a/main.py
+++ b/main.py
@@ -11,7 +11,11 @@ def send_post_request(prompt):
     url = "http://localhost:19194/"  # Update this to your actual server URL
     payload = prompt.title()
     response = requests.post(url, json=payload)
-    return response.text
+    json = response.json()
+    if "response" in json:
+        return json["response"]
+    else:
+        return json
 
 
 # Custom CSS to fix the input box at the bottom of the page