Update app.py

app.py

@@ -1,6 +1,9 @@
 import streamlit as st
 from tf_coder.value_search import colab_interface
 from tf_coder.value_search import value_search_settings
+import io
+from contextlib import redirect_stdout
+
 def get_problem():
   """Specifies a problem to run TF-Coder on. Edit this function!"""
   # A dict mapping input variable names to input tensors.
@@ -32,7 +35,7 @@ settings = value_search_settings.from_dict({
 
 i = st.text_area("input tensor","second test")
 
-inputs, output, constants, description = get_problem()
-t = colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
-for e in ["benchmark", "count", "index", "settings", "solutions", "statistics", "total_time", "value_set", "values_by_weight"]:
-    st.write(eval(f"t.{e}"))
+with io.StringIO() as buf, redirect_stdout(buf):
+    colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
+    output = buf.getvalue()
+st.write(output)