Upload 3 files

app.py

@@ -0,0 +1,69 @@
+from dotenv import load_dotenv
+import streamlit as st
+from langchain_experimental.agents import create_csv_agent
+from langchain.llms import HuggingFaceHub
+from htmlTemplate import css, bot_template, user_template
+
+load_dotenv()
+
+def main():
+    st.set_page_config(
+        page_title= "QnA with CSV",
+        page_icon=":robot_face:",
+        layout="wide"
+    )
+    st.write(css, unsafe_allow_html=True)
+
+    st.title("QnA with CSV 🤖")
+    user_csv = st.file_uploader(label = "", type= "csv")
+    if not user_csv:
+        st.warning("Please Upload your CSV file 🚨")
+
+    llm = HuggingFaceHub(
+            repo_id = "mistralai/Mistral-7B-Instruct-v0.2",
+            model_kwargs = {
+                'max_new_tokens': 249, 
+                'temperature': 0.3,
+            }
+        )
+    user_input = st.chat_input("Ask your Question about CSV files",
+                               disabled= not user_csv)
+    
+    with st.sidebar:
+        st.subheader("Example Questions:")
+        example_questions = [
+            "What is the total number of rows in the CSV?",
+            "Can you show me the first 5 rows of the CSV?",
+            "What are the column names in the CSV?",
+            "How many columns does the CSV have?",
+            "What is the data type of a specific column in the CSV?",
+            "Can you provide a summary statistics for the numerical columns?",
+            "Are there any missing values in the CSV?",
+            "Can you filter the data based on a specific condition?",
+            "What is the average value of a numerical column?",
+        ]
+
+        selected_example = st.selectbox("Select an example question:", example_questions, disabled= not user_csv)
+
+        if not user_csv:
+            st.warning("Please Upload your CSV file 🚨")
+        
+        if st.button("Use Selected Example"):
+            user_input = selected_example
+    
+    if user_csv is not None:
+        agent = create_csv_agent(
+            llm = llm,
+            path = user_csv,
+            verbose = True,
+            handle_parsing_errors=True
+        )
+
+        if user_input is not None and user_input != "":
+            st.write(user_template.replace("{{MSG}}",user_input), unsafe_allow_html= True)         
+            with st.spinner("Processing..."):   
+                response = agent.run(user_input)
+                st.write(bot_template.replace("{{MSG}}",response), unsafe_allow_html= True)
+
+if __name__ == "__main__":
+    main()

htmlTemplate.py

@@ -0,0 +1,74 @@
+# Updated CSS
+# CSS Styles
+css = '''
+<style>
+    /* Styling for the chat messages */
+    .chat-message {
+        padding: 1rem;
+        border-radius: 0.5rem;
+        margin-bottom: 1rem;
+        display: flex;
+        border: 1px solid #d3d3d3; /* Add a subtle border */
+    }
+
+    /* Styling for user messages */
+    .chat-message.user {
+        background-color: #2b313e;
+    }
+
+    /* Styling for bot messages */
+    .chat-message.bot {
+        background-color: #475063;
+    }
+
+    /* Styling for the avatar */
+    .chat-message .avatar {
+        width: 15%; /* Adjust avatar size */
+    }
+
+    /* Styling for the avatar image */
+    .chat-message .avatar img {
+        max-width: 60px;
+        max-height: 60px;
+        border-radius: 50%;
+        object-fit: cover;
+    }
+
+    /* Styling for the message content */
+    .chat-message .message {
+        width: auto; /* Adjust message width */
+        padding: 0.75rem;
+        color: #fff;
+        margin-right: 1rem; /* Add margin to the left of the message */
+        margin-left: 1rem;
+    }
+
+    /* Styling for strong (name) in the message */
+    .chat-message .message strong {
+        margin-right: 0.25rem; /* Adjust the margin as needed */
+    }
+</style>
+'''
+
+# HTML Templates for Bot and User Messages
+bot_template = '''
+<div class="chat-message bot">
+    <div class="avatar">
+        <img src="https://i.ibb.co/3pvQJ2B/bot-icon.jpg">
+    </div>
+    <div class="message">
+        <strong>Baymax:</strong> {{MSG}}
+    </div>
+</div>
+'''
+
+user_template = '''
+<div class="chat-message user">
+    <div class="avatar">
+        <img src="https://i.ibb.co/HY8rRpL/human.jpg">
+    </div>    
+    <div class="message">
+        <strong>Tadashi:</strong> {{MSG}}
+    </div>
+</div>
+'''

requirements.txt

@@ -0,0 +1,7 @@
+huggingface_hub
+python-dotenv
+langchain
+sentence-transformers
+streamlit
+langchain-experimental
+tabulate