Update app.py

app.py

@@ -42,6 +42,10 @@ else:
     st.write(f"Data Preview ({csv_file.name}):")
     st.dataframe(data.head())
 
+# Display column names
+st.write("**Available Columns:**")
+st.write(", ".join(data.columns.tolist()))
+
 # Step 2: Load CSV data into SQLite database
 db_file = 'my_database.db'
 conn = sqlite3.connect(db_file)
@@ -63,18 +67,34 @@ agent_executor = create_sql_agent(
     agent_executor_kwargs={"return_intermediate_steps": True}
 )
 
+# Step 3: Sample Questions
+st.write("**Sample Questions:**")
+sample_questions = [
+    "Provide an analysis of the data.",
+    "Summarize the data for me.",
+    "Can you offer any recommendations based on the data sets?",
+    "What are the total sales across regions?"
+]
+
+def set_sample_question(question):
+    st.session_state['user_input'] = question
+    process_input()
+
+for question in sample_questions:
+    st.button(question, on_click=set_sample_question, args=(question,))
+
 # Step 4: Define the callback function
 def process_input():
-    user_prompt = st.session_state['user_input']
+    user_prompt = st.session_state.get('user_input', '')
 
     if user_prompt:
         try:
             # Append user message to history
             st.session_state.history.append({"role": "user", "content": user_prompt})
 
-            # Use the agent to get the response WITHOUT 'return_intermediate_steps'
+            # Use the agent to get the response
             with st.spinner("Processing..."):
-                response = agent_executor(user_prompt)  # Removed 'return_intermediate_steps' here
+                response = agent_executor(user_prompt)
 
             # Extract the final answer and the data from intermediate steps
             final_answer = response['output']
@@ -123,16 +143,20 @@ def process_input():
             st.session_state.history.append({"role": "assistant", "content": insights})
         except Exception as e:
             logging.error(f"An error occurred: {e}")
-            assistant_response = f"Error: {e}"
+
+            # Check for specific errors related to missing columns
+            if "no such column" in str(e).lower():
+                assistant_response = "Error: One or more columns referenced do not exist in the data."
+            else:
+                assistant_response = f"Error: {e}"
+
             st.session_state.history.append({"role": "assistant", "content": assistant_response})
 
         # Reset user input
         st.session_state['user_input'] = ''
 
- 
-
-
 # Step 5: Display conversation history
+st.write("## Conversation History")
 for message in st.session_state.history:
     if message['role'] == 'user':
         st.markdown(f"**User:** {message['content']}")