Update app.py

app.py

@@ -54,8 +54,11 @@ if prompt := st.chat_input("What is up?"):
         "--method", "global",
         prompt
     ]
-    result = subprocess.run(command, capture_output=True, text=True, check=True)
-    response = result.stdout
+    try:
+        result = subprocess.run(command, capture_output=True, text=True, check=True)
+        response = result.stdout
+    except subprocess.CalledProcessError as e:
+        response = e.output
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):