Update app.py

app.py

@@ -6,10 +6,11 @@ from contextlib import redirect_stdout
 from streamlit_ace import st_ace
 st.set_page_config(page_icon='👩‍💻', layout="wide")
 
+st.write('## Inputs')
+inputs = st_ace(placeholder="The input tensor(s) specified as a dictionary", value="{'rows': [10, 20, 30],\n'cols': [1,2,3,4]}", language="python", theme="solarized_dark")
+st.write('## Outputs')
+output = st_ace(placeholder="The output tensor", value="[[11, 12, 13, 14],\n[21, 22, 23, 24],\n[31, 32, 33, 34]]", language="python", theme="solarized_dark")
 
-#inputs = st.text_area('The input tensor(s) specified as key-value pairs', placeholder="{'rows': [10, 20, 30],'cols': [1,2,3,4]}")
-inputs = st_ace(value="{'rows': [10, 20, 30],\n'cols': [1,2,3,4]}", language="python", theme="solarized_dark")
-st.write(type(eval(inputs)))
 st.sidebar.header("Settings:")
 settings_kwargs = dict()
 settings_kwargs["require_all_inputs_used"] = st.sidebar.checkbox("Require All Inputs", value=True)
@@ -26,14 +27,10 @@ settings = value_search_settings.from_dict({
   })
 
 with io.StringIO() as buf, redirect_stdout(buf):
-    output = [[11, 12, 13, 14],
-            [21, 22, 23, 24],
-            [31, 32, 33, 34]]  
-
     constants = []
 
     description = 'add two vectors with broadcasting to get a matrix'
     inputs = {'rows': [10, 20, 30],'cols': [1,2,3,4]}
-    results = colab_interface.run_value_search_from_colab(inputs, output, constants, description, settings)
+    results = colab_interface.run_value_search_from_colab(eval(inputs), eval(output), constants, description, settings)
     stdout = buf.getvalue()
 st.code(stdout, language='bash')