fixes

app.py

@@ -5,14 +5,13 @@ DEMO_SERVER = "4.208.9.167:80"
 
 def run_query( prompt):
     POLICY = "./cce_policy.txt"
-    if prompt != True:
-        return("⛔ Error: invalid user input for query")
-    if prompt.len() == 0:
+    if len(prompt) == 0:
         return("⛔ Error: please add your query for automated code completion")
     try:
         with SecureSession(f"http://{DEMO_SERVER}", POLICY) as secure_session:
             res = secure_session.post(endpoint="/generate", json={"input_text": prompt})
-            return("✅ Query successful\n" + res.text.replace('\\n', '\n'))
+            cleaned = res.text.replace('\\n', '\n').split('\n\n')[0].split(':"')[1]
+            return("✅ Query successful\n" + cleaned)
     except Exception as err:
         return(f"⛔ Query failed!\n{err}")