Fix output

src/app.py

@@ -121,13 +121,14 @@ def main():
                 return get_response(input_text, depth + 1)
 
     if st.button("Generate"):
+        start = time()
         if input_text:
             with st.status("Generating response...") as status:
                 config.status = status
                 config.status.update(label=":question: Breaking down question")
                 response = get_response(input_text)
-                st.text(response)
-                config.status.update(label="Finished!", state="complete", expanded=True)
+                st.info(response)
+                config.status.update(label=f"Finished! ({time() - start:.2f}s)", state="complete", expanded=True)
         else:
             st.warning("Please enter some text to generate a response.")