chore: make it compatible with chatgpt

2024-10-13 10:25:06 -04:00 · 2024-10-13 10:25:06 -04:00 · fc357d17f2
commit fc357d17f2
parent dbd326504c
1 changed files with 5 additions and 1 deletions
--- 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