app.py

import streamlit as st
import numpy as np
import tensorflow as tf
from tf_coder.value_search import colab_interface
from tf_coder.value_search import value_search_settings as settings_module

inputs = st.text_area('A dict mapping input variable names to input tensors.', placeholder='''
    {
    'rows': [10, 20, 30],
    'cols': [1, 2, 3, 4],
}
    ''')
output = st.text_area('The corresponding output tensor.', placeholder='''
         [[11, 12, 13, 14],
          [21, 22, 23, 24],
          [31, 32, 33, 34]]
    ''')
    
constants = st.text_area('A list of relevant scalar constants, if any.', placeholder='[]' )

description = st.text_input('An English natural language description of the tensor manipulation', placeholder='Add two vectors with broadcasting to get a matrix')