edit interface

app.py

@@ -15,39 +15,41 @@ st.set_page_config(page_title="Table Question Answering", layout="wide")
 st.title("📊 Table Question Answering")
 st.write("""
 Welcome to the **Table Question Answering** app! 
-You can upload or paste your own table and ask questions about it.
+You can upload your own table and ask questions about it.
 The model will analyze the table and provide accurate answers.
 """)
 
-
-# Define the table data
-data = {
-    "year": [1896, 1900, 1904, 2004, 2008, 2012],
-    "city": ["athens", "paris", "st. louis", "athens", "beijing", "london"]
-}
-table = pd.DataFrame(data)
-table = table.astype(str)  #ensure all values are strings
-
-st.title("Table Question Answering")
-st.write("### Input Table")
-st.dataframe(table)
-
-query = st.text_input("Ask a question about the table:", "In which year did beijing host the Olympic Games?")
-
-if "history" not in st.session_state:
-    st.session_state.history = []
-
-#process query and display result
-if st.button("Get Answer"):
-    result = pipe(table=table, query=query)
-    st.session_state.history.append({"query": query, "answer": result["answer"]})
-    st.write("### Answer")
-    st.write(result["answer"])
-
-#show history
-if st.session_state.history:
-    st.write("### Previous Questions and Answers")
-    for item in st.session_state.history:
-        st.write(f"**Q:** {item['query']}")
-        st.write(f"**A:** {item['answer']}")
-        st.write("---")
+# Step 1: Table input
+st.header("📂 Step 1: Upload your table")
+# Option to upload a excel file
+uploaded_file = st.file_uploader("Upload your Excel file:", type=["xlsx"])
+if uploaded_file is not None:
+    try:
+        table = pd.read_excel(uploaded_file)
+        st.success("Table uploaded successfully!")
+        st.write("### Preview of Uploaded Table:")
+        st.dataframe(table, use_container_width=True)
+    except Exception as e:
+        st.error(f"Error reading the file: {e}")
+        table = None
+else:
+    st.info("Please upload an Excel file to proceed.")
+    table = None
+
+st.divider()
+st.header("❓ Step 2: Ask a Question")
+query = st.text_input("Ask a question about the table:")
+
+
+st.divider()
+if st.button("🔍 Get Answer"):
+    if table is not None and query:
+        table = table.astype(str)  # Ensure all values are strings
+        result = pipe(table=table, query=query)
+        st.success("Answer:")
+        st.markdown(f"### {result['answer']}")
+    else:
+        st.warning("Please upload a table and enter a question before clicking 'Get Answer'.")
+
+st.divider()
+st.markdown("**❤️**")