Update app.py

app.py

@@ -1,42 +1,31 @@
 import streamlit as st
-import tempfile
 
+# ✅ Correct imports (new structure)
 from langchain_community.document_loaders import PyPDFLoader, TextLoader
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
-from langchain.llms import HuggingFacePipeline
-from langchain.chains import RetrievalQA
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
+from langchain_community.llms import HuggingFacePipeline
 
+from langchain.chains import RetrievalQA
 from transformers import pipeline
 
-# -------------------------------
-# Page Config
-# -------------------------------
-st.set_page_config(page_title="RAG Chatbot", layout="wide")
-st.title("📄 Chat with Your Documents (RAG)")
-st.write("🚀 App started successfully")
 
 # -------------------------------
-# Load Documents (FIXED)
+# Load Documents
 # -------------------------------
 def load_documents(uploaded_files):
     documents = []
-
     for file in uploaded_files:
-        # Save file safely using temp file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=file.name) as tmp:
-            tmp.write(file.getbuffer())
-            temp_path = tmp.name
+        with open(file.name, "wb") as f:
+            f.write(file.getbuffer())
 
-        # Load based on type
         if file.name.endswith(".pdf"):
-            loader = PyPDFLoader(temp_path)
+            loader = PyPDFLoader(file.name)
         else:
-            loader = TextLoader(temp_path)
+            loader = TextLoader(file.name)
 
         documents.extend(loader.load())
-
     return documents
 
 
@@ -51,39 +40,30 @@ def split_documents(documents):
     return splitter.split_documents(documents)
 
 
-# -------------------------------
-# Cached Embeddings (IMPORTANT)
-# -------------------------------
-@st.cache_resource
-def get_embeddings():
-    return HuggingFaceEmbeddings(
-        model_name="sentence-transformers/all-MiniLM-L6-v2"
-    )
-
-
 # -------------------------------
 # Create Vector Store
 # -------------------------------
 def create_vectorstore(chunks):
-    embeddings = get_embeddings()
+    embeddings = HuggingFaceEmbeddings(
+        model_name="sentence-transformers/all-MiniLM-L6-v2"
+    )
     return FAISS.from_documents(chunks, embeddings)
 
 
 # -------------------------------
-# Cached LLM (IMPORTANT)
+# Load Local LLM (FREE)
 # -------------------------------
-@st.cache_resource
 def load_llm():
     pipe = pipeline(
-        "text-generation",
-        model="google/flan-t5-small",  # lightweight model
-        max_length=256
+        "text2text-generation",   # ✅ FIXED
+        model="google/flan-t5-base",
+        max_length=512
     )
     return HuggingFacePipeline(pipeline=pipe)
 
 
 # -------------------------------
-# Build QA Chain
+# Build QA Chain (with strict prompt)
 # -------------------------------
 def build_qa(vectorstore):
     llm = load_llm()
@@ -91,37 +71,35 @@ def build_qa(vectorstore):
 
     qa = RetrievalQA.from_chain_type(
         llm=llm,
-        retriever=retriever,
-        return_source_documents=False
+        retriever=retriever
     )
     return qa
 
 
 # -------------------------------
-# UI - Upload
+# Streamlit UI
 # -------------------------------
+st.set_page_config(page_title="RAG Chatbot", layout="wide")
+st.title("📄 Chat with Your Documents (RAG)")
+
 uploaded_files = st.file_uploader(
     "Upload PDF or TXT files",
     accept_multiple_files=True
 )
 
 if uploaded_files:
-    with st.spinner("📄 Processing documents..."):
+    with st.spinner("Processing documents..."):
         docs = load_documents(uploaded_files)
         chunks = split_documents(docs)
         vectorstore = create_vectorstore(chunks)
         qa_chain = build_qa(vectorstore)
 
-    st.success("✅ Documents ready!")
+    st.success("Documents ready!")
 
-    # -------------------------------
-    # User Query
-    # -------------------------------
-    query = st.text_input("💬 Ask a question from your documents")
+    query = st.text_input("Ask a question from your documents")
 
     if query:
-        with st.spinner("🤖 Generating answer..."):
+        with st.spinner("Generating answer..."):
             result = qa_chain.run(query)
-
-        st.markdown("### 🧠 Answer:")
-        st.write(result)
+            st.write("### Answer:")
+            st.write(result)