Upload app.py

app.py

@@ -65,7 +65,7 @@ if "value" not in st.session_state:
 if "faq" not in st.session_state:
     st.session_state["faq"] = None
 
-st.divider()
+# st.divider()
 
 def upload_file():
     uploaded_file = st.file_uploader("Upload a file", type=["pdf"])
@@ -114,9 +114,9 @@ def create_pickle_file(filepath):
         pickle.dump(faiss_index, f)
 
 
-uploaded_file_name = upload_file()
-if uploaded_file_name is not None:
-    create_pickle_file(uploaded_file_name)
+# uploaded_file_name = upload_file()
+# if uploaded_file_name is not None:
+#     create_pickle_file(uploaded_file_name)
 
 def highlight_pdf(file_path, text_to_highlight, page_numbers):
     # Create a temporary file to save the modified PDF
@@ -136,11 +136,9 @@ def highlight_pdf(file_path, text_to_highlight, page_numbers):
     # Tokenize the text into words
     words = text_to_highlight.split()
 
-   
-
     # Remove stopwords
-    # stop_words = set(stopwords.words("english"))
-    # words = [word for word in words if word.lower() not in stop_words]
+    stop_words = set(stopwords.words("english"))
+    words = [word for word in words if word.lower() not in stop_words]
     
     print(words)
     
@@ -213,7 +211,7 @@ def get_faiss_semantic_index():
             with open(index_path, "rb") as f:
                 faiss_index = pickle.load(f)
                 sleep(0.1)
-        st.write("Embeddings loaded successfully.")
+        # st.write("Embeddings loaded successfully.")
         return faiss_index
     except Exception as e:
         st.error(f"Error loading embeddings: {e}")
@@ -224,14 +222,17 @@ print(faiss_index)
 # def promt_engineer(text):
 PROMPT_TEMPLATE = """
 Instructions:
---------------------------------------------------------
-Answer the question based only on the following context:
-Please dont show human instructions in the answer
-----------------------------------------------------------
+-----------------------------------------------------------------------------------------
+Answer the question  only based on the below context:
+- You're a Vedic AI expert in the Hindu Vedic scriptures.
+- Always try to provide Keep it simple answers in nice format without incomplete sentence.
+- Give the answer atleast 5 seperate lines addition to the title info.
+- provide Title: <title> Chapter: <chapter> Text No: <textnumber> Page No: <pagenumber> 
+------------------------------------------------------------------------------------------
 
 {context}
 
------------------------------------------------------------
+----------------------------------------------------------------------------------
 
 Answer the question based on the above context: {question}
 """
@@ -259,7 +260,7 @@ def chat_actions():
     # query_embedding = model.encode(st.session_state["chat_input"])
     query = st.session_state["chat_input"]
     if faiss_index is not None:
-        docs = faiss_index.similarity_search(query, k=2)
+        docs = faiss_index.similarity_search(query, k=6)
     else:
         st.error("Failed to load embeddings.")
     # docs = faiss_index.similarity_search(query, k=2)
@@ -278,7 +279,7 @@ def chat_actions():
     result = ""
     try:
         llm = HuggingFaceHub(
-            repo_id="meta-llama/Meta-Llama-3-8B-Instruct"
+            repo_id="meta-llama/Meta-Llama-3-8B-Instruct", model_kwargs={"temperature": 0.1, "max_new_tokens": 256, "task":"text-generation"}
         )
         response_text = llm.invoke(prompt)
         escaped_query = re.escape(query)
@@ -317,11 +318,14 @@ with st.sidebar:
 
     faq = st.selectbox(
     "Select Your Favorite Scriptures",
-    ("Why does atheism exist even when all questions are answered in Bhagavad Gita?", 
+    ("what is jeeva and paramathma?",
+     "What are the Krishna told to Arjuna?",
+     "What are the key points from Krishna?",
+     "Why does atheism exist even when all questions are answered in Bhagavad Gita?", 
      "Why don’t all souls surrender to Lord Krishna, although he has demonstrated that everyone is part and parcel of Him, and all can be liberated from all sufferings by surrendering to Him?",
      "Why do souls misuse their independence by rebelling against Lord Krishna?",
      "How do I put an end to my suffering in this world?",
-     "What are the key points from Krishna?"),
+     "what is the reason behind Krishna decided to go far war?"),
     # index=None,
     # placeholder="Select scriptures...",
     key="faq",