import pandas as pd
import torch
from io import StringIO
import streamlit as st
from app.tapas import execute_query

query = st.text_input(label='Enter your query')
st.caption('Multiple queries separated by comma(,)')
# st.write('The current movie title is', title)
uploaded_file = st.file_uploader("Choose a csv file")

if uploaded_file is not None:
     dataframe = pd.read_csv(uploaded_file)         
     if query:
          query_result = execute_query(query.split(','), dataframe)
          st.markdown('**'Prediction'**')
          st.markdown('_'+query_result+'_')
     st.dataframe(dataframe)