Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,10 @@
 import streamlit as st
-from langchain.document_loaders import PyPDFLoader, TextLoader, UnstructuredWordDocumentLoader, CSVLoader
+from langchain_community.document_loaders import (
+    PyPDFLoader,
+    TextLoader,
+    UnstructuredWordDocumentLoader,
+    CSVLoader
+)
 from langchain.vectorstores import FAISS
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.embeddings import HuggingFaceEmbeddings
@@ -50,6 +55,7 @@ vectorstore = load_files(uploaded_files) if uploaded_files else None
 
 if vectorstore:
     retriever = vectorstore.as_retriever()
+    
     llm = HuggingFaceEndpoint(
         repo_id="AI-Sweden-Models/Llama-3-8B-instruct",
         task="text-generation",
@@ -67,9 +73,11 @@ if vectorstore:
     if "messages" not in st.session_state:
         st.session_state.messages = []
 
+    # Display previous messages
     for msg in st.session_state.messages:
         st.chat_message(msg["role"]).markdown(msg["content"])
 
+    # User input
     user_input = st.chat_input("Ask something...")
     if user_input:
         st.chat_message("user").markdown(user_input)
@@ -81,6 +89,7 @@ if vectorstore:
 else:
     st.warning("Upload files to start querying.")
 
+# Clear chat button
 if st.button("Clear Chat"):
     st.session_state.messages = []
     st.experimental_rerun()