feat: add sidebar

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import numpy as np
 from llm import load_llm, response_generator
 from sql import csv_to_sqlite, run_sql_query
 
@@ -13,6 +14,7 @@ llm = load_llm(repo_id, filename)
 st.title("CSV TO SQL")
 st.write("To start, Upload your CSV below 👇")
 if st.button("Example prompt"):
+    st.session_state.csv_file = "./data/sales.csv"
     st.session_state.db_name = "sales"
     st.session_state.table_name = "sales"
     csv_to_sqlite("./data/sales.csv", "sales", "sales")
@@ -42,6 +44,7 @@ with st.expander("Upload CSV"):
     table_name = st.text_input("Table Name")
     if st.button("Save"):
         if csv_file and db_name and table_name:
+            st.session_state.csv_file = csv_file
             st.session_state.db_name = db_name
             st.session_state.table_name = table_name
 
@@ -58,7 +61,10 @@ if "messages" not in st.session_state:
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         if "content" in message:
-            st.markdown(message["content"])
+            if message["role"] == "user":
+                st.markdown(message["content"])
+            else:
+                st.code(message["content"])
         if "result" in message:
             st.dataframe(message["result"])
 
@@ -84,10 +90,24 @@ if prompt := st.chat_input(
             messages=st.session_state.messages,
             question=prompt,
         )
-        response = st.markdown(response_sql)
+        response = st.code(response_sql)
         result = run_sql_query(db_name=st.session_state.db_name, query=response_sql)
         st.markdown(result)
         st.table(result)
 
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response_sql})
+
+with st.sidebar:
+    st.title("Data Previewer")
+    st.write("You can see you CSV file content here")
+    if (
+        "csv_file" in st.session_state
+        and "db_name" in st.session_state
+        and "table_name" in st.session_state
+    ):
+        result = run_sql_query(
+            db_name=st.session_state.db_name,
+            query=f"SELECT * FROM {st.session_state.table_name}",
+        )
+        st.dataframe(result)