Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 # Set the page layout for Streamlit
 st.set_page_config(layout="wide")
 
-# Initialize TAPAS pipeline for table-based question answering
+# Initialize TAPAS pipeline for table-based question answering (multilingual)
 tqa = pipeline(task="table-question-answering", 
               model="google/tapas-large-finetuned-wtq",
               device=0)  # Assuming GPU is available, otherwise set device="cpu"
@@ -24,6 +24,12 @@ st.markdown("""
     Upload your data and ask questions about the data to generate visualizations.
 """)
 
+# Language Selection
+language = st.selectbox(
+    "Select the language of your question",
+    ("English", "German", "French", "Spanish", "Italian", "Others")
+)
+
 # File uploader in the sidebar
 file_name = st.sidebar.file_uploader("Upload file:", type=['csv', 'xlsx'])
 
@@ -56,7 +62,7 @@ else:
         st.error(f"Error reading file: {str(e)}")
 
     # User input for the question
-    question = st.text_input('Ask your graph-related question')
+    question = st.text_input(f'Ask your graph-related question in {language}')
 
     with st.spinner():
         if st.button('Generate Graph'):