Final Commmit

app.py

@@ -1,8 +1,9 @@
 import time
 import streamlit as st
-from llama_index import ServiceContext, StorageContext, set_global_service_context, VectorStoreIndex, SimpleDirectoryReader, Document
+from llama_index import ServiceContext, StorageContext, set_global_service_context, VectorStoreIndex, Document
 from llama_index.embeddings import LangchainEmbedding
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
+from llama_index.chat_engine.condense_question import CondenseQuestionChatEngine
 from llama_index.llms import LlamaCPP
 from llama_index.llms.llama_utils import messages_to_prompt, completion_to_prompt
 from PyPDF2 import PdfReader
@@ -62,6 +63,9 @@ def main():
     # Storage Context
     storage_context = StorageContext.from_defaults()
     st.title("Llama-CPP Local LLM with RAG (Phi-2 RAG)")
+    # Credits
+    st.markdown(
+        "Made with ❤️️ By Danyaal Majid & Muhammad Bin Asif Using [HF Spaces](https://huggingface.co/spaces/DanyaalMajid/NLP-Final-LocalLLM-RAG)")
 
     pdf = st.file_uploader("Upload a PDF file", type=["pdf"])
 
@@ -72,8 +76,14 @@ def main():
         storage_context.docstore.add_documents(nodes)
         index = (VectorStoreIndex.from_documents(
             documents, service_context=service_context, storage_context=storage_context, llm=llm))
-        chat_engine = index.as_chat_engine(chat_mode="simple", verbose=True)
-
+        # chat_engine = index.as_chat_engine(chat_mode="simple", verbose=True)
+        custom_prompt = ""
+        query_engine = index.as_query_engine()
+        chat_engine = CondenseQuestionChatEngine.from_defaults(
+            query_engine=query_engine,
+            condense_question_prompt=custom_prompt,
+            verbose=True,
+        )
         # Initialize chat history
         if "messages" not in st.session_state:
             st.session_state.messages = []
@@ -110,10 +120,6 @@ def main():
             st.session_state.messages.append(
                 {"role": "assistant", "content": full_response})
 
-    # Credits
-    st.markdown(
-        "Made By Danyaal Majid & Muhammad Bin Asif [HuggingFace](https://huggingface.co/spaces/DanyaalMajid/NLP-Final-LocalLLM-RAG).")
-
 
 if __name__ == "__main__":
     main()